From c9e7c407612e3746c4a218344d0b2be8916a7a6f Mon Sep 17 00:00:00 2001 From: Chen Weihang Date: Tue, 31 May 2022 18:37:45 +0800 Subject: [PATCH] [Phi] Polish assign kernel copy impl (#43061) * fix assign kernel copy impl * fix test failed --- paddle/phi/kernels/assign_kernel.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/phi/kernels/assign_kernel.cc b/paddle/phi/kernels/assign_kernel.cc index 5ed9d72a50..2349bf990a 100644 --- a/paddle/phi/kernels/assign_kernel.cc +++ b/paddle/phi/kernels/assign_kernel.cc @@ -26,7 +26,7 @@ template void AssignKernel(const Context& dev_ctx, const DenseTensor& x, DenseTensor* out) { - Copy(dev_ctx, x, x.place(), false, out); + paddle::framework::TensorCopy(x, x.place(), out); } template -- GitLab