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 cc09dc0d81e90ee67b9dec2a2ad6c55d6be6ad51
3 changes: 2 additions & 1 deletion CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -245,6 +245,7 @@ if(USE_CUDA)
list(POP_BACK CUDA_ARCHS CUDA_LAST_SUPPORTED_ARCH)
list(TRANSFORM CUDA_ARCHS APPEND "-real")
list(APPEND CUDA_ARCHS "${CUDA_LAST_SUPPORTED_ARCH}-real" "${CUDA_LAST_SUPPORTED_ARCH}-virtual")
message(STATUS "CUDA_ARCHITECTURES: ${CUDA_ARCHS}")
if(USE_DEBUG)
set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -g")
else()
Expand Down Expand Up @@ -569,7 +570,7 @@ if(__INTEGRATE_OPENCL)
endif()

if(USE_CUDA)
message(STATUS "${CUDA_ARCHS}")

set_target_properties(
lightgbm_objs
PROPERTIES
Expand Down
Loading