[cmake] [CUDA] ignore CUDA-specific source files in non-CUDA builds (f…

…ixes #6267) (#6268)
microsoft · Jan 12, 2024 · 2e3543c · 2e3543c
1 parent 5516533
commit 2e3543c
Showing 1 changed file with 8 additions and 2 deletions.
diff --git 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)