mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01: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
29 lines
740 B
C++
29 lines
740 B
C++
#include "caffe2/serialize/file_adapter.h"
|
|
#include <c10/util/Exception.h>
|
|
#include "caffe2/core/common.h"
|
|
|
|
namespace caffe2 {
|
|
namespace serialize {
|
|
|
|
FileAdapter::FileAdapter(const std::string& file_name) {
|
|
file_stream_.open(file_name, std::ifstream::in | std::ifstream::binary);
|
|
if (!file_stream_) {
|
|
AT_ERROR("open file failed, file path: ", file_name);
|
|
}
|
|
istream_adapter_ = caffe2::make_unique<IStreamAdapter>(&file_stream_);
|
|
}
|
|
|
|
size_t FileAdapter::size() const {
|
|
return istream_adapter_->size();
|
|
}
|
|
|
|
size_t FileAdapter::read(uint64_t pos, void* buf, size_t n, const char* what)
|
|
const {
|
|
return istream_adapter_->read(pos, buf, n, what);
|
|
}
|
|
|
|
FileAdapter::~FileAdapter() {}
|
|
|
|
} // namespace serialize
|
|
} // namespace caffe2
|