fix FeedAndSplitTensorIntoLocalScopes (#11817)

7 years ago · 6711b7b5f1
parent 8080ffd7eb
commit 6711b7b5f1
1 changed files with 3 additions and 0 deletions
--- 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() {