messed up the logic, this should not match the OG recipe

regro-cf-autotick-bot · Jun 8, 2022 · 38cf28c · 38cf28c
1 parent 24a3f70
commit 38cf28c
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/recipe/build.sh b/recipe/build.sh
@@ -2,8 +2,10 @@
 
 set -euxo pipefail
 
-if [[ ${cuda_compiler_version} != "None" ]]; then
+if [[ ${cuda_compiler_version} != "None" && "$target_platform" == linux-64 ]]; then
     export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX"
+    export FORCE_CUDA="1"
+    export CC="$GCC"
     if [[ ${cuda_compiler_version} == 9.0* ]]; then
         export TORCH_CUDA_ARCH_LIST="$TORCH_CUDA_ARCH_LIST;6.0;7.0"
     elif [[ ${cuda_compiler_version} == 9.2* ]]; then
@@ -22,11 +24,6 @@ if [[ ${cuda_compiler_version} != "None" ]]; then
     fi
 fi
 
-if [[ ${cuda_compiler_version} == "None"  && "$target_platform" == linux-64 ]]; then
-    export FORCE_CUDA="1"
-    export CC="$GCC"
-fi
-
 
 echo "Installing"
 ${PYTHON} -m pip install . -vv