fix FeedAndSplitTensorIntoLocalScopes (#11817) (#11852)

chengduo · web-flow · commit 110a1a986a04 · 2018-07-02T09:11:55.000+08:00
diff --git 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() {

Original file line number	Diff line number	Diff line change
`@@ -253,6 +253,9 @@ void ParallelExecutor::FeedAndSplitTensorIntoLocalScopes(`
`253`	`253`	`t->set_lod(lod_tensors[j].lod());`
`254`	`254`	`}`
`255`	`255`	`}`
	`256`	`+ for (auto &p : member_->places_) {`
	`257`	`+ platform::DeviceContextPool::Instance().Get(p)->Wait();`
	`258`	`+ }`
`256`	`259`	`}`
`257`	`260`
`258`	`261`	`ParallelExecutor::~ParallelExecutor() {`