diff --git a/conda/environments/bench_ann_cuda-118_arch-x86_64.yaml b/conda/environments/bench_ann_cuda-118_arch-x86_64.yaml index 98b07c2c3f..051dad7eaa 100644 --- 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 index e3ca128340..11d14c7be9 100644 --- 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]