[XPU][Phi kernels] empty support bfloat16 (#60212)

dynamicheart · web-flow · commit 03639fe96892 · 2023-12-21T17:13:34.000+08:00
diff --git a/paddle/phi/backends/xpu/xpu2_op_list.cc b/paddle/phi/backends/xpu/xpu2_op_list.cc
@@ -329,7 +329,8 @@ XPUOpMap& get_kl2_ops() {
                      phi::DataType::BOOL,
                      phi::DataType::FLOAT16,
                      phi::DataType::FLOAT32,
-                     phi::DataType::FLOAT64})},
+                     phi::DataType::FLOAT64,
+                     phi::DataType::BFLOAT16})},
       {"embedding_grad",
        XPUKernelSet({phi::DataType::FLOAT32,
                      phi::DataType::FLOAT16,
diff --git a/paddle/phi/backends/xpu/xpu3_op_list.cc b/paddle/phi/backends/xpu/xpu3_op_list.cc
@@ -315,7 +315,8 @@ XPUOpMap& get_kl3_ops() {
                      phi::DataType::BOOL,
                      phi::DataType::FLOAT16,
                      phi::DataType::FLOAT32,
-                     phi::DataType::FLOAT64})},
+                     phi::DataType::FLOAT64,
+                     phi::DataType::BFLOAT16})},
       {"embedding_grad",
        XPUKernelSet({phi::DataType::FLOAT32,
                      phi::DataType::FLOAT16,
diff --git a/paddle/phi/kernels/empty_kernel.cc b/paddle/phi/kernels/empty_kernel.cc
@@ -125,7 +125,8 @@ PD_REGISTER_KERNEL(empty,
                    int,
                    int64_t,
                    bool,
-                   phi::dtype::float16) {}
+                   phi::dtype::float16,
+                   phi::dtype::bfloat16) {}
 PD_REGISTER_KERNEL(empty_like,
                    XPU,
                    ALL_LAYOUT,
@@ -138,7 +139,8 @@ PD_REGISTER_KERNEL(empty_like,
                    int,
                    int64_t,
                    bool,
-                   phi::dtype::float16) {
+                   phi::dtype::float16,
+                   phi::dtype::bfloat16) {
   kernel->InputAt(0).SetBackend(phi::Backend::ALL_BACKEND);
 }
 #endif
diff --git a/test/xpu/test_empty_op_xpu.py b/test/xpu/test_empty_op_xpu.py
@@ -58,6 +58,7 @@ def verify_output(self, outs):
                 'uint8',
                 'float16',
                 'int16',
+                'uint16',
             ]:
                 max_value = np.nanmax(outs[0])
                 min_value = np.nanmin(outs[0])
