Skip to content

Commit 09f58ce

Browse files
authored
Update build-cuda.sh
1 parent 9e4f73e commit 09f58ce

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

.github/scripts/build-cuda.sh

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,10 @@ elif [ "${build_arch}" = "aarch64" ]; then
1212
build_capability="75;80;90"
1313

1414
# CUDA 12.8+: Add sm100/sm120
15-
[[ "${cuda_version}" == 12.8.* || "${cuda_version}" == 12.9.* || "${cuda_version}" == 13.*.* ]] && build_capability="75;80;90;100;120"
15+
[[ "${cuda_version}" == 12.8.* || "${cuda_version}" == 12.9.* ]] && build_capability="75;80;90;100;120"
16+
17+
# CUDA 13.0+: Add sm100/sm110/sm120
18+
[[ "${cuda_version}" == 13.*.* ]] && build_capability="75;80;90;100;110;120"
1619
else
1720
# By default, target Pascal through Hopper.
1821
build_capability="60;70;75;80;86;89;90"

0 commit comments

Comments
 (0)