arch specific deps

rapidsai · Oct 18, 2024 · f95aec7 · f95aec7
1 parent f061f27
commit f95aec7
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 6 deletions.
diff --git a/conda/environments/bench_ann_cuda-118_arch-aarch64.yaml b/conda/environments/bench_ann_cuda-118_arch-aarch64.yaml
@@ -25,8 +25,6 @@ dependencies:
 - glog>=0.6.0
 - h5py>=3.8.0
 - hnswlib=0.6.2
-- libaio
-- libboost-devel
 - libcublas-dev=11.11.3.6
 - libcublas=11.11.3.6
 - libcurand-dev=10.3.0.86
@@ -37,7 +35,6 @@ dependencies:
 - libcusparse=11.7.5.86
 - librmm==24.12.*,>=0.0.0a0
 - matplotlib
-- mkl-devel
 - nccl>=2.19
 - ninja
 - nlohmann_json>=3.11.2

diff --git a/conda/environments/bench_ann_cuda-125_arch-aarch64.yaml b/conda/environments/bench_ann_cuda-125_arch-aarch64.yaml
@@ -26,15 +26,12 @@ dependencies:
 - glog>=0.6.0
 - h5py>=3.8.0
 - hnswlib=0.6.2
-- libaio
-- libboost-devel
 - libcublas-dev
 - libcurand-dev
 - libcusolver-dev
 - libcusparse-dev
 - librmm==24.12.*,>=0.0.0a0
 - matplotlib
-- mkl-devel
 - nccl>=2.19
 - ninja
 - nlohmann_json>=3.11.2

diff --git a/dependencies.yaml b/dependencies.yaml
@@ -485,6 +485,9 @@ dependencies:
               - mkl-devel
               - libaio
               - libboost-devel
+          - matrix:
+              arch: aarch64
+            packages: null
   bench_python:
     common:
       - output_types: [conda, pyproject, requirements]