add more cuda deps in testing

conda-forge · Jan 31, 2025 · 510d074 · 510d074
1 parent 8591bb6
commit 510d074
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/recipe/meta.yaml b/recipe/meta.yaml
@@ -211,7 +211,9 @@ test:
     # cmake needs a compiler to run package detection, see
     # https://discourse.cmake.org/t/questions-about-find-package-cli-msvc/6194
     - {{ compiler('cxx') }}
+    # for CMake config to find cuda & nvrtc
     - {{ compiler('cuda') }}    # [cuda_compiler_version != "None"]
+    - cuda-nvrtc-dev            # [cuda_compiler_version != "None"]
     - cmake
     - ninja
     - pkg-config
@@ -368,6 +370,8 @@ outputs:
       requires:
         - {{ compiler('c') }}
         - {{ compiler('cxx') }}
+        # for torch.compile tests
+        - {{ compiler('cuda') }}       # [cuda_compiler_version != "None"]
         - ninja
         - boto3
         - hypothesis