Add config for USE_CUDA.

Author: trivialfis

CMakeLists.txt

@@ -59,6 +59,13 @@ ADD_LIBRARY(cpuh2o4gpu STATIC ${CPU_SOURCES} $<TARGET_OBJECTS:commonh2o4gpu>)
 TARGET_LINK_LIBRARIES(cpuh2o4gpu ${BLAS_LIBRARIES})
 #============= BUILD CPU LIBRARY
 
+if (USE_CUDA)
+  set(HG_USE_CUDA 1)
+else()
+  set(HG_USE_CUDA 0)
+endif(USE_CUDA)
+add_subdirectory(${CMAKE_CURRENT_LIST_DIR}/src/common)
+
 if(USE_CUDA)
   FIND_PACKAGE(CUDA 8.0 REQUIRED)
   FIND_PACKAGE(NVML REQUIRED)

src/common/CMakeLists.txt

@@ -0,0 +1 @@
+configure_file(${CMAKE_CURRENT_SOURCE_DIR}/utils.h.in ${CMAKE_CURRENT_SOURCE_DIR}/utils.h)

src/common/utils.h renamed to src/common/utils.h.in

@@ -10,7 +10,7 @@
 
 #include "cblas/cblas.h"
 
-#define USE_CUDA() 1
+#define USE_CUDA() @HG_USE_CUDA@
 
 template<typename T>
 void self_dot(std::vector<T> array_in, int n, int dim,