Skip to content

Commit 18c07bd

Browse files
fix
1 parent 77db267 commit 18c07bd

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

.github/scripts/build-cuda.sh

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,14 +6,14 @@ declare cuda_targets
66

77
set -xeuo pipefail
88

9-
if [ -n "${cuda_targets}" ]; then
9+
if [[ -v cuda_targets ]; then
10+
build_capability="${cuda_targets}"
11+
else
1012
# By default, target Maxwell through Hopper.
1113
build_capability="50;52;60;61;70;75;80;86;89;90"
1214

1315
# CUDA 12.8: Add sm100 and sm120; remove < sm75 to align with PyTorch 2.7+cu128 minimum
1416
[[ "${cuda_version}" == 12.8.* ]] && build_capability="75;80;86;89;90;100;120"
15-
else
16-
build_capability="${cuda_targets}"
1717
fi
1818

1919
[[ "${build_os}" = windows-* ]] && python3 -m pip install ninja

0 commit comments

Comments
 (0)