mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
a918f1d9af
Summary: To implement a stream is very annoying, since it is closely defined with the underlying storage streambuffer. So in this PR, we add ReadAdapterInterface and PyTorchStreamReader will use it. We implement IStreamAdapter as a wrapper of std::istream. And keep the user interface unchanged. Pull Request resolved: https://github.com/pytorch/pytorch/pull/15551 Reviewed By: zrphercule Differential Revision: D13568907 Pulled By: houseroad fbshipit-source-id: 93708cb801248a6c101f35cb14d1631029365c3c
21 lines
514 B
C++
21 lines
514 B
C++
#pragma once
|
|
|
|
#include <cstddef>
|
|
#include <cstdint>
|
|
|
|
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 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
|