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)