Wait code

bade5798 · Yu Yang · 4a330094 · bade5798
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

paddle/fluid/framework/parallel_executor.cc paddle/fluid/framework/parallel_executor.cc +2 -1

未找到文件。
--- 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());