Skip to content

Commit e4bd63d

Browse files
authored
Merge pull request #9240 from JiayiFeng/fix_bug_in_recordio
Fix a bug of RecordIO
2 parents 873cb9b + 0b2f1b3 commit e4bd63d

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

paddle/fluid/recordio/header.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,8 +29,8 @@ Header::Header(uint32_t num, uint32_t sum, Compressor c, uint32_t cs)
2929

3030
bool Header::Parse(std::istream& is) {
3131
uint32_t magic;
32-
size_t read_size =
33-
is.readsome(reinterpret_cast<char*>(&magic), sizeof(uint32_t));
32+
is.read(reinterpret_cast<char*>(&magic), sizeof(uint32_t));
33+
size_t read_size = is.gcount();
3434
if (read_size < sizeof(uint32_t)) {
3535
return false;
3636
}

paddle/fluid/recordio/scanner.cc

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@ Scanner::Scanner(const std::string &filename) {
2828
}
2929

3030
void Scanner::Reset() {
31+
stream_->clear();
3132
stream_->seekg(0, std::ios::beg);
3233
ParseNextChunk();
3334
}

0 commit comments

Comments
 (0)