Clean place check in ctc_align_kernel.cu, etc (#73159)

co63oc · web-flow · commit c8859e052260 · 2025-06-09T14:54:51.000+08:00
diff --git a/paddle/phi/kernels/gpu/ctc_align_kernel.cu b/paddle/phi/kernels/gpu/ctc_align_kernel.cu
@@ -87,11 +87,6 @@ void CTCAlignOpCUDAKernel(const Context& dev_ctx,
                           int padding_value,
                           DenseTensor* output,
                           DenseTensor* output_length) {
-  PADDLE_ENFORCE_EQ(dev_ctx.GetPlace().GetType() == phi::AllocationType::GPU,
-                    true,
-                    common::errors::InvalidArgument(
-                        "CTCAlign operator CUDA kernel must use CUDAPlace "
-                        "rather than CPUPlace."));
   const T* tokens = input.data<T>();
   auto stream = dev_ctx.stream();
 
diff --git a/paddle/phi/kernels/onednn/conv_function.h b/paddle/phi/kernels/onednn/conv_function.h
@@ -292,11 +292,6 @@ void ConvOnednn(const Context& dev_ctx,
                 bool fuse_residual_connection,
                 bool force_fp32_output,
                 DenseTensor* out) {
-  PADDLE_ENFORCE_EQ(dev_ctx.GetPlace().GetType(),
-                    AllocationType::CPU,
-                    common::errors::PreconditionNotMet(
-                        "Operator DNNL Conv must use CPUPlace"));
-
   bool is_INT8 = funcs::is_int8<T>();
 
   auto dst_dt = GetDstType(is_INT8,
diff --git a/paddle/phi/kernels/onednn/conv_grad_kernel.cc b/paddle/phi/kernels/onednn/conv_grad_kernel.cc
@@ -51,10 +51,6 @@ void ConvGradKernel(const Context& dev_ctx,
                     const std::string& data_format,
                     DenseTensor* input_grad,
                     DenseTensor* filter_grad) {
-  PADDLE_ENFORCE_EQ(dev_ctx.GetPlace().GetType(),
-                    AllocationType::CPU,
-                    common::errors::PreconditionNotMet(
-                        "Operator oneDNN ConvGrad must use CPUPlace"));
   const auto& onednn_engine = dev_ctx.GetEngine();
 
   bool is_test = dev_ctx.HasDnnAttr("is_test")
diff --git a/paddle/phi/kernels/onednn/conv_transpose_kernel.cc b/paddle/phi/kernels/onednn/conv_transpose_kernel.cc
@@ -488,11 +488,6 @@ void Conv2dTransposeKernel(const Context& dev_ctx,
                            const std::vector<int>& dilations,
                            const std::string& data_format UNUSED,
                            DenseTensor* out) {
-  PADDLE_ENFORCE_EQ(dev_ctx.GetPlace().GetType(),
-                    AllocationType::CPU,
-                    common::errors::PreconditionNotMet(
-                        "Operator oneDNN Conv must use CPUPlace"));
-
   const bool is_BFLOAT16 =
       dev_ctx.HasDnnAttr("mkldnn_data_type")
           ? PADDLE_GET_CONST(std::string,
@@ -544,11 +539,6 @@ void Conv2dTransposeBiasKernel(const Context& dev_ctx,
                                const std::vector<int>& dilations,
                                const std::string& data_format UNUSED,
                                DenseTensor* out) {
-  PADDLE_ENFORCE_EQ(dev_ctx.GetPlace().GetType(),
-                    AllocationType::CPU,
-                    common::errors::PreconditionNotMet(
-                        "Operator oneDNN Conv must use CPUPlace"));
-
   const bool is_BFLOAT16 =
       dev_ctx.HasDnnAttr("mkldnn_data_type")
           ? PADDLE_GET_CONST(std::string,
diff --git a/paddle/phi/kernels/onednn/prelu_kernel.cc b/paddle/phi/kernels/onednn/prelu_kernel.cc
@@ -25,10 +25,6 @@ void PReluKernel(const Context& dev_ctx,
                  const std::string& data_format,
                  const std::string& mode,
                  DenseTensor* out) {
-  PADDLE_ENFORCE_EQ(dev_ctx.GetPlace().GetType(),
-                    AllocationType::CPU,
-                    common::errors::PreconditionNotMet(
-                        "Operator oneDNN PReLU must use CPUPlace"));
   if (out && out->numel() == 0) {
     dev_ctx.template Alloc<T>(out);
     return;
diff --git a/paddle/phi/kernels/xpu/sigmoid_cross_entropy_with_logits_grad_kernel.cc b/paddle/phi/kernels/xpu/sigmoid_cross_entropy_with_logits_grad_kernel.cc
@@ -37,10 +37,6 @@ void SigmoidCrossEntropyWithLogitsGradKernel(
     int ignore_index,
     DenseTensor* in_grad) {
   using XPUType = typename XPUTypeTrait<T>::Type;
-  PADDLE_ENFORCE_EQ(x.place().GetType() == phi::AllocationType::XPU,
-                    true,
-                    errors::Unavailable("This kernel only runs on XPU."));
-
   dev_ctx.template Alloc<T>(in_grad);
 
   // allocate temp memory
diff --git a/paddle/phi/kernels/xpu/sigmoid_cross_entropy_with_logits_kernel.cc b/paddle/phi/kernels/xpu/sigmoid_cross_entropy_with_logits_kernel.cc
@@ -36,9 +36,6 @@ void SigmoidCrossEntropyWithLogitsKernel(
     int ignore_index,
     DenseTensor* out) {
   using XPUType = typename XPUTypeTrait<T>::Type;
-  PADDLE_ENFORCE_EQ(x.place().GetType() == phi::AllocationType::XPU,
-                    true,
-                    errors::Unavailable("This kernel only runs on XPU."));
 
   dev_ctx.template Alloc<T>(out);
   xpu::ctx_guard RAII_GUARD(dev_ctx.x_context());
