Add -D PADDLE_WITH_CUDA in cmake/configure.cmake

8 years ago · 5f51d0afc4
parent 77e07833f5
commit 5f51d0afc4
1 changed files with 5 additions and 0 deletions
--- a/cmake/configure.cmake
+++ b/cmake/configure.cmake
@ -49,11 +49,16 @@ if(NOT WITH_GOLANG)
 endif(NOT WITH_GOLANG)

 if(NOT WITH_GPU)
+    # Will gradually remove uses of PADDLE_ONLY_CPU in source files,
+    # so could we remove -DPADDLE_ONLY_CPU.
+    # c.f. https://github.com/PaddlePaddle/Paddle/issues/4588
    add_definitions(-DPADDLE_ONLY_CPU)
    add_definitions(-DHPPL_STUB_FUNC)

    list(APPEND CMAKE_CXX_SOURCE_FILE_EXTENSIONS cu)
 else()
+    add_definitions(-DPADDLE_WITH_CUDA)
+
    FIND_PACKAGE(CUDA REQUIRED)

    if(${CUDA_VERSION_MAJOR} VERSION_LESS 7)