fix FillInferBuf (PaddlePaddle#218)

Co-authored-by: root <[email protected]>
zmxdream · Feb 23, 2023 · 8fa81f6 · 8fa81f6
1 parent 7ab8d80
commit 8fa81f6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/paddle/fluid/framework/data_feed.cu b/paddle/fluid/framework/data_feed.cu
@@ -2133,8 +2133,9 @@ int GraphDataGenerator::FillInferBuf() {
     global_infer_node_type_start[infer_cursor] += total_row_;
     infer_node_end_ = global_infer_node_type_start[infer_cursor];
     cursor_ = infer_cursor;
+    return 1;
   }
-  return 1;
+  return 0;
 }
 
 void GraphDataGenerator::ClearSampleState() {