|
|
|
@ -95,11 +95,26 @@ if(NOT WITH_GPU)
|
|
|
|
|
add_definitions(-DHPPL_STUB_FUNC)
|
|
|
|
|
list(APPEND CMAKE_CXX_SOURCE_FILE_EXTENSIONS cu)
|
|
|
|
|
else()
|
|
|
|
|
if(${CUDA_VERSION_MAJOR} GREATER 6)
|
|
|
|
|
if(COMPILER_SUPPORT_CXX11)
|
|
|
|
|
LIST(APPEND CUDA_NVCC_FLAGS -std=c++11)
|
|
|
|
|
endif()
|
|
|
|
|
endif()
|
|
|
|
|
|
|
|
|
|
# TODO(yuyang18): Change it to remove std=c++11 in cuda compile.
|
|
|
|
|
set(CUDA_PROPAGATE_HOST_FLAGS OFF)
|
|
|
|
|
if(NOT CUDNN_FOUND)
|
|
|
|
|
message(FATAL_ERROR "Paddle need cudnn to compile")
|
|
|
|
|
endif()
|
|
|
|
|
set(CUDA_NVCC_FLAGS ${CUDA_NVCC_FLAGS} "-g -O3 --use_fast_math")
|
|
|
|
|
|
|
|
|
|
if(WITH_AVX)
|
|
|
|
|
if(AVX_FOUND)
|
|
|
|
|
set(CUDA_NVCC_FLAGS ${CUDA_NVCC_FLAGS} "-Xcompiler -mavx")
|
|
|
|
|
endif(AVX_FOUND)
|
|
|
|
|
else(WITH_AVX)
|
|
|
|
|
set(CUDA_NVCC_FLAGS ${CUDA_NVCC_FLAGS} "-Xcompiler -msse3")
|
|
|
|
|
endif(WITH_AVX)
|
|
|
|
|
|
|
|
|
|
if(WITH_DSO)
|
|
|
|
|
set(CUDA_LIBRARIES "")
|
|
|
|
|