-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathCMakeLists.txt
23 lines (15 loc) · 1017 Bytes
/
CMakeLists.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
PROJECT(currennt-pp)
CMAKE_MINIMUM_REQUIRED (VERSION 2.8)
FIND_PACKAGE (CUDA)
FIND_PACKAGE (Boost 1.48.0 REQUIRED COMPONENTS program_options system filesystem random thread REQUIRED)
INCLUDE_DIRECTORIES (${Boost_INCLUDE_DIR})
LINK_DIRECTORIES (${Boost_LIBRARY_DIR})
SET (CMAKE_BUILD_TYPE Release)
SET (CUDA_NVCC_FLAGS "${CUDA_NVCC_FLAGS} -std=c++11 --generate-code arch=compute_50,code=sm_50 --generate-code arch=compute_61,code=sm_61 -O3")
SET (CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11 -lpthread -Wno-format-security")
FILE (GLOB_RECURSE src_lib currennt_lib/*.cpp currennt_lib/*.hpp currennt_lib/*.h currennt_lib/*.cu currennt_lib/*.cuh)
FILE (GLOB_RECURSE src_trainer currennt/*.cpp currennt/*.hpp currennt/*.h currennt/*.cu currennt/*.cuh)
CUDA_ADD_EXECUTABLE (${PROJECT_NAME} ${src_lib} ${src_trainer})
CUDA_ADD_CUBLAS_TO_TARGET (${PROJECT_NAME})
TARGET_LINK_LIBRARIES (${PROJECT_NAME} netcdf)
TARGET_LINK_LIBRARIES (${PROJECT_NAME} ${Boost_LIBRARIES})