【pir】modify Decomp_gen for mutable_attr (#68539)

xiaoguoguo626807 · web-flow · commit d72a8527c61b · 2024-10-09T10:30:50.000+08:00
* modify paddletest

* modify paddletest

* remove dynamic shape

* modify ci

* modify mutable attribute gen for decomp
diff --git a/paddle/fluid/primitive/codegen/templates/decomp/generated_decomp.j2 b/paddle/fluid/primitive/codegen/templates/decomp/generated_decomp.j2
@@ -15,6 +15,7 @@ namespace paddle {
 namespace dialect {
 using IntArray = paddle::experimental::IntArray;
 {% macro sig(fwd_name, class_name, inputs, attrs, outputs) %}
+
 {% set input_names=[] %}
 {% set attr_names=[] %}
 {% set output_names=[] %}
@@ -72,7 +73,7 @@ std::vector<std::vector<pir::Value>> {{class_name}}::Decomp(pir::Operation* op)
     {% if attrs %}
       {% for item in attrs %}
       {% do attr_names.append(item.name) %}
-      {% if item.typename == "Scalar" and item.support_tensor %}
+      {% if item.typename.startswith("Scalar") and item.support_tensor %}
 
   Tensor {{item.name}}_(std::make_shared<primitive::LazyTensor>(op_obj.{{item.name}}()));
 
diff --git a/paddle/phi/ops/yaml/op_compat.yaml b/paddle/phi/ops/yaml/op_compat.yaml
@@ -3618,22 +3618,22 @@
   outputs:
     out : Out
   int_array:
-      sections :
-          data_type : int
-          support_tensor : true
+    sections :
+      data_type : int
+      support_tensor : true
   scalar :
-      axis :
-        data_type : int
-        support_tensor : true
+    axis :
+      data_type : int
+      support_tensor : true
   extra :
     attrs : [bool use_mkldnn = false, str mkldnn_data_type = "float32"]
 
 - op : split_with_num
   scalar :
-      axis :
-        data_type : int
-        support_tensor : true
-        tensor_name : AxisTensor
+    axis :
+      data_type : int
+      support_tensor : true
+      tensor_name : AxisTensor
 
 - op : sqrt
   backward : sqrt_grad, sqrt_double_grad (sqrt_grad_grad)
