add pir_debug flag (#63841)

zyfncg · web-flow · commit eb9a3a2164bd · 2024-04-25T13:04:11.000+08:00
diff --git a/paddle/common/flags.cc b/paddle/common/flags.cc
@@ -1489,6 +1489,10 @@ PHI_DEFINE_EXPORTED_int32(
     "been dropped when you are profiling, try increasing this value.");
 
 PHI_DEFINE_EXPORTED_bool(print_ir, false, "Whether print ir debug str.");
+
+PHI_DEFINE_EXPORTED_bool(pir_debug,
+                         false,
+                         "Whether print more pir debug info.");
 PHI_DEFINE_EXPORTED_bool(prim_skip_dynamic,
                          false,
                          "Whether to skip decomposing op with dynamic shape.");
diff --git 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/common/flags.h"
 #include "paddle/pir/include/core/block.h"
 #include "paddle/pir/include/core/builtin_attribute.h"
 #include "paddle/pir/include/core/builtin_type.h"
@@ -27,6 +28,8 @@
 #include "paddle/pir/include/core/utils.h"
 #include "paddle/pir/include/core/value.h"
 
+COMMON_DECLARE_bool(pir_debug);
+
 namespace pir {
 
 namespace {
@@ -182,6 +185,10 @@ void IrPrinter::PrintOperationWithNoRegion(Operation* op) {
 
   os << " \"" << op->name() << "\"";
 
+  if (FLAGS_pir_debug) {
+    os << " [id:" << op->id() << "]";
+  }
+
   // TODO(lyk): add API to get operands directly
   PrintOpOperands(op);
 
