use phi::Copy

Author: m3ngyang

paddle/phi/kernels/cpu/put_along_axis_grad_kernel.cc

@@ -15,11 +15,11 @@
 #include "paddle/phi/kernels/put_along_axis_grad_kernel.h"
 
 #include "paddle/fluid/framework/convert_utils.h"
-#include "paddle/fluid/framework/tensor_util.h"
 #include "paddle/fluid/operators/gather_scatter_kernel.h"
 #include "paddle/fluid/platform/place.h"
 #include "paddle/phi/backends/cpu/cpu_context.h"
 #include "paddle/phi/core/kernel_registry.h"
+#include "paddle/phi/kernels/copy_kernel.h"
 
 namespace phi {
 
@@ -40,7 +40,7 @@ void PutAlongAxisGradKernel(const Context& dev_ctx,
  const auto& index_type =
  paddle::framework::TransToProtoVarType(index.dtype());
  if (x_grad) {
- paddle::framework::TensorCopy(out_grad, dev_ctx.GetPlace(), x_grad);
+ phi::Copy(dev_ctx, out_grad, dev_ctx.GetPlace(), false, x_grad);
  if (index_type == paddle::framework::proto::VarType::INT32) {
  paddle::operators::cpu_scatter_input_grad_kernel<T, int32_t>(
  // Here passing an unused argument out_grad, because it's

paddle/phi/kernels/cpu/put_along_axis_kernel.cc

@@ -15,11 +15,11 @@
 #include "paddle/phi/kernels/put_along_axis_kernel.h"
 
 #include "paddle/fluid/framework/convert_utils.h"
-#include "paddle/fluid/framework/tensor_util.h"
 #include "paddle/fluid/operators/gather_scatter_kernel.h"
 #include "paddle/fluid/platform/place.h"
 #include "paddle/phi/backends/cpu/cpu_context.h"
 #include "paddle/phi/core/kernel_registry.h"
+#include "paddle/phi/kernels/copy_kernel.h"
 
 namespace phi {
 
@@ -36,7 +36,7 @@ void PutAlongAxisKernel(const Context& dev_ctx,
  true,
  errors::PreconditionNotMet("PutAlongAxisOpKernel only runs on CPU."));
 
- paddle::framework::TensorCopy(x, dev_ctx.GetPlace(), out);
+ phi::Copy(dev_ctx, x, dev_ctx.GetPlace(), false, out);
  const auto& index_type =
  paddle::framework::TransToProtoVarType(index.dtype());
  if (reduce == "add") {

paddle/phi/kernels/gpu/put_along_axis_grad_kernel.cu

@@ -15,11 +15,11 @@
 #include "paddle/phi/kernels/put_along_axis_grad_kernel.h"
 
 #include "paddle/fluid/framework/convert_utils.h"
-#include "paddle/fluid/framework/tensor_util.h"
 #include "paddle/fluid/operators/gather_scatter_kernel.h"
 #include "paddle/fluid/platform/place.h"
 #include "paddle/phi/backends/gpu/gpu_context.h"
 #include "paddle/phi/core/kernel_registry.h"
+#include "paddle/phi/kernels/copy_kernel.h"
 
 namespace phi {
 
@@ -40,7 +40,7 @@ void PutAlongAxisGradKernel(const Context& dev_ctx,
  const auto& index_type =
  paddle::framework::TransToProtoVarType(index.dtype());
  if (x_grad) {
- paddle::framework::TensorCopy(out_grad, dev_ctx.GetPlace(), x_grad);
+ phi::Copy(dev_ctx, out_grad, dev_ctx.GetPlace(), false, x_grad);
  if (index_type == paddle::framework::proto::VarType::INT32) {
  paddle::operators::gpu_scatter_input_grad_kernel<T, int32_t>(
  out_grad, axis, index, *x_grad, dev_ctx);

paddle/phi/kernels/gpu/put_along_axis_kernel.cu

@@ -19,6 +19,7 @@
 #include "paddle/fluid/platform/place.h"
 #include "paddle/phi/backends/gpu/gpu_context.h"
 #include "paddle/phi/core/kernel_registry.h"
+#include "paddle/phi/kernels/copy_kernel.h"
 
 namespace phi {
 
@@ -38,7 +39,7 @@ void PutAlongAxisKernel(const Context& dev_ctx,
  const auto& index_type =
  paddle::framework::TransToProtoVarType(index.dtype());
 
- paddle::framework::TensorCopy(x, dev_ctx.GetPlace(), out);
+ phi::Copy(dev_ctx, x, dev_ctx.GetPlace(), false, out);
  if (reduce == "add") {
  if (index_type == paddle::framework::proto::VarType::INT32) {
  paddle::operators::gpu_scatter_add_kernel<T, int32_t>(