fix a bug

7 years ago · c346a345e0
parent 7e526a6cb8
commit c346a345e0
1 changed files with 2 additions and 2 deletions
--- a/paddle/fluid/recordio/header.cc
+++ b/paddle/fluid/recordio/header.cc
@ -29,8 +29,8 @@ Header::Header(uint32_t num, uint32_t sum, Compressor c, uint32_t cs)

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