update sig; test=develop

paddle/phi/ops/compat/embedding_sig.cc

@@ -17,15 +17,29 @@
 namespace phi {
 
 KernelSignature EmbeddingOpArgumentMapping(const ArgumentMappingContext& ctx) {
-  return KernelSignature("embedding", {"Ids", "W"}, {"padding_idx"}, {"Out"});
+  if (ctx.IsDenseTensorInput("W")) {
+    LOG(ERROR) << "is dense here";
+    return KernelSignature("embedding", {"Ids", "W"}, {"padding_idx"}, {"Out"});
+  } else {
+    LOG(ERROR) << "is selcted rows";
+    return KernelSignature(
+        "sparse_weight_embedding", {"Ids", "W"}, {"padding_idx"}, {"Out"});
+  }
 }
 
 KernelSignature EmbeddingGradOpArgumentMapping(
     const ArgumentMappingContext& ctx) {
-  return KernelSignature("embedding_grad",
-                         {"Ids", "W", GradVarName("Out")},
-                         {"padding_idx"},
-                         {GradVarName("W")});
+  if (ctx.IsDenseTensorInput("W")) {
+    return KernelSignature("embedding_grad",
+                           {"Ids", "W", GradVarName("Out")},
+                           {"padding_idx"},
+                           {GradVarName("W")});
+  } else {
+    return KernelSignature("sparse_weight_embedding_grad",
+                           {"Ids", "W", GradVarName("Out")},
+                           {"padding_idx"},
+                           {GradVarName("W")});
+  }
 }
 
 }  // namespace phi