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"