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 |