Skip to content

Commit 50be19c

Browse files
jaslmatthewdouglas
andauthored
Add Thor support (#1764)
* Update build-cuda.sh --------- Co-authored-by: Matthew Douglas <[email protected]>
1 parent c72aae7 commit 50be19c

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

.github/scripts/build-cuda.sh

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,8 +11,11 @@ if [[ -v cuda_targets ]]; then
1111
elif [ "${build_arch}" = "aarch64" ]; then
1212
build_capability="75;80;90"
1313

14-
# 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"
14+
# CUDA 12.8-12.9: Add sm100/sm120
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)