[cmake] [CUDA] ignore CUDA-specific source files in non-CUDA builds (fixes #6267) (#6268)

2e3543c0 · Samuel Johnson · GitHub · 5516533c · 2e3543c0
Unverified Commit 2e3543c0 authored Jan 11, 2024 by Samuel Johnson Committed by GitHub Jan 11, 2024
Show whitespace changes
Inline Side-by-side

Showing with 8 additions and 2 deletions

CMakeLists.txt CMakeLists.txt +8 -2

No files found.
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -428,7 +428,10 @@ file(
      src/network/*.cpp
      src/treelearner/*.cpp
      src/utils/*.cpp
-if(USE_CUDA)
+)
+file(
+    GLOB
+    LGBM_CUDA_SOURCES
      src/treelearner/*.cu
      src/boosting/cuda/*.cpp
      src/boosting/cuda/*.cu
@@ -442,9 +445,12 @@ if(USE_CUDA)
      src/io/cuda/*.cpp
      src/cuda/*.cpp
      src/cuda/*.cu
-endif()
 )

+if(USE_CUDA)
+  list(APPEND SOURCES ${LGBM_CUDA_SOURCES})
+endif()
+
 add_library(lightgbm_objs OBJECT ${SOURCES})

 if(BUILD_CLI)