From 78eb2d8776315eb46d7e78a30c78a3e35a41d649 Mon Sep 17 00:00:00 2001
From: seemingwang <seemingwang@users.noreply.github.com>
Date: Tue, 14 Feb 2023 13:51:16 +0800
Subject: [PATCH] support int8 for embedding (#50413)

---
 paddle/phi/kernels/cpu/embedding_kernel.cc | 1 +
 paddle/phi/kernels/gpu/embedding_kernel.cu | 1 +
 2 files changed, 2 insertions(+)

diff --git a/paddle/phi/kernels/cpu/embedding_kernel.cc b/paddle/phi/kernels/cpu/embedding_kernel.cc
index 7accdffe134..cfba1787a15 100644
--- 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) {}
diff --git a/paddle/phi/kernels/gpu/embedding_kernel.cu b/paddle/phi/kernels/gpu/embedding_kernel.cu
index b6bf4bce425..fdf453522e1 100644
--- 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) {}
-- 
GitLab