Trying script_env

facebookresearch · May 20, 2024 · 9e26015 · 9e26015
1 parent c8d836e
commit 9e26015
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/conda/faiss-gpu/meta.yaml b/conda/faiss-gpu/meta.yaml
@@ -41,12 +41,11 @@ outputs:
       string: "h{{ PKG_HASH }}_{{ number }}_cuda{{ cudatoolkit }}{{ suffix }}"
       run_exports:
         - {{ pin_compatible('libfaiss', exact=True) }}
-      env:
-        {% if cudatoolkit == '11.4.4' %}
-        - FAISS_FLATTEN_CONDA_INCLUDES: "1"
-        {% endif %}
       script_env:
         - CUDA_ARCHS
+        {% if cudatoolkit == '11.4.4' %}
+        - FAISS_FLATTEN_CONDA_INCLUDES = "1"
+        {% endif %}
     requirements:
       build:
         - {{ compiler('cxx') }}