Merge pull request #10160 from JiayiFeng/develop

fix fetch_op_handle

Merge pull request #10160 from JiayiFeng/develop
fix fetch_op_handle
e5f2cb80 · Yu Yang · GitHub · 44fa8238 · 00b54d58 · e5f2cb80
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

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

未找到文件。
--- a/paddle/fluid/framework/details/fetch_op_handle.cc
+++ b/paddle/fluid/framework/details/fetch_op_handle.cc
@@ -66,7 +66,7 @@ void FetchOpHandle::RunImpl() {
    auto &t = var->Get<framework::LoDTensor>();
    if (platform::is_gpu_place(t.place())) {
 #ifdef PADDLE_WITH_CUDA
-      TensorCopy(t, cpu, *dev_ctxes_[t.place()], &tensors_[i]);
+      TensorCopy(t, cpu, *dev_ctxes_[t.place()], &tensors_[i], true);
      dev_ctxes_.at(t.place())->Wait();
 #endif
    } else {