support int8 for embedding (#50413)

78eb2d87 · seemingwang · GitHub · abad724e · 78eb2d87 · 78eb2d87
隐藏空白更改
内联并排

Showing with 2 addition and 0 deletion

paddle/phi/kernels/cpu/embedding_kernel.cc paddle/phi/kernels/cpu/embedding_kernel.cc +1 -0

paddle/phi/kernels/gpu/embedding_kernel.cu paddle/phi/kernels/gpu/embedding_kernel.cu +1 -0

未找到文件。
--- a/paddle/phi/kernels/cpu/embedding_kernel.cc
+++ b/paddle/phi/kernels/cpu/embedding_kernel.cc
@@ -115,4 +115,5 @@ PD_REGISTER_KERNEL(embedding,
                   phi::EmbeddingKernel,
                   float,
                   double,
+                   int8_t,
                   phi::dtype::bfloat16) {}
--- a/paddle/phi/kernels/gpu/embedding_kernel.cu
+++ b/paddle/phi/kernels/gpu/embedding_kernel.cu
@@ -134,5 +134,6 @@ PD_REGISTER_KERNEL(embedding,
                   phi::EmbeddingKernel,
                   float,
                   double,
+                   int8_t,
                   phi::dtype::float16,
                   phi::dtype::bfloat16) {}