Add config for USE_CUDA.

h2oai · Jul 25, 2018 · c61e8da · c61e8da
1 parent 6c8ff4e
commit c61e8da
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 1 deletion.
diff --git a/CMakeLists.txt b/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)

diff --git a/src/common/CMakeLists.txt b/src/common/CMakeLists.txt
@@ -0,0 +1 @@
+configure_file(${CMAKE_CURRENT_SOURCE_DIR}/utils.h.in ${CMAKE_CURRENT_SOURCE_DIR}/utils.h)
diff --git a/src/common/utils.h → src/common/utils.h.in b/src/common/utils.h → 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,
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		configure_file(${CMAKE_CURRENT_SOURCE_DIR}/utils.h.in ${CMAKE_CURRENT_SOURCE_DIR}/utils.h)