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();
-  }
 }

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