mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
71ca600af9
Summary: Since caffe2 and torch have been consolidated, CAFFE2_API should be merged with TORCH_API. Addresses a TODO. Manually edited some references of the removed `CAFFE2_API`: * `CONTRIBUTING.md` * `caffe2/proto/CMakeLists.txt` * `cmake/ProtoBuf.cmake` * `c10/macros/Export.h` * `torch/csrc/WindowsTorchApiMacro.h` Pull Request resolved: https://github.com/pytorch/pytorch/pull/49496 Reviewed By: malfet, samestep Differential Revision: D25600726 Pulled By: janeyx99 fbshipit-source-id: 7e068d959e397ac183c097d7e9a9afeca5ddd782
24 lines
556 B
C++
24 lines
556 B
C++
#pragma once
|
|
|
|
#include <cstddef>
|
|
#include <cstdint>
|
|
|
|
#include "c10/macros/Macros.h"
|
|
|
|
namespace caffe2 {
|
|
namespace serialize {
|
|
|
|
// this is the interface for the (file/stream/memory) reader in
|
|
// PyTorchStreamReader. with this interface, we can extend the support
|
|
// besides standard istream
|
|
class TORCH_API ReadAdapterInterface {
|
|
public:
|
|
virtual size_t size() const = 0;
|
|
virtual size_t read(uint64_t pos, void* buf, size_t n, const char* what = "")
|
|
const = 0;
|
|
virtual ~ReadAdapterInterface();
|
|
};
|
|
|
|
} // namespace serialize
|
|
} // namespace caffe2
|