refine patch

JiayiFeng · JiayiFeng · commit 07d38a9b9aff · 2018-03-19T14:32:57.000+08:00
diff --git a/paddle/fluid/operators/reader/create_shuffle_reader_op.cc b/paddle/fluid/operators/reader/create_shuffle_reader_op.cc
@@ -34,6 +34,9 @@ class ShuffleReader : public framework::DecoratedReader {
   }
 
   void ReadNext(std::vector<framework::LoDTensor>* out) override {
+    if (!HasNext()) {
+      PADDLE_THROW("There is no next data!");
+    }
     if (iteration_pos_ >= buffer_.size()) {
       VLOG(10) << "Resetting shuffle buffer";
       ReadIntoBuffers();

Original file line number	Diff line number	Diff line change
`@@ -34,6 +34,9 @@ class ShuffleReader : public framework::DecoratedReader {`
`34`	`34`	`}`
`35`	`35`
`36`	`36`	`void ReadNext(std::vector<framework::LoDTensor>* out) override {`
	`37`	`+ if (!HasNext()) {`
	`38`	`+ PADDLE_THROW("There is no next data!");`
	`39`	`+ }`
`37`	`40`	`if (iteration_pos_ >= buffer_.size()) {`
`38`	`41`	`VLOG(10) << "Resetting shuffle buffer";`
`39`	`42`	`ReadIntoBuffers();`