Skip to content

GitLab

"include/vscode:/vscode.git/clone" did not exist on "d1d218c329b57e97b8938adfb1643e210ebe197c"
Unverified Commit d6fc12b3 authored by Robin Dong's avatar Robin Dong Committed by GitHub
Browse files

[CUDA] Add CUDA_ARCHITECTURES to fix CMake warnings (#3754) (#4268)

parent 90677e43
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
CMakeLists.txt
View file @ d6fc12b3
......@@ -210,6 +210,7 @@ if(USE_CUDA)
function(add_histogram hsize hname hadd hconst hdir)
add_library(histo${hsize}${hname} OBJECT src/treelearner/kernels/histogram${hsize}.cu)
set_target_properties(histo${hsize}${hname} PROPERTIES CUDA_SEPARABLE_COMPILATION ON)
set_target_properties(histo${hsize}${hname} PROPERTIES CUDA_ARCHITECTURES OFF)
if(hadd)
list(APPEND histograms histo${hsize}${hname})
set(histograms ${histograms} PARENT_SCOPE)
......@@ -434,11 +435,13 @@ endif()
if(USE_CUDA)
set_target_properties(lightgbm PROPERTIES CUDA_RESOLVE_DEVICE_SYMBOLS ON)
set_target_properties(lightgbm PROPERTIES CUDA_ARCHITECTURES OFF)
TARGET_LINK_LIBRARIES(
lightgbm
${histograms}
)
set_target_properties(_lightgbm PROPERTIES CUDA_RESOLVE_DEVICE_SYMBOLS ON)
set_target_properties(_lightgbm PROPERTIES CUDA_ARCHITECTURES OFF)
TARGET_LINK_LIBRARIES(
_lightgbm
${histograms}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment