[Phi] Polish assign kernel copy impl (#43061)

* fix assign kernel copy impl * fix test failed

[Phi] Polish assign kernel copy impl (#43061)
* fix assign kernel copy impl * fix test failed
c9e7c407 · Chen Weihang · GitHub · 172739d4 · c9e7c407
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddle/phi/kernels/assign_kernel.cc paddle/phi/kernels/assign_kernel.cc +1 -1

未找到文件。
--- a/paddle/phi/kernels/assign_kernel.cc
+++ b/paddle/phi/kernels/assign_kernel.cc
@@ -26,7 +26,7 @@ template <typename Context>
 void AssignKernel(const Context& dev_ctx,
                  const DenseTensor& x,
                  DenseTensor* out) {
-  Copy<Context>(dev_ctx, x, x.place(), false, out);
+  paddle::framework::TensorCopy(x, x.place(), out);
 }

 template <typename Context>