1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
|
#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_ = std::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
|
