[XPU] float64 support for elementwise_add (#64542)

houj04 · web-flow · commit fbff860a672b · 2024-05-24T14:02:44.000+08:00
diff --git a/paddle/phi/backends/xpu/xpu3_op_list.cc b/paddle/phi/backends/xpu/xpu3_op_list.cc
@@ -275,7 +275,8 @@ XPUOpMap& get_kl3_ops() {
                      phi::DataType::INT64,
                      phi::DataType::INT32})},
       {"elementwise_add",
-       XPUKernelSet({phi::DataType::FLOAT32,
+       XPUKernelSet({phi::DataType::FLOAT64,
+                     phi::DataType::FLOAT32,
                      phi::DataType::FLOAT16,
                      phi::DataType::BFLOAT16,
                      phi::DataType::INT64,
diff --git a/paddle/phi/kernels/xpu/elementwise_add_kernel.cc b/paddle/phi/kernels/xpu/elementwise_add_kernel.cc
@@ -37,6 +37,7 @@ void AddKernel(const Context& dev_ctx,
   if (x.dtype() == phi::DataType::FLOAT32 &&
       (y.dtype() == phi::DataType::BFLOAT16 ||
        y.dtype() == phi::DataType::FLOAT16)) {
+    // special case for "float32 + bfloat16", or "float32 + float16"
     auto dev_version =
         phi::backends::xpu::get_xpu_version(dev_ctx.GetPlace().GetDeviceId());
     if (dev_version >= phi::backends::xpu::XPUVersion::XPU3 &&
@@ -123,6 +124,7 @@ PD_REGISTER_KERNEL(add,
                    XPU,
                    ALL_LAYOUT,
                    phi::AddKernel,
+                   double,
                    phi::dtype::float16,
                    phi::dtype::bfloat16,
                    float,
