From 84ec80925dc67a42d768e8205ca95be15870a942 Mon Sep 17 00:00:00 2001
From: zhangbo9674 <82555433+zhangbo9674@users.noreply.github.com>
Date: Fri, 1 Sep 2023 09:46:22 +0800
Subject: [PATCH] [IR] Refine Int64 attribute translator logic (#56842)

* fix bug

* fix bug
---
 .../ir_adaptor/translator/attribute_translator.cc     | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/paddle/fluid/ir_adaptor/translator/attribute_translator.cc b/paddle/fluid/ir_adaptor/translator/attribute_translator.cc
index cc471b41396..f6a4b94f2bf 100644
--- 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()(
-- 
GitLab