summaryrefslogtreecommitdiff
path: root/infra/nncc/cmake/packages/Caffe/CMakeLists.txt
blob: 51b723eddde4d26ab3ac5fc3b5f77501fa0d6f8d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
file(GLOB CORE_SOURCES "${CaffeSource_DIR}/src/caffe/*.cpp" "${CaffeSource_DIR}/src/caffe/util/*.cpp")
file(GLOB LAYER_SOURCES "${CaffeSource_DIR}/src/caffe/layers/*.cpp")

add_library(caffe SHARED ${CORE_SOURCES} ${LAYER_SOURCES})
target_compile_definitions(caffe PUBLIC CPU_ONLY)
target_include_directories(caffe PUBLIC ${CaffeSource_DIR}/include)
target_include_directories(caffe PRIVATE ${Boost_INCLUDE_DIRS})
target_include_directories(caffe PRIVATE ${HDF5_INCLUDE_DIRS})
target_include_directories(caffe PRIVATE ${Atlas_INCLUDE_DIRS})
target_link_libraries(caffe caffeproto)
target_link_libraries(caffe glog)
target_link_libraries(caffe gflags)
target_link_libraries(caffe ${Boost_LIBRARIES})
target_link_libraries(caffe ${HDF5_LIBRARIES} ${HDF5_HL_LIBRARIES})
target_link_libraries(caffe ${Atlas_LIBRARIES})