Conditional on variable declaration.

rapidsai · Jan 11, 2024 · 1ad4439 · 1ad4439
1 parent 5a27d41
commit 1ad4439
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/conda/recipes/pynvjitlink/meta.yaml b/conda/recipes/pynvjitlink/meta.yaml
@@ -2,8 +2,13 @@
 
 {% set data = load_file_data("pyproject.toml") %}
 
+{% if cuda_compiler_version is defined %}
 {% set cuda_major = cuda_compiler_version.split(".")[0]|int %}
 {% set cuda_minor = cuda_compiler_version.split(".")[1]|int %}
+{% else %}
+{% set cuda_major = 0 %}
+{% set cuda_minor = 0 %}
+{% endif %}
 
 package:
   name: pynvjitlink