mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
Summary: Add a rule to wrap proto_utils.h and depend on that, rather than relying on a glob which violates package boundaries. Reviewed By: igorsugak Differential Revision: D29273453 fbshipit-source-id: 08f198a03d06ee2fdf61f5dbe1d0087db22aec8b
46 lines
1.4 KiB
C++
46 lines
1.4 KiB
C++
#include "caffe2/utils/proto_wrap.h"
|
|
|
|
#include <google/protobuf/stubs/common.h>
|
|
#include <google/protobuf/generated_message_util.h>
|
|
|
|
namespace ONNX_NAMESPACE {
|
|
|
|
// ONNX wrapper functions for protobuf's GetEmptyStringAlreadyInited() function
|
|
// used to avoid duplicated global variable in the case when protobuf
|
|
// is built with hidden visibility.
|
|
TORCH_API const ::std::string& GetEmptyStringAlreadyInited() {
|
|
return ::google::protobuf::internal::GetEmptyStringAlreadyInited();
|
|
}
|
|
|
|
} // namespace ONNX_NAMESPACE
|
|
|
|
namespace caffe2 {
|
|
|
|
// Caffe2 wrapper functions for protobuf's GetEmptyStringAlreadyInited() function
|
|
// used to avoid duplicated global variable in the case when protobuf
|
|
// is built with hidden visibility.
|
|
TORCH_API const ::std::string& GetEmptyStringAlreadyInited() {
|
|
return ::google::protobuf::internal::GetEmptyStringAlreadyInited();
|
|
}
|
|
|
|
void ShutdownProtobufLibrary() {
|
|
::google::protobuf::ShutdownProtobufLibrary();
|
|
}
|
|
|
|
} // namespace caffe2
|
|
|
|
namespace torch {
|
|
|
|
// Caffe2 wrapper functions for protobuf's GetEmptyStringAlreadyInited() function
|
|
// used to avoid duplicated global variable in the case when protobuf
|
|
// is built with hidden visibility.
|
|
TORCH_API const ::std::string& GetEmptyStringAlreadyInited() {
|
|
return ::google::protobuf::internal::GetEmptyStringAlreadyInited();
|
|
}
|
|
|
|
void ShutdownProtobufLibrary() {
|
|
::google::protobuf::ShutdownProtobufLibrary();
|
|
}
|
|
|
|
} // namespace torch
|