1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
if(BUILD_CAFFE2)
file(GLOB Caffe2_PROTOBUF_FILES "${CMAKE_CURRENT_SOURCE_DIR}/*.proto")
else()
set(Caffe2_PROTOBUF_FILES "${CMAKE_CURRENT_SOURCE_DIR}/torch.proto;${CMAKE_CURRENT_SOURCE_DIR}/caffe2.proto")
endif()
caffe2_protobuf_generate_cpp_py(Caffe2_PROTO_SRCS Caffe2_PROTO_HEADERS Caffe2_PROTO_PY ${Caffe2_PROTOBUF_FILES})
add_library(Caffe2_PROTO OBJECT ${Caffe2_PROTO_HEADERS} ${Caffe2_PROTO_SRCS})
if(MSVC)
if(BUILD_SHARED_LIBS)
set(TORCH_API_DEFINE "-DTORCH_API=__declspec(dllexport)")
else()
set(TORCH_API_DEFINE "-DTORCH_API=")
endif()
else()
set(TORCH_API_DEFINE "-DTORCH_API=")
endif()
target_compile_definitions(
Caffe2_PROTO PRIVATE ${TORCH_API_DEFINE})
install(FILES ${Caffe2_PROTO_HEADERS} DESTINATION include/caffe2/proto)
|
