mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
60e8dc4374
Fixes #ISSUE_NUMBER Pull Request resolved: https://github.com/pytorch/pytorch/pull/134925 Approved by: https://github.com/ezyang
38 lines
1.2 KiB
C++
38 lines
1.2 KiB
C++
#ifndef CAFFE2_UTILS_PROTO_WRAP_H_
|
|
#define CAFFE2_UTILS_PROTO_WRAP_H_
|
|
|
|
#include <c10/util/Logging.h>
|
|
|
|
namespace caffe2 {
|
|
|
|
// A wrapper function to shut down protobuf library (this is needed in ASAN
|
|
// testing and valgrind cases to avoid protobuf appearing to "leak" memory).
|
|
TORCH_API void ShutdownProtobufLibrary();
|
|
|
|
// 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();
|
|
} // namespace caffe2
|
|
|
|
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();
|
|
|
|
} // namespace ONNX_NAMESPACE
|
|
|
|
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();
|
|
|
|
void ShutdownProtobufLibrary();
|
|
|
|
} // namespace torch
|
|
#endif // CAFFE2_UTILS_PROTO_WRAP_H_
|