fix num_blocks==2

7 years ago · 3666d7c02f
parent 139ae08fdf
commit 3666d7c02f
1 changed files with 2 additions and 2 deletions
--- a/paddle/fluid/operators/listen_and_serv_op.cc
+++ b/paddle/fluid/operators/listen_and_serv_op.cc
@ -157,8 +157,8 @@ class ListenAndServOp : public framework::OperatorBase {
        }));
      }
      for (int i = 0; i < num_blocks - 2; ++i) fs[i].wait();
-      // Run global block at final step
-      if (num_blocks > 2) {
+      // Run global block at final step, or block1 if there are only 2 blocks
+      if (num_blocks >= 2) {
        try {
          executor.Run(*program, &recv_scope, num_blocks - 1,
                       false /*create_local_scope*/, false /*create_vars*/);