From 74ac097f8d59f5d9be2ab4adf3b5c22ef5ce0e46 Mon Sep 17 00:00:00 2001
From: Eli Uriegas <eliuriegas@fb.com>
Date: Wed, 9 Dec 2020 16:21:38 -0800
Subject: [PATCH] manywheel: Fix TORCH_CUDA_ARCH_LIST not getting populated

There was a variable mismatch when it came to TORCH_CUDA_ARCH_LIST

Signed-off-by: Eli Uriegas <eliuriegas@fb.com>
---
 manywheel/build.sh | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

diff --git a/manywheel/build.sh b/manywheel/build.sh
index 24a905ce9..92da0e3b8 100755
--- a/manywheel/build.sh
+++ b/manywheel/build.sh
@@ -50,22 +50,22 @@ fi
 
 cuda_version_nodot=$(echo $CUDA_VERSION | tr -d '.')
 
-torch_cuda_arch_list="3.7;5.0;6.0;7.0"
+TORCH_CUDA_ARCH_LIST="3.7;5.0;6.0;7.0"
 case ${CUDA_VERSION} in
     11.1)
-        torch_cuda_arch_list="5.0;7.0;8.0;8.6"  # removing some to prevent bloated binary size
+        TORCH_CUDA_ARCH_LIST="5.0;7.0;8.0;8.6"  # removing some to prevent bloated binary size
         EXTRA_CAFFE2_CMAKE_FLAGS+=("-DATEN_NO_TEST=ON")
         ;;
     11.0)
-        torch_cuda_arch_list="$TORCH_CUDA_ARCH_LIST;7.5;8.0"
+        TORCH_CUDA_ARCH_LIST="${TORCH_CUDA_ARCH_LIST};7.5;8.0"
         EXTRA_CAFFE2_CMAKE_FLAGS+=("-DATEN_NO_TEST=ON")
         ;;
     10.*)
-        torch_cuda_arch_list="$TORCH_CUDA_ARCH_LIST;7.5"
+        TORCH_CUDA_ARCH_LIST="${TORCH_CUDA_ARCH_LIST};7.5"
         EXTRA_CAFFE2_CMAKE_FLAGS+=("-DATEN_NO_TEST=ON")
         ;;
     9.*)
-        torch_cuda_arch_list="${TORCH_CUDA_ARCH_LIST}"
+        TORCH_CUDA_ARCH_LIST="${TORCH_CUDA_ARCH_LIST}"
         ;;
     *)
         echo "unknown cuda version $CUDA_VERSION"
@@ -74,7 +74,7 @@ case ${CUDA_VERSION} in
 esac
 
 if [[ -n "$OVERRIDE_TORCH_CUDA_ARCH_LIST" ]]; then
-    torch_cuda_arch_list="$OVERRIDE_TORCH_CUDA_ARCH_LIST"
+    TORCH_CUDA_ARCH_LIST="$OVERRIDE_TORCH_CUDA_ARCH_LIST"
 
     # Prune CUDA again with new arch list. Unfortunately, we need to re-install CUDA to prune it again
     override_gencode=""
@@ -87,8 +87,8 @@ if [[ -n "$OVERRIDE_TORCH_CUDA_ARCH_LIST" ]]; then
     bash "$(dirname "$SCRIPTPATH")"/common/install_cuda.sh "${CUDA_VERSION}"
 fi
 
-export TORCH_CUDA_ARCH_LIST=$torch_cuda_arch_list
-echo "$TORCH_CUDA_ARCH_LIST"
+export TORCH_CUDA_ARCH_LIST=${TORCH_CUDA_ARCH_LIST}
+echo "${TORCH_CUDA_ARCH_LIST}"
 
 # Package directories
 WHEELHOUSE_DIR="wheelhouse$cuda_version_nodot"