diff --git a/src/io/dmlc_simple.cpp b/src/io/dmlc_simple.cpp index 3fbf34734..0448bd578 100644 --- a/src/io/dmlc_simple.cpp +++ b/src/io/dmlc_simple.cpp @@ -153,7 +153,7 @@ class StdFile : public dmlc::Stream { return std::fread(ptr, 1, size, fp); } virtual void Write(const void *ptr, size_t size) { - std::fwrite(ptr, size, 1, fp); + Check(std::fwrite(ptr, size, 1, fp) == 1, "StdFile::Write: fwrite error!"); } virtual void Seek(size_t pos) { std::fseek(fp, static_cast(pos), SEEK_SET); // NOLINT(*) diff --git a/src/utils/io.h b/src/utils/io.h index 5b366e51c..1fd09310e 100644 --- a/src/utils/io.h +++ b/src/utils/io.h @@ -33,7 +33,7 @@ class FileStream : public ISeekStream { return std::fread(ptr, size, 1, fp); } virtual void Write(const void *ptr, size_t size) { - std::fwrite(ptr, size, 1, fp); + Check(std::fwrite(ptr, size, 1, fp) == 1, "FileStream::Write: fwrite error!"); } virtual void Seek(size_t pos) { std::fseek(fp, static_cast(pos), SEEK_SET); // NOLINT(*)