diff --git a/.github/scripts/build-cuda.sh b/.github/scripts/build-cuda.sh index 81fd410ac..32874e70b 100644 --- a/.github/scripts/build-cuda.sh +++ b/.github/scripts/build-cuda.sh @@ -11,8 +11,11 @@ if [[ -v cuda_targets ]]; then elif [ "${build_arch}" = "aarch64" ]; then build_capability="75;80;90" - # CUDA 12.8+: Add sm100/sm120 - [[ "${cuda_version}" == 12.8.* || "${cuda_version}" == 12.9.* || "${cuda_version}" == 13.*.* ]] && build_capability="75;80;90;100;120" + # CUDA 12.8-12.9: Add sm100/sm120 + [[ "${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 # By default, target Pascal through Hopper. build_capability="60;70;75;80;86;89;90"