mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
0a809fc8b1
Summary: Properly annotated all apis for cpu front. Checked with cmake using cmake -DUSE_ATEN=ON -DUSE_CUDA=OFF -DBUILD_ATEN=ON and resulting libcaffe2.so has about 11k symbols. Pull Request resolved: https://github.com/pytorch/pytorch/pull/10504 Reviewed By: ezyang Differential Revision: D9316491 Pulled By: Yangqing fbshipit-source-id: 215659abf350af7032e9a4b0f28a856babab2454
15 lines
381 B
C++
15 lines
381 B
C++
#ifndef CAFFE2_UTILS_PROTO_WRAP_H_
|
|
#define CAFFE2_UTILS_PROTO_WRAP_H_
|
|
|
|
#include "caffe2/core/common.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).
|
|
CAFFE2_API void ShutdownProtobufLibrary();
|
|
|
|
} // namespace caffe2
|
|
|
|
#endif // CAFFE2_UTILS_PROTO_WRAP_H_
|