[CustomDevice] register bf16 empty kernel for custom devices (#62140)

SylarTiaNII · web-flow · commit b09e0d72cdba · 2024-02-29T08:44:17.000+08:00
diff --git a/paddle/phi/kernels/empty_kernel.cc b/paddle/phi/kernels/empty_kernel.cc
@@ -158,7 +158,8 @@ PD_REGISTER_KERNEL(empty,
                    int,
                    int64_t,
                    bool,
-                   phi::dtype::float16) {}
+                   phi::dtype::float16,
+                   phi::dtype::bfloat16) {}
 PD_REGISTER_KERNEL(empty_like,
                    Custom,
                    ALL_LAYOUT,
@@ -171,7 +172,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
