Skip to content

Commit d65d6e1

Browse files
authored
Update build-cuda.sh
1 parent b378e24 commit d65d6e1

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

.github/scripts/build-cuda.sh

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,9 +5,12 @@ declare cuda_version
55

66
set -xeuo pipefail
77
build_capability="50;52;60;61;70;75;80;86;89;90;100;120"
8-
[[ "${cuda_version}" == 11.7.* ]] && build_capability=${build_capability%???????????????????????}
9-
[[ "${cuda_version}" == 11.8.* ]] && build_capability=${build_capability%??????????????????}
10-
[[ "${cuda_version}" < 12.7 ]] && build_capability=${build_capability%??????????????????}
8+
remove_for_11_7=";86;89;90;100;120"
9+
remove_for_11_8=";89;90;100;120"
10+
remove_for_lt_12_7=";100;120"
11+
[[ "${cuda_version}" == 11.7.* ]] && build_capability=$(sed 's|'"$remove_for_11_7"'||g' <<< "$build_capability")
12+
[[ "${cuda_version}" == 11.8.* ]] && build_capability=$(sed 's|'"$remove_for_11_8"'||g' <<< "$build_capability")
13+
[[ "${cuda_version}" < 12.7 ]] && build_capability=$(sed 's|'"$remove_for_lt_12_7"'||g; s|'"${remove_for_lt_12_7#;}"';||g' <<< "$build_capability")
1114
[[ "${build_os}" = windows-* ]] && python3 -m pip install ninja
1215

1316
if [ "${build_os:0:6}" == ubuntu ]; then

0 commit comments

Comments
 (0)