diff --git a/.github/scripts/build-cuda.sh b/.github/scripts/build-cuda.sh index 630a79c13..6c618b0cc 100644 --- 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 index 64d01e91d..693a2655d 100644 --- 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 index c9eefb751..d16e94690 100644 --- 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()