mirror of
https://github.com/zebrajr/pytorch.git
synced 2026-01-15 12:15:51 +00:00
7cfd23636c
Fixes #ISSUE_NUMBER Pull Request resolved: https://github.com/pytorch/pytorch/pull/134935 Approved by: https://github.com/ezyang
22 lines
541 B
C++
22 lines
541 B
C++
#pragma once
|
|
|
|
#include <cstddef>
|
|
#include <cstdint>
|
|
|
|
#include <c10/macros/Export.h>
|
|
|
|
namespace caffe2::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() = default;
|
|
};
|
|
|
|
} // namespace caffe2::serialize
|