mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
20bda0057e
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
15 lines
378 B
C++
15 lines
378 B
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();
|
|
|
|
} // namespace caffe2
|
|
|
|
#endif // CAFFE2_UTILS_PROTO_WRAP_H_
|