From 088bcb8ea8f54fa8ec4fd3bb6b782a96646c6fe0 Mon Sep 17 00:00:00 2001
From: gitliuyf <liuyinfeng_sx@163.com>
Date: Thu, 10 Oct 2024 16:47:33 +0800
Subject: [PATCH] [xpu]: modify xdnn embedding to paddle_embedding
 name;test=develop

---
 lite/kernels/xpu/__xpu__mmdnn_compute.cc | 3 ++-
 lite/kernels/xpu/lookup_table_compute.cc | 2 +-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/lite/kernels/xpu/__xpu__mmdnn_compute.cc b/lite/kernels/xpu/__xpu__mmdnn_compute.cc
index 10e8e3c318d..adb338ce80c 100644
--- a/lite/kernels/xpu/__xpu__mmdnn_compute.cc
+++ b/lite/kernels/xpu/__xpu__mmdnn_compute.cc
@@ -1112,7 +1112,7 @@ void XPUMmdnnBidEmbGrnnAttCompute2::Run() {
   int table_m = param.emb_tbl->dims()[0];
   int embed_dim = param.emb_tbl->dims()[1];
 
-  int r = xdnn::embedding<float, int64_t>(
+  int r = xdnn::paddle_embedding<float, int64_t>(
       ctx.GetRawContext(),
       param.emb_tbl->data<float>(),
       param.id0->data<int64_t>(),
@@ -1122,6 +1122,7 @@ void XPUMmdnnBidEmbGrnnAttCompute2::Run() {
       num,
       128000);
   CHECK_EQ(r, 0);
+
 }
 
 class XPUMmdnnBidEmbAttCompute
diff --git a/lite/kernels/xpu/lookup_table_compute.cc b/lite/kernels/xpu/lookup_table_compute.cc
index efbdc6aa628..a3548ea6972 100644
--- a/lite/kernels/xpu/lookup_table_compute.cc
+++ b/lite/kernels/xpu/lookup_table_compute.cc
@@ -30,7 +30,7 @@ void LookupTableCompute<WType, IDType, PType>::Run() {
   int xm = param.W->dims()[0];
   int n = param.W->dims()[1];
 
-  int r = xdnn::embedding<WType, IDType>(
+  int r = xdnn::paddle_embedding<WType, IDType>(
       ctx.GetRawContext(), /* context */
       param.W->template data<WType>(),
       param.Ids->template data<IDType>(),