fix FeedAndSplitTensorIntoLocalScopes (#11817) (#11852)

110a1a98 · chengduo · GitHub · 1ad80714 · 110a1a98
隐藏空白更改
内联并排

Showing with 3 addition and 0 deletion

paddle/fluid/framework/parallel_executor.cc paddle/fluid/framework/parallel_executor.cc +3 -0

未找到文件。
--- a/paddle/fluid/framework/parallel_executor.cc
+++ b/paddle/fluid/framework/parallel_executor.cc
@@ -253,6 +253,9 @@ void ParallelExecutor::FeedAndSplitTensorIntoLocalScopes(
      t->set_lod(lod_tensors[j].lod());
    }
  }
+  for (auto &p : member_->places_) {
+    platform::DeviceContextPool::Instance().Get(p)->Wait();
+  }
 }

 ParallelExecutor::~ParallelExecutor() {