xref: /aosp_15_r20/external/pytorch/caffe2/utils/proto_wrap.h (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1*da0073e9SAndroid Build Coastguard Worker #ifndef CAFFE2_UTILS_PROTO_WRAP_H_
2*da0073e9SAndroid Build Coastguard Worker #define CAFFE2_UTILS_PROTO_WRAP_H_
3*da0073e9SAndroid Build Coastguard Worker 
4*da0073e9SAndroid Build Coastguard Worker #include <c10/util/Logging.h>
5*da0073e9SAndroid Build Coastguard Worker 
6*da0073e9SAndroid Build Coastguard Worker namespace caffe2 {
7*da0073e9SAndroid Build Coastguard Worker 
8*da0073e9SAndroid Build Coastguard Worker // A wrapper function to shut down protobuf library (this is needed in ASAN
9*da0073e9SAndroid Build Coastguard Worker // testing and valgrind cases to avoid protobuf appearing to "leak" memory).
10*da0073e9SAndroid Build Coastguard Worker TORCH_API void ShutdownProtobufLibrary();
11*da0073e9SAndroid Build Coastguard Worker 
12*da0073e9SAndroid Build Coastguard Worker } // namespace caffe2
13*da0073e9SAndroid Build Coastguard Worker 
14*da0073e9SAndroid Build Coastguard Worker #endif // CAFFE2_UTILS_PROTO_WRAP_H_
15