Skip to content

P
Paddle

PaddlePaddle / Paddle
大约 1 年 前同步成功

通知 2298
Star 20931
Fork 5422
P
Paddle

体验新版 GitCode，发现更多精彩内容 >>

未验证 提交 78eb2d87 编写于 作者: S seemingwang 提交者： GitHub
浏览文件
操作

support int8 for embedding (#50413)

上级 abad724e
显示空白变更内容
内联 并排
Showing with 2 addition and 0 deletion
paddle/phi/kernels/cpu/embedding_kernel.cc
浏览文件 @ 78eb2d87
...@@ -115,4 +115,5 @@ PD_REGISTER_KERNEL(embedding, ...@@ -115,4 +115,5 @@ PD_REGISTER_KERNEL(embedding,
phi::EmbeddingKernel, phi::EmbeddingKernel,
float, float,
double, double,
int8_t,
phi::dtype::bfloat16) {} phi::dtype::bfloat16) {}
paddle/phi/kernels/gpu/embedding_kernel.cu
浏览文件 @ 78eb2d87
...@@ -134,5 +134,6 @@ PD_REGISTER_KERNEL(embedding, ...@@ -134,5 +134,6 @@ PD_REGISTER_KERNEL(embedding,
phi::EmbeddingKernel, phi::EmbeddingKernel,
float, float,
double, double,
int8_t,
phi::dtype::float16, phi::dtype::float16,
phi::dtype::bfloat16) {} phi::dtype::bfloat16) {}
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或