[IR] Refine Int64 attribute translator logic (#56842)

* fix bug * fix bug
PaddlePaddle · Sep 1, 2023 · 84ec809 · 84ec809
1 parent 0d08135
commit 84ec809
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/paddle/fluid/ir_adaptor/translator/attribute_translator.cc b/paddle/fluid/ir_adaptor/translator/attribute_translator.cc
@@ -176,6 +176,16 @@ class Int64ArrayAttributeVisitor : public AttributeVisitor {
   }
 };
 
+class Int64AttributeVisitor : public AttributeVisitor {
+ public:
+  using AttributeVisitor::AttributeVisitor;
+
+  ir::Attribute operator()(int is) override {
+    VLOG(10) << "translating int to Int64Attribute";
+    return ir::Int64Attribute::get(ctx, is);
+  }
+};
+
 class IntArrayAttributeVisitor : public AttributeVisitor {
  public:
   using AttributeVisitor::AttributeVisitor;
@@ -229,6 +239,7 @@ AttributeTranslator::AttributeTranslator() {
       new PlaceAttributeVisitor();
   special_visitors["ir::ArrayAttribute<ir::Int64Attribute>"] =
       new Int64ArrayAttributeVisitor();
+  special_visitors["ir::Int64Attribute"] = new Int64AttributeVisitor();
 }
 
 ir::Attribute AttributeTranslator::operator()(