Unverified Commit 50be19c3 authored by Jun Jiang's avatar Jun Jiang Committed by GitHub
Browse files

Add Thor support (#1764)



* Update build-cuda.sh

---------
Co-authored-by: default avatarMatthew Douglas <38992547+matthewdouglas@users.noreply.github.com>
parent c72aae78
...@@ -11,8 +11,11 @@ if [[ -v cuda_targets ]]; then ...@@ -11,8 +11,11 @@ if [[ -v cuda_targets ]]; then
elif [ "${build_arch}" = "aarch64" ]; then elif [ "${build_arch}" = "aarch64" ]; then
build_capability="75;80;90" build_capability="75;80;90"
# CUDA 12.8+: Add sm100/sm120 # CUDA 12.8-12.9: Add sm100/sm120
[[ "${cuda_version}" == 12.8.* || "${cuda_version}" == 12.9.* || "${cuda_version}" == 13.*.* ]] && build_capability="75;80;90;100;120" [[ "${cuda_version}" == 12.8.* || "${cuda_version}" == 12.9.* ]] && build_capability="75;80;90;100;120"
# CUDA 13.0+: Add sm100/sm110/sm120
[[ "${cuda_version}" == 13.*.* ]] && build_capability="75;80;90;100;110;120"
else else
# By default, target Pascal through Hopper. # By default, target Pascal through Hopper.
build_capability="60;70;75;80;86;89;90" build_capability="60;70;75;80;86;89;90"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment