fix FeedAndSplitTensorIntoLocalScopes (#11817)

6711b7b5 · chengduo · GitHub · 8080ffd7 · 6711b7b5
显示空白变更内容
内联并排

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() {