Adding openblas to nn_bench deps

rapidsai · Oct 7, 2023 · daffaf4 · daffaf4
1 parent 667b95c
commit daffaf4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/conda/environments/bench_ann_cuda-118_arch-x86_64.yaml b/conda/environments/bench_ann_cuda-118_arch-x86_64.yaml
@@ -33,6 +33,7 @@ dependencies:
 - ninja
 - nlohmann_json>=3.11.2
 - nvcc_linux-64=11.8
+- openblas
 - rmm=23.12.*
 - scikit-build>=0.13.1
 - sysroot_linux-64==2.17

diff --git a/dependencies.yaml b/dependencies.yaml
@@ -227,6 +227,7 @@ dependencies:
           - h5py>=3.8.0
           - benchmark>=1.8.2
           - rmm=23.12.*
+          - openblas
   nn_bench_python:
     common:
       - output_types: [conda]