From f57a91bb4fe393a62729d929c785b36aa6c7816d Mon Sep 17 00:00:00 2001
From: SigureMo <sigure.qaq@gmail.com>
Date: Wed, 6 Mar 2024 07:50:17 +0000
Subject: [PATCH 1/2] [PIR] Filter out attribute `op_callstack` when print
 program

---
 paddle/pir/src/core/ir_printer.cc | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/paddle/pir/src/core/ir_printer.cc b/paddle/pir/src/core/ir_printer.cc
index de75d6d2fc6036..c32154140e7393 100644
--- a/paddle/pir/src/core/ir_printer.cc
+++ b/paddle/pir/src/core/ir_printer.cc
@@ -17,6 +17,7 @@
 #include <string>
 #include <unordered_map>
 
+#include "paddle/fluid/framework/op_proto_maker.h"
 #include "paddle/pir/include/core/block.h"
 #include "paddle/pir/include/core/builtin_attribute.h"
 #include "paddle/pir/include/core/builtin_type.h"
@@ -279,6 +280,11 @@ void IrPrinter::PrintAttributeMap(Operation* op) {
   AttributeMap attributes = op->attributes();
   std::map<std::string, Attribute, std::less<>> order_attributes(
       attributes.begin(), attributes.end());
+
+  // Filter out the callstack attribute
+  order_attributes.erase(
+      paddle::framework::OpProtoAndCheckerMaker::OpCreationCallstackAttrName());
+
   os << " {";
 
   pir::detail::PrintInterleave(

From 2d0a1f0560b78170daafad21687bf199ed257555 Mon Sep 17 00:00:00 2001
From: SigureMo <sigure.qaq@gmail.com>
Date: Wed, 6 Mar 2024 08:36:50 +0000
Subject: [PATCH 2/2] dont include fluid

---
 paddle/pir/src/core/ir_printer.cc | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/paddle/pir/src/core/ir_printer.cc b/paddle/pir/src/core/ir_printer.cc
index c32154140e7393..e2bc7757f9de46 100644
--- a/paddle/pir/src/core/ir_printer.cc
+++ b/paddle/pir/src/core/ir_printer.cc
@@ -17,7 +17,6 @@
 #include <string>
 #include <unordered_map>
 
-#include "paddle/fluid/framework/op_proto_maker.h"
 #include "paddle/pir/include/core/block.h"
 #include "paddle/pir/include/core/builtin_attribute.h"
 #include "paddle/pir/include/core/builtin_type.h"
@@ -282,8 +281,7 @@ void IrPrinter::PrintAttributeMap(Operation* op) {
       attributes.begin(), attributes.end());
 
   // Filter out the callstack attribute
-  order_attributes.erase(
-      paddle::framework::OpProtoAndCheckerMaker::OpCreationCallstackAttrName());
+  order_attributes.erase("op_callstack");
 
   os << " {";