From 85465402f92c0ce58c1d67af8b62b6836bc0b27e Mon Sep 17 00:00:00 2001 From: Christopher Harris Date: Mon, 7 Oct 2024 19:13:36 +0000 Subject: [PATCH] add matx cuda dependencies --- conda/environments/all_cuda-125_arch-x86_64.yaml | 3 +++ conda/environments/dev_cuda-125_arch-x86_64.yaml | 3 +++ dependencies.yaml | 6 ++++-- 3 files changed, 10 insertions(+), 2 deletions(-) diff --git a/conda/environments/all_cuda-125_arch-x86_64.yaml b/conda/environments/all_cuda-125_arch-x86_64.yaml index 1a45c575c6..0281cf6b68 100644 --- a/conda/environments/all_cuda-125_arch-x86_64.yaml +++ b/conda/environments/all_cuda-125_arch-x86_64.yaml @@ -57,8 +57,11 @@ dependencies: - isort - jsonpatch>=1.33 - kfp +- libcublas-dev - libcudf=24.10 - libcufft-dev +- libcurand-dev +- libcusolver-dev - librdkafka>=1.9.2,<1.10.0a0 - libtool - libwebp=1.3.2 diff --git a/conda/environments/dev_cuda-125_arch-x86_64.yaml b/conda/environments/dev_cuda-125_arch-x86_64.yaml index 47e717f3f0..af599fb7de 100644 --- a/conda/environments/dev_cuda-125_arch-x86_64.yaml +++ b/conda/environments/dev_cuda-125_arch-x86_64.yaml @@ -47,8 +47,11 @@ dependencies: - include-what-you-use=0.20 - ipython - isort +- libcublas-dev - libcudf=24.10 - libcufft-dev +- libcurand-dev +- libcusolver-dev - librdkafka>=1.9.2,<1.10.0a0 - libtool - libwebp=1.3.2 diff --git a/dependencies.yaml b/dependencies.yaml index 7242c9ea5f..01546f6bae 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -214,10 +214,12 @@ dependencies: - cuda-nvml-dev=12.5 - cuda-nvrtc-dev=12.5 - cuda-nvtx-dev=12.5 - # - cuda-tools=12.5 + - cuda-sanitizer-api - cuda-version=12.5 + - libcublas-dev # required by matx - libcufft-dev # required by matx - - cuda-sanitizer-api + - libcusolver-dev # required by matx + - libcurand-dev # required by matx