Update build-cuda.sh

Author: matthewdouglas

.github/scripts/build-cuda.sh

@@ -4,13 +4,16 @@ declare build_os
 declare cuda_version
 
 set -xeuo pipefail
-build_capability="50;52;60;61;70;75;80;86;89;90;100;120"
-remove_for_11_7=";89;90;100;120"
-remove_for_11_8=";100;120"
-remove_for_lt_12_7=";100;120"
-[[ "${cuda_version}" == 11.7.* ]] && build_capability=$(sed 's|'"$remove_for_11_7"'||g' <<< "$build_capability")
-[[ "${cuda_version}" == 11.8.* ]] && build_capability=$(sed 's|'"$remove_for_11_8"'||g' <<< "$build_capability")
-[[ "${cuda_version}" < 12.7 ]] && build_capability=$(sed 's|'"$remove_for_lt_12_7"'||g; s|'"${remove_for_lt_12_7#;}"';||g' <<< "$build_capability")
+
+# By default, target Maxwell through Hopper.
+build_capability = "50;52;60;61;70;75;80;86;89;90"
+
+# CUDA 11.7: Remove sm89 and sm90
+[[ "${cuda_version}" == 11.7.* ]] && build_capability="50;52;60;61;70;75;80;86"
+
+# CUDA 12.8: Add sm100 and sm120; remove sm50 through sm61
+[[ "${cuda_version}" == 12.8.* ]] && build_capability="70;75;80;86;89;90;100;120"
+
 [[ "${build_os}" = windows-* ]] && python3 -m pip install ninja
 
 if [ "${build_os:0:6}" == ubuntu ]; then