add share data xpu kernel (#71229)

AndSonder · web-flow · commit 839844b1f188 · 2025-02-24T11:56:08.000+08:00
diff --git a/paddle/phi/backends/xpu/xpu3_op_list.cc b/paddle/phi/backends/xpu/xpu3_op_list.cc
@@ -1264,6 +1264,16 @@ XPUOpMap& get_kl3_ops() {
        XPUKernelSet({phi::DataType::FLOAT32})},
       {"sigmoid_cross_entropy_with_logits",
        XPUKernelSet({phi::DataType::FLOAT32})},
+      {"share_data",
+       XPUKernelSet({phi::DataType::BOOL,
+                     phi::DataType::INT32,
+                     phi::DataType::INT8,
+                     phi::DataType::UINT8,
+                     phi::DataType::INT64,
+                     phi::DataType::FLOAT32,
+                     phi::DataType::FLOAT64,
+                     phi::DataType::FLOAT16,
+                     phi::DataType::BFLOAT16})},
       {"shape",
        XPUKernelSet({phi::DataType::FLOAT32,
                      phi::DataType::INT64,
diff --git a/paddle/phi/kernels/xpu/share_data_kernel.cc b/paddle/phi/kernels/xpu/share_data_kernel.cc
@@ -0,0 +1,30 @@
+// Copyright (c) 2025 PaddlePaddle Authors. All Rights Reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     https://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#include "paddle/phi/common/data_type.h"
+#include "paddle/phi/kernels/impl/share_data_kernel_impl.h"
+
+PD_REGISTER_KERNEL(share_data,
+                   XPU,
+                   ALL_LAYOUT,
+                   phi::ShareDataKernel,
+                   bool,
+                   int,
+                   int8_t,
+                   uint8_t,
+                   int64_t,
+                   float,
+                   double,
+                   phi::dtype::float16,
+                   phi::dtype::bfloat16) {}
