mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
71ca600af9
Summary: Since caffe2 and torch have been consolidated, CAFFE2_API should be merged with TORCH_API. Addresses a TODO. Manually edited some references of the removed `CAFFE2_API`: * `CONTRIBUTING.md` * `caffe2/proto/CMakeLists.txt` * `cmake/ProtoBuf.cmake` * `c10/macros/Export.h` * `torch/csrc/WindowsTorchApiMacro.h` Pull Request resolved: https://github.com/pytorch/pytorch/pull/49496 Reviewed By: malfet, samestep Differential Revision: D25600726 Pulled By: janeyx99 fbshipit-source-id: 7e068d959e397ac183c097d7e9a9afeca5ddd782
24 lines
767 B
CMake
24 lines
767 B
CMake
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)
|