diff --git a/paddle/phi/kernels/cpu/embedding_grad_kernel.cc b/paddle/phi/kernels/cpu/embedding_grad_kernel.cc
index 67c28eefc87f0a9160f107204b3c5ca4c4df8521..4a6b1014277d70d013788fcb84ce5cc47eb2d8ec 100644
--- a/paddle/phi/kernels/cpu/embedding_grad_kernel.cc
+++ b/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) {}
diff --git a/paddle/phi/kernels/cpu/embedding_kernel.cc b/paddle/phi/kernels/cpu/embedding_kernel.cc
index 63ea7004d42254760b41653a36656b9dbc3add3e..6c92e9a660a7e50bb9cbe91a350b88cb884290f6 100644
--- a/paddle/phi/kernels/cpu/embedding_kernel.cc
+++ b/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) {}
diff --git a/paddle/phi/kernels/cpu/sparse_weight_embedding_grad_kernel.cc b/paddle/phi/kernels/cpu/sparse_weight_embedding_grad_kernel.cc
index 743faa3e43e96396e96d69baa0e2babe08ad5287..89237d3f6e87696c0efc59b0818a41913bd147d0 100644
--- a/paddle/phi/kernels/cpu/sparse_weight_embedding_grad_kernel.cc
+++ b/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) {}