From d5571c690497fba31a86b44cea866be184e8925a Mon Sep 17 00:00:00 2001
From: Alexandria Barghi <abarghi@nvidia.com>
Date: Fri, 11 Oct 2024 13:19:38 -0700
Subject: [PATCH] fixes to dependencies.yaml

---
 .../all_cuda-118_arch-x86_64.yaml             |  8 ++++++-
 .../all_cuda-121_arch-x86_64.yaml             |  8 ++++++-
 .../all_cuda-124_arch-x86_64.yaml             |  8 ++++++-
 dependencies.yaml                             | 22 ++++++++++++++++---
 4 files changed, 40 insertions(+), 6 deletions(-)

diff --git a/conda/environments/all_cuda-118_arch-x86_64.yaml b/conda/environments/all_cuda-118_arch-x86_64.yaml
index 4380b82..a006610 100644
--- a/conda/environments/all_cuda-118_arch-x86_64.yaml
+++ b/conda/environments/all_cuda-118_arch-x86_64.yaml
@@ -9,20 +9,26 @@ channels:
 - nvidia
 dependencies:
 - breathe
+- c-compiler
 - cmake>=3.26.4,!=3.30.0
 - cuda-nvtx
 - cuda-version=11.8
 - cudatoolkit
 - cudf==24.12.*,>=0.0.0a0
+- cudnn=8.8
 - cugraph==24.12.*,>=0.0.0a0
 - cupy>=12.0.0
+- cxx-compiler
 - cython>=3.0.0
 - dask-cudf==24.12.*,>=0.0.0a0
 - doxygen
 - graphviz
 - ipython
+- libraft-headers==24.12.*,>=0.0.0a0
+- librmm==24.12.*,>=0.0.0a0
+- nanobind>=0.2.0
 - nbsphinx
-- ninja
+- nccl>=2.19
 - notebook>=0.5.0
 - numba>=0.57
 - numpy>=1.23,<3.0a0
diff --git a/conda/environments/all_cuda-121_arch-x86_64.yaml b/conda/environments/all_cuda-121_arch-x86_64.yaml
index a868c5c..8fd6284 100644
--- a/conda/environments/all_cuda-121_arch-x86_64.yaml
+++ b/conda/environments/all_cuda-121_arch-x86_64.yaml
@@ -9,14 +9,17 @@ channels:
 - nvidia
 dependencies:
 - breathe
+- c-compiler
 - cmake>=3.26.4,!=3.30.0
 - cuda-cudart-dev
 - cuda-nvtx-dev
 - cuda-profiler-api
 - cuda-version=12.1
 - cudf==24.12.*,>=0.0.0a0
+- cudnn=8.8
 - cugraph==24.12.*,>=0.0.0a0
 - cupy>=12.0.0
+- cxx-compiler
 - cython>=3.0.0
 - dask-cudf==24.12.*,>=0.0.0a0
 - doxygen
@@ -26,8 +29,11 @@ dependencies:
 - libcurand-dev
 - libcusolver-dev
 - libcusparse-dev
+- libraft-headers==24.12.*,>=0.0.0a0
+- librmm==24.12.*,>=0.0.0a0
+- nanobind>=0.2.0
 - nbsphinx
-- ninja
+- nccl>=2.19
 - notebook>=0.5.0
 - numba>=0.57
 - numpy>=1.23,<3.0a0
diff --git a/conda/environments/all_cuda-124_arch-x86_64.yaml b/conda/environments/all_cuda-124_arch-x86_64.yaml
index 20dcef6..32ea733 100644
--- a/conda/environments/all_cuda-124_arch-x86_64.yaml
+++ b/conda/environments/all_cuda-124_arch-x86_64.yaml
@@ -9,14 +9,17 @@ channels:
 - nvidia
 dependencies:
 - breathe
+- c-compiler
 - cmake>=3.26.4,!=3.30.0
 - cuda-cudart-dev
 - cuda-nvtx-dev
 - cuda-profiler-api
 - cuda-version=12.4
 - cudf==24.12.*,>=0.0.0a0
+- cudnn=8.8
 - cugraph==24.12.*,>=0.0.0a0
 - cupy>=12.0.0
+- cxx-compiler
 - cython>=3.0.0
 - dask-cudf==24.12.*,>=0.0.0a0
 - doxygen
@@ -26,8 +29,11 @@ dependencies:
 - libcurand-dev
 - libcusolver-dev
 - libcusparse-dev
+- libraft-headers==24.12.*,>=0.0.0a0
+- librmm==24.12.*,>=0.0.0a0
+- nanobind>=0.2.0
 - nbsphinx
-- ninja
+- nccl>=2.19
 - notebook>=0.5.0
 - numba>=0.57
 - numpy>=1.23,<3.0a0
diff --git a/dependencies.yaml b/dependencies.yaml
index 29ceca6..f349ec8 100644
--- a/dependencies.yaml
+++ b/dependencies.yaml
@@ -204,10 +204,20 @@ dependencies:
               - cuda-nvtx
   common_build:
     common:
-      - output_types: [conda, pyproject]
+      - output_types: [requirements, pyproject]
         packages:
-          - &cmake_ver cmake>=3.26.4,!=3.30.0
           - ninja
+      - output_types: [conda]
+        packages:
+          - &cmake_ver cmake>=3.26.4,!=3.30.0
+          - cudnn=8.8
+          - c-compiler
+          - cxx-compiler
+          - libraft-headers==24.12.*,>=0.0.0a0
+          - librmm==24.12.*,>=0.0.0a0
+          - nanobind>=0.2.0
+          - &cython cython>=3.0.0
+          - &nccl nccl>=2.19
 
   docs:
     common:
@@ -254,7 +264,7 @@ dependencies:
     common:
       - output_types: [conda, pyproject, requirements]
         packages:
-          - cython>=3.0.0
+          - *cython
       - output_types: conda
         packages:
           - scikit-build-core>=0.10.0
@@ -288,6 +298,12 @@ dependencies:
       - output_types: [conda]
         packages:
           - wget
+  test_cpp:
+    common:
+      - output_types: conda
+        packages:
+          - *cmake_ver
+          - *nccl
   test_python_common:
     common:
       - output_types: [conda, pyproject]