@@ -48,11 +48,12 @@ struct MemoryFixSizeBuffer : public dmlc::SeekStream {
4848 curr_ptr_ += nread;
4949 return nread;
5050 }
51- void Write (const void *ptr, std::size_t size) override {
52- if (size == 0 ) return ;
51+ std:: size_t Write (const void *ptr, std::size_t size) override {
52+ if (size == 0 ) return 0 ;
5353 CHECK_LE (curr_ptr_ + size, buffer_size_);
5454 std::memcpy (p_buffer_ + curr_ptr_, ptr, size);
5555 curr_ptr_ += size;
56+ return size;
5657 }
5758 void Seek (std::size_t pos) override {
5859 if (pos == kSeekEnd ) {
@@ -91,13 +92,14 @@ struct MemoryBufferStream : public dmlc::SeekStream {
9192 curr_ptr_ += nread;
9293 return nread;
9394 }
94- void Write (const void *ptr, size_t size) override {
95- if (size == 0 ) return ;
95+ std:: size_t Write (const void *ptr, size_t size) override {
96+ if (size == 0 ) return 0 ;
9697 if (curr_ptr_ + size > p_buffer_->length ()) {
9798 p_buffer_->resize (curr_ptr_+size);
9899 }
99100 std::memcpy (&(*p_buffer_)[0 ] + curr_ptr_, ptr, size);
100101 curr_ptr_ += size;
102+ return size;
101103 }
102104 void Seek (size_t pos) override {
103105 curr_ptr_ = static_cast <size_t >(pos);
@@ -127,8 +129,9 @@ class PeekableInStream : public dmlc::Stream {
127129 size_t Read (void * dptr, size_t size) override ;
128130 virtual size_t PeekRead (void * dptr, size_t size);
129131
130- void Write (const void *, size_t ) override {
132+ std:: size_t Write (const void *, size_t ) override {
131133 LOG (FATAL) << " Not implemented"
134+ return 0 ;
132135 }
133136
134137 private:
@@ -155,8 +158,9 @@ class FixedSizeStream : public PeekableInStream {
155158 [[nodiscard]] std::size_t Tell () const { return pointer_; }
156159 void Seek (size_t pos);
157160
158- void Write (const void *, size_t ) override {
161+ std:: size_t Write (const void *, size_t ) override {
159162 LOG (FATAL) << " Not implemented"
163+ return 0 ;
160164 }
161165
162166 /* !
0 commit comments