Merge remote-tracking branch 'origin/main'

bitsandbytes-foundation · Jan 22, 2025 · 2d1fe52 · 2d1fe52
2 parents 1bc0fd8 + d65d6e1
commit 2d1fe52
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 6 deletions.
diff --git a/.github/scripts/build-cuda.sh b/.github/scripts/build-cuda.sh
@@ -5,8 +5,12 @@ 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%???}
+remove_for_11_7=";86;89;90;100;120"
+remove_for_11_8=";89;90;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")
 [[ "${build_os}" = windows-* ]] && python3 -m pip install ninja
 
 if [ "${build_os:0:6}" == ubuntu ]; then

diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,8 +1,8 @@
-### 0.43.4
+### 0.45.1
 
 #### Improvements:
 
-- Initial Support Blackwell B100 GPUs and RTX 50 series GPUs
+- Initial Support Blackwell B100 GPUs, RTX 50 Blackwell series GPUs and Jetson Thor Blackwell
 
 ### 0.43.3
 

diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -130,8 +130,8 @@ if(BUILD_CUDA)
 
         # CUDA 12.8 adds support for RTX 50 Blackwell.
         if (CMAKE_CUDA_COMPILER_VERSION VERSION_GREATER_EQUAL "12.8")
-            list(APPEND CMAKE_CUDA_ARCHITECTURES_ALL 120)
-            list(APPEND CMAKE_CUDA_ARCHITECTURES_ALL_MAJOR 120)
+            list(APPEND CMAKE_CUDA_ARCHITECTURES_ALL 101 120)
+            list(APPEND CMAKE_CUDA_ARCHITECTURES_ALL_MAJOR 101 120)
         endif()
     endif()