fix cuda 12.7 build-cuda.sh

bitsandbytes-foundation · Jan 22, 2025 · b378e24 · b378e24
1 parent b803bdf
commit b378e24
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/.github/scripts/build-cuda.sh b/.github/scripts/build-cuda.sh
@@ -5,8 +5,9 @@ declare cuda_version
 
 set -xeuo pipefail
 build_capability="50;52;60;61;70;75;80;86;89;90;100;120"
-[[ "${cuda_version}" == 11.7.* ]] && build_capability=${build_capability%??????????????????}
-[[ "${cuda_version}" == 11.8.* ]] && build_capability=${build_capability%?????????}
+[[ "${cuda_version}" == 11.7.* ]] && build_capability=${build_capability%???????????????????????}
+[[ "${cuda_version}" == 11.8.* ]] && build_capability=${build_capability%??????????????????}
+[[ "${cuda_version}" < 12.7 ]] && build_capability=${build_capability%??????????????????}
 [[ "${build_os}" = windows-* ]] && python3 -m pip install ninja
 
 if [ "${build_os:0:6}" == ubuntu ]; then