fix new ir translator c_embedding failure

PaddlePaddle · Dec 5, 2023 · bfa1210 · bfa1210
1 parent c873fb3
commit bfa1210
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/paddle/fluid/ir_adaptor/translator/op_translator.cc b/paddle/fluid/ir_adaptor/translator/op_translator.cc
@@ -2684,6 +2684,16 @@ struct SliceOpTranscriber : public OpTranscriber {
   }
 };
 
+struct CEmbeddingOpTranscriber : public OpTranscriber {
+  void HandleNonexistentAttribute(pir::IrContext* ctx,
+                                  pir::AttributeMap* attribute_map,
+                                  const OpAttributeInfo& info) override {
+    if (info.name == "vocab_size") {
+      (*attribute_map)[info.name] = pir::Int64Attribute::get(ctx, -1);
+    }
+  }
+};
+
 OpTranslator::OpTranslator() {
   pir::IrContext* ctx = pir::IrContext::Instance();
   ctx->GetOrRegisterDialect<paddle::dialect::OperatorDialect>();
@@ -2750,6 +2760,7 @@ OpTranslator::OpTranslator() {
   special_handlers["elementwise_min_grad"] = ElementwiseGradTranscriber();
   special_handlers["elementwise_mod_grad"] = ElementwiseGradTranscriber();
   special_handlers["elementwise_floordiv_grad"] = ElementwiseGradTranscriber();
+  special_handlers["c_embedding"] = CEmbeddingOpTranscriber();
 }
 
 }  // namespace translator