Commit e4a9a94c authored by Matthew Douglas's avatar Matthew Douglas
Browse files

Update build-cuda.sh

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