PaddlePaddle
diff --git a/‎paddle/fluid/platform/device/xpu/xpu_op_list.cc
Copy file name to clipboard
+18-6Lines changed: 18 additions & 6 deletions b/‎paddle/fluid/platform/device/xpu/xpu_op_list.cc
Copy file name to clipboard
+18-6Lines changed: 18 additions & 6 deletions
diff --git a/‎paddle/fluid/pybind/place.cc
Copy file name to clipboard
+1Lines changed: 1 addition & 0 deletions b/‎paddle/fluid/pybind/place.cc
Copy file name to clipboard
+1Lines changed: 1 addition & 0 deletions
diff --git a/‎paddle/phi/backends/CMakeLists.txt
Copy file name to clipboard
+8-2Lines changed: 8 additions & 2 deletions b/‎paddle/phi/backends/CMakeLists.txt
Copy file name to clipboard
+8-2Lines changed: 8 additions & 2 deletions
@@ -101,9 +101,15 @@ std::vector<vartype::Type> get_xpu_kp_op_support_type(
 
 std::vector<vartype::Type> get_xpu_op_support_type(
     const std::string& op_name, phi::backends::xpu::XPUVersion version) {
-  auto& ops = version == phi::backends::xpu::XPUVersion::XPU1
-                  ? phi::backends::xpu::get_kl1_ops()
-                  : phi::backends::xpu::get_kl2_ops();
+  phi::backends::xpu::XPUOpMap ops;
+  if (version == phi::backends::xpu::XPUVersion::XPU1) {
+    ops = phi::backends::xpu::get_kl1_ops();
+  } else if (version == phi::backends::xpu::XPUVersion::XPU2) {
+    ops = phi::backends::xpu::get_kl2_ops();
+  } else {
+    ops = phi::backends::xpu::get_kl3_ops();
+  }
+
   std::vector<vartype::Type> res;
   if (ops.find(op_name) != ops.end()) {
     auto& dtypes = ops[op_name];
@@ -115,9 +121,15 @@ std::vector<vartype::Type> get_xpu_op_support_type(
 }
 
 XPUOpListMap get_xpu_op_list(phi::backends::xpu::XPUVersion version) {
-  auto& ops = version == phi::backends::xpu::XPUVersion::XPU1
-                  ? phi::backends::xpu::get_kl1_ops()
-                  : phi::backends::xpu::get_kl2_ops();
+  phi::backends::xpu::XPUOpMap ops;
+  if (version == phi::backends::xpu::XPUVersion::XPU1) {
+    ops = phi::backends::xpu::get_kl1_ops();
+  } else if (version == phi::backends::xpu::XPUVersion::XPU2) {
+    ops = phi::backends::xpu::get_kl2_ops();
+  } else {
+    ops = phi::backends::xpu::get_kl3_ops();
+  }
+
   XPUOpListMap res;
   for (auto& op : ops) {
     std::vector<vartype::Type> op_types;
 
@@ -459,6 +459,7 @@ void BindPlace(pybind11::module &m) {  // NOLINT
   py::enum_<phi::backends::xpu::XPUVersion>(m, "XPUVersion", py::arithmetic())
       .value("XPU1", phi::backends::xpu::XPUVersion::XPU1)
       .value("XPU2", phi::backends::xpu::XPUVersion::XPU2)
+      .value("XPU3", phi::backends::xpu::XPUVersion::XPU3)
       .export_values();
   m.def("get_xpu_device_count", platform::GetXPUDeviceCount);
   m.def("get_xpu_device_version",
 
@@ -20,8 +20,14 @@ endif()
 
 if(WITH_XPU)
   list(APPEND BACKENDS_SRCS xpu/xpu_context.cc xpu/xpu_info.cc)
-  list(APPEND BACKENDS_SRCS xpu/xpu_op_list.cc xpu/xpu1_op_list.cc
-       xpu/xpu2_op_list.cc xpu/xpu_l3_strategy.cc)
+  list(
+    APPEND
+    BACKENDS_SRCS
+    xpu/xpu_op_list.cc
+    xpu/xpu1_op_list.cc
+    xpu/xpu2_op_list.cc
+    xpu/xpu3_op_list.cc
+    xpu/xpu_l3_strategy.cc)
   list(APPEND BACKENDS_DEPS phi_dynload_xpti)
 endif()