Add dependency on CUDA.

rapidsai · Apr 1, 2024 · a31833b · a31833b
1 parent fda0aab
commit a31833b
Showing 1 changed file with 41 additions and 2 deletions.
diff --git a/dependencies.yaml b/dependencies.yaml
@@ -4,21 +4,60 @@ files:
     output: none
     includes:
       - build
+      - cuda_version
+      - cuda
       - docs
 channels:
   - rapidsai
   - conda-forge
 dependencies:
   build:
     common:
-      - output_types: [conda]
+      - output_types: conda
         packages:
           - binutils
           - cmake
           - gxx <=11
           - make
+  cuda_version:
+    specific:
+      - output_types: conda
+        matrices:
+          - matrix:
+              cuda: "11.8"
+            packages:
+              - cuda-version=11.8
+          - matrix:
+              cuda: "12.0"
+            packages:
+              - cuda-version=12.0
+          - matrix:
+              cuda: "12.2"
+            packages:
+              - cuda-version=12.2
+  cuda:
+    specific:
+      - output_types: conda
+        matrices:
+          - matrix:
+              cuda: "12.*"
+            packages:
+              - cuda-nvcc
+              - cuda-cudart-dev
+          - matrix:
+              arch: x86_64
+              cuda: "11.8"
+            packages:
+              - nvcc_linux-64=11.8
+              - cudatoolkit
+          - matrix:
+              arch: aarch64
+              cuda: "11.8"
+            packages:
+              - nvcc_linux-aarch64=11.8
+              - cudatoolkit
   docs:
     common:
-      - output_types: [conda]
+      - output_types: conda
         packages:
           - doxygen