Fix merge op

3f88fad0 · Yu Yang · c42c4a67 · 3f88fad0
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

paddle/fluid/framework/details/fetch_op_handle.cc paddle/fluid/framework/details/fetch_op_handle.cc +2 -2

未找到文件。
--- a/paddle/fluid/framework/details/fetch_op_handle.cc
+++ b/paddle/fluid/framework/details/fetch_op_handle.cc
@@ -67,9 +67,9 @@ void FetchOpHandle::RunImpl() {
      tensors_[i].ShareDataWith(t);
      tensors_[i].set_lod(t.lod());
    }
-
-    this->WaitAndMergeCPUTensors();
  }
+
+  this->WaitAndMergeCPUTensors();
 }

 std::string FetchOpHandle::Name() const { return "Fetch"; }