From daffaf475ff17f561d6d64ed2e4cd2f204686c6a Mon Sep 17 00:00:00 2001 From: "Corey J. Nolet" Date: Fri, 6 Oct 2023 22:04:27 -0400 Subject: [PATCH] Adding openblas to nn_bench deps --- conda/environments/bench_ann_cuda-118_arch-x86_64.yaml | 1 + dependencies.yaml | 1 + 2 files changed, 2 insertions(+) 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]