Skip to content

Commit 52c2f25

Browse files
authored
Fix override_gencode population (#675)
1 parent 8ffd0f9 commit 52c2f25

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

manywheel/build.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,7 @@ if [[ -n "$OVERRIDE_TORCH_CUDA_ARCH_LIST" ]]; then
8383

8484
# Prune CUDA again with new arch list. Unfortunately, we need to re-install CUDA to prune it again
8585
override_gencode=""
86-
for arch in ${torch_cuda_arch_list//;/ } ; do
86+
for arch in ${TORCH_CUDA_ARCH_LIST//;/ } ; do
8787
arch_code=$(echo "$arch" | tr -d .)
8888
override_gencode="${override_gencode}-gencode arch=compute_$arch_code,code=sm_$arch_code "
8989
done

0 commit comments

Comments
 (0)