Skip to content

Commit 509ed8f

Browse files
Drop sm70 from cu128 build targets to align with pytorch
1 parent 722dba7 commit 509ed8f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

.github/scripts/build-cuda.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,8 @@ set -xeuo pipefail
88
# By default, target Maxwell through Hopper.
99
build_capability="50;52;60;61;70;75;80;86;89;90"
1010

11-
# CUDA 12.8: Add sm100 and sm120; remove sm50 through sm61
12-
[[ "${cuda_version}" == 12.8.* ]] && build_capability="70;75;80;86;89;90;100;120"
11+
# CUDA 12.8: Add sm100 and sm120; remove < sm75 to align with PyTorch 2.7+cu128 minimum
12+
[[ "${cuda_version}" == 12.8.* ]] && build_capability="75;80;86;89;90;100;120"
1313

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

0 commit comments

Comments
 (0)