                    int64_t,
                    float,
                    double,
-                   phi::dtype::float16) {}
+                   phi::dtype::float16,
+                   phi::dtype::bfloat16) {}
 PD_REGISTER_KERNEL(greater_equal,
                    KPS,
                    ALL_LAYOUT,

                   int64_t,
                   float,
                   double,
                   phi::dtype::float16) {}

PD_REGISTER_KERNEL(equal_all,
                   KPS,

                   int64_t,
                   float,
                   double,
                   phi::dtype::float16,
                   phi::dtype::bfloat16) {}
PD_REGISTER_KERNEL(not_equal,
                   KPS,
                   ALL_LAYOUT,
