Skip to content

[CUDA] fix setting of CUDA architectures and enable support for NVIDIA Blackwell #6812

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 18 commits into from
Feb 2, 2025
Merged

[CUDA] fix setting of CUDA architectures and enable support for NVIDIA Blackwell #6812

Changes from 1 commit
Commits
Show all changes
18 commits
Select commit Hold shift + click to select a range
2df25bd
Update CMakeLists.txt
StrikerRUS Jan 31, 2025
f569fac
Update cuda.yml
StrikerRUS Jan 31, 2025
6918b18
Update cuda.yml
StrikerRUS Feb 2, 2025
4218424
Update CMakeLists.txt
StrikerRUS Feb 2, 2025
cca1a03
Update cuda.yml
StrikerRUS Feb 2, 2025
8648091
Update CMakeLists.txt
StrikerRUS Feb 2, 2025
f44d835
Update CMakeLists.txt
StrikerRUS Feb 2, 2025
2fc0a4f
Update CMakeLists.txt
StrikerRUS Feb 2, 2025
c32e639
Update CMakeLists.txt
StrikerRUS Feb 2, 2025
9a9c3cb
Update CMakeLists.txt
StrikerRUS Feb 2, 2025
e6751d9
Update CMakeLists.txt
StrikerRUS Feb 2, 2025
d9b8c26
Update CMakeLists.txt
StrikerRUS Feb 2, 2025
b1a9d31
Update CMakeLists.txt
StrikerRUS Feb 2, 2025
903d9e0
Update CMakeLists.txt
StrikerRUS Feb 2, 2025
a026971
Update cuda.yml
StrikerRUS Feb 2, 2025
e9bf0c6
Update CMakeLists.txt
StrikerRUS Feb 2, 2025
cc09dc0
Update CMakeLists.txt
StrikerRUS Feb 2, 2025
be70aa5
Update check-python-dists.sh
StrikerRUS Feb 2, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Update CMakeLists.txt
  • Loading branch information
@StrikerRUS
StrikerRUS authored Feb 2, 2025
commit 421842431d1c0f0a88d612dffba772ef5138bb12
3 changes: 3 additions & 0 deletions CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -224,6 +224,8 @@ if(USE_CUDA)
# reference for mapping of CUDA toolkit component versions to supported architectures ("compute capabilities"):
# https://en.wikipedia.org/wiki/CUDA#GPUs_supported
set(CUDA_ARCHS "60" "61" "62" "70" "75")
message(STATUS "---------------")
message(STATUS ${CUDA_VERSION})
if(CUDA_VERSION VERSION_GREATER_EQUAL "110")
Copy link
Collaborator Author

@StrikerRUS StrikerRUS Feb 2, 2025 •
edited
Loading

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

"110" means exactly 110 version during comparison, VERSION_GREATER_EQUAL doesn't know whether and where we want to put a .: 11.0 or maybe 1.10.

list(APPEND CUDA_ARCHS "80")
endif()
Expand All @@ -238,6 +240,7 @@ if(USE_CUDA)
list(APPEND CUDA_ARCHS "90")
endif()
if(CUDA_VERSION VERSION_GREATER_EQUAL "128")
message(STATUS "Using CUDA 128")
list(APPEND CUDA_ARCHS "100")
list(APPEND CUDA_ARCHS "120")
endif()
Expand Down