fix cpu bf16 bug; test=develop

paddle/phi/kernels/cpu/embedding_grad_kernel.cc

@@ -197,7 +197,7 @@ PD_REGISTER_KERNEL(embedding_grad,
                    phi::EmbeddingGradKernel,
                    float,
                    double,
-                   phi::dtype::float16) {}
+                   phi::dtype::bfloat16) {}
 
 PD_REGISTER_KERNEL(embedding_sparse_grad,
                    CPU,
@@ -205,4 +205,4 @@ PD_REGISTER_KERNEL(embedding_sparse_grad,
                    phi::EmbeddingSparseGradKernel,
                    float,
                    double,
-                   phi::dtype::float16) {}
+                   phi::dtype::bfloat16) {}

paddle/phi/kernels/cpu/embedding_kernel.cc

@@ -105,4 +105,4 @@ PD_REGISTER_KERNEL(embedding,
                    phi::EmbeddingKernel,
                    float,
                    double,
-                   phi::dtype::float16) {}
+                   phi::dtype::bfloat16) {}

paddle/phi/kernels/cpu/sparse_weight_embedding_grad_kernel.cc

@@ -197,7 +197,7 @@ PD_REGISTER_KERNEL(sparse_weight_embedding_grad,
                    phi::SparseWeightEmbeddingGradKernel,
                    float,
                    double,
-                   phi::dtype::float16) {}
+                   phi::dtype::bfloat16) {}
 
 PD_REGISTER_KERNEL(sparse_weight_embedding_sparse_grad,
                    CPU,
@@ -205,4 +205,4 @@ PD_REGISTER_KERNEL(sparse_weight_embedding_sparse_grad,
                    phi::SparseWeightEmbeddingSparseGradKernel,
                    float,
                    double,
-                   phi::dtype::float16) {}
+                   phi::dtype::bfloat16) {}