update sig; test=develop

paddle/phi/ops/compat/embedding_sig.cc

@@ -17,15 +17,29 @@
 namespace phi {
 
 KernelSignature EmbeddingOpArgumentMapping(const ArgumentMappingContext& ctx) {
+  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) {
+  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