mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
564d00f364
This reverts commit 7cfd23636c8fa6fcbb8bf3ea34e15b847ec9ad9d. Reverted https://github.com/pytorch/pytorch/pull/134935 on behalf of https://github.com/izaitsevfb due to breaks internal builds, caffe2 is still used internally ([comment](https://github.com/pytorch/pytorch/pull/134935#issuecomment-2349368152))
41 lines
1.0 KiB
C++
41 lines
1.0 KiB
C++
#include "caffe2/serialize/istream_adapter.h"
|
|
#include <c10/util/Exception.h>
|
|
|
|
namespace caffe2 {
|
|
namespace serialize {
|
|
|
|
IStreamAdapter::IStreamAdapter(std::istream* istream) : istream_(istream) {}
|
|
|
|
size_t IStreamAdapter::size() const {
|
|
auto prev_pos = istream_->tellg();
|
|
validate("getting the current position");
|
|
istream_->seekg(0, istream_->end);
|
|
validate("seeking to end");
|
|
auto result = istream_->tellg();
|
|
validate("getting size");
|
|
istream_->seekg(prev_pos);
|
|
validate("seeking to the original position");
|
|
return result;
|
|
}
|
|
|
|
size_t IStreamAdapter::read(uint64_t pos, void* buf, size_t n, const char* what)
|
|
const {
|
|
istream_->seekg(pos);
|
|
validate(what);
|
|
istream_->read(static_cast<char*>(buf), n);
|
|
validate(what);
|
|
return n;
|
|
}
|
|
|
|
void IStreamAdapter::validate(const char* what) const {
|
|
if (!*istream_) {
|
|
AT_ERROR("istream reader failed: ", what, ".");
|
|
}
|
|
}
|
|
|
|
// NOLINTNEXTLINE(modernize-use-equals-default)
|
|
IStreamAdapter::~IStreamAdapter() {}
|
|
|
|
} // namespace serialize
|
|
} // namespace caffe2
|