Update put_along_axis_grad_kernel.cc (#73049)

Author: co63oc

paddle/phi/kernels/cpu/put_along_axis_grad_kernel.cc

@@ -35,11 +35,6 @@ void PutAlongAxisGradKernel(const Context& dev_ctx,
                             bool include_self,
                             DenseTensor* x_grad,
                             DenseTensor* value_grad) {
-  PADDLE_ENFORCE_EQ(
-      dev_ctx.GetPlace().GetType() == phi::AllocationType::CPU,
-      true,
-      errors::PreconditionNotMet("PutAlongAxisGradOpKernel only runs on CPU."));
-
   const auto& index_type = index.dtype();
   if (x_grad) {
     phi::Copy(dev_ctx, out_grad, dev_ctx.GetPlace(), false, x_grad);

paddle/phi/kernels/cpu/put_along_axis_kernel.cc

@@ -32,11 +32,6 @@ void PutAlongAxisKernel(const Context& dev_ctx,
                         const std::string& reduce,
                         bool include_self,
                         DenseTensor* out) {
-  PADDLE_ENFORCE_EQ(
-      dev_ctx.GetPlace().GetType() == phi::AllocationType::CPU,
-      true,
-      errors::PreconditionNotMet("PutAlongAxisOpKernel only runs on CPU."));
-
   phi::Copy(dev_ctx, x, dev_ctx.GetPlace(), false, out);
   const auto& index_type = index.dtype();
   if (reduce == "add") {