[PIR] Add op_callstack to Pir

paddle/cinn/hlir/framework/pir/utils.cc

@@ -314,9 +314,17 @@ static utils::Attribute ConvertArrayAttribute(
         CASE_ATTRIBUTE(float, FloatAttribute)
       } else if (attr_vec[0].isa<::pir::DoubleAttribute>()) {
         CASE_ATTRIBUTE(double, DoubleAttribute)
+      } else if (attr_vec[0].isa<::pir::StrAttribute>()) {
+        std::vector<std::string> dst_attr;
+        for (auto element : attr_vec) {
+          dst_attr.push_back(
+              element.dyn_cast<::pir::StrAttribute>().AsString());
+        }
+        dst_attr = dst_attr;
       } else {
-        LOG(FATAL) << "only support bool/int32/int64/float/double attribute in "
-                      "ArrayAttribute";
+        LOG(FATAL)
+            << "only support bool/int32/int64/float/double/string attribute in "
+               "ArrayAttribute";
       }
     }
   } else {

paddle/fluid/pir/dialect/op_generator/python_c_gen.py

@@ -52,6 +52,9 @@
 #include "paddle/fluid/pybind/op_function_common.h"
 #include "paddle/phi/common/int_array.h"
 #include "paddle/phi/core/enforce.h"
+#include "paddle/fluid/pir/dialect/operator/ir/api_builder.h"
+#include "paddle/fluid/pybind/op_callstack_utils.h"
+#include "paddle/fluid/framework/op_proto_maker.h"
 
 
 {body}
@@ -71,8 +74,24 @@
         {attrs}
 
         // Call ir static api
+        pir::InsertionPoint before_insertion_point =
+          paddle::dialect::ApiBuilder::Instance().GetCurrentInsertionPoint();
+        auto before_insertion_iterator = before_insertion_point.second;
+        pir::Attribute callstack_info_attr = get_op_callstack_info();
+        before_insertion_iterator--;
         auto static_api_out = paddle::dialect::{api_name}({args});
-
+        before_insertion_iterator++;
+        pir::InsertionPoint after_insertion_point =
+            paddle::dialect::ApiBuilder::Instance().GetCurrentInsertionPoint();
+        PADDLE_ENFORCE_EQ(before_insertion_point.first,after_insertion_point.first);
+        auto after_insertion_iterator = after_insertion_point.second;
+        for (auto block_iterator = before_insertion_iterator;
+            block_iterator != after_insertion_iterator;
+            block_iterator++) {{
+               block_iterator->set_attribute(paddle::framework::OpProtoAndCheckerMaker::
+                                        OpCreationCallstackAttrName(),
+                                    callstack_info_attr);
+        }}
         return ToPyObject(static_api_out);
     }} catch (...) {{
         ThrowExceptionToPython(std::current_exception());
@@ -94,8 +113,24 @@
         {attrs}
 
         // Call ir static api
+        pir::InsertionPoint before_insertion_point =
+          paddle::dialect::ApiBuilder::Instance().GetCurrentInsertionPoint();
+        auto before_insertion_iterator = before_insertion_point.second;
+        pir::Attribute callstack_info_attr = get_op_callstack_info();
+        before_insertion_iterator--;
         paddle::dialect::{api_name}({args});
-
+        before_insertion_iterator++;
+        pir::InsertionPoint after_insertion_point =
+            paddle::dialect::ApiBuilder::Instance().GetCurrentInsertionPoint();
+        PADDLE_ENFORCE_EQ(before_insertion_point.first,after_insertion_point.first);
+        auto after_insertion_iterator = after_insertion_point.second;
+        for (auto block_iterator = before_insertion_iterator;
+            block_iterator != after_insertion_iterator;
+            block_iterator++) {{
+               block_iterator->set_attribute(paddle::framework::OpProtoAndCheckerMaker::
+                                        OpCreationCallstackAttrName(),
+                                    callstack_info_attr);
+        }}
         return nullptr;
     }} catch (...) {{
         ThrowExceptionToPython(std::current_exception());
@@ -129,7 +164,24 @@
         {cast_attrs}
 
         // Call ir static api
+        pir::InsertionPoint before_insertion_point =
+          paddle::dialect::ApiBuilder::Instance().GetCurrentInsertionPoint();
+        auto before_insertion_iterator = before_insertion_point.second;
+        pir::Attribute callstack_info_attr = get_op_callstack_info();
+        before_insertion_iterator--;
         auto static_api_out = paddle::dialect::{api_name}({args_with_mutable_attrs});
+        before_insertion_iterator++;
+        pir::InsertionPoint after_insertion_point =
+            paddle::dialect::ApiBuilder::Instance().GetCurrentInsertionPoint();
+        PADDLE_ENFORCE_EQ(before_insertion_point.first,after_insertion_point.first);
+        auto after_insertion_iterator = after_insertion_point.second;
+        for (auto block_iterator = before_insertion_iterator;
+            block_iterator != after_insertion_iterator;
+            block_iterator++) {{
+               block_iterator->set_attribute(paddle::framework::OpProtoAndCheckerMaker::
+                                        OpCreationCallstackAttrName(),
+                                    callstack_info_attr);
+        }}
         return ToPyObject(static_api_out);
 
 

paddle/fluid/pir/dialect/operator/ir/manual_api.cc

@@ -19,6 +19,7 @@
 #include "paddle/fluid/pir/dialect/operator/ir/pd_op.h"
 #include "paddle/pir/include/core/builtin_op.h"
 #include "paddle/pir/include/core/parameter.h"
+
 namespace paddle {
 namespace dialect {
 
@@ -241,6 +242,7 @@ std::tuple<pir::Value, pir::Value> fused_gemm_epilogue(pir::Value x,
       {"trans_x", pir::BoolAttribute::get(ctx, trans_x)},
       {"trans_y", pir::BoolAttribute::get(ctx, trans_y)},
       {"activation", pir::StrAttribute::get(ctx, activation)}};
+
   auto fused_gemm_epilogue_op =
       ApiBuilder::Instance()
           .GetBuilder()

paddle/fluid/pybind/CMakeLists.txt

@@ -151,7 +151,8 @@ set(PYBIND_SRCS
     auto_parallel_py.cc
     eval_frame_tools.cc
     cpython_internals.c
-    eval_frame.c)
+    eval_frame.c
+    op_callstack_utils.cc)
 
 if(NOT WITH_SHARED_IR)
   # Note: We want to compile pir source into paddle.so directly, because