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