Wait code

7 years ago · bade579826
parent 4a330094f9
commit bade579826
1 changed files with 2 additions and 1 deletions
--- a/paddle/fluid/framework/parallel_executor.cc
+++ b/paddle/fluid/framework/parallel_executor.cc
@ -193,7 +193,8 @@ struct FetchOpHandle : public OpHandle {

  void Run() override {
    for (auto *input : inputs_) {
-      input->generated_op_->Wait(nullptr);
+      auto *var = static_cast<VarHandle *>(input);
+      var->generated_op_->Wait(this->dev_ctx_[var->place_]);
    }

    tensors_.resize(inputs_.size());