[NewIR]new ir support op fusion and fusion merge pass (PaddlePaddle#5…

…6111) * update * update * update * remove cerr * update * fix compile error * polish code * fix compile bug * move op fusion to cinn folder * update * using cinn dialect op * [NewIR]Refine CINN Dilact directory * fix conflict * fix deps * update * fix unittest deps * fix header bug * update * fix proto dtype depency bug * remove useless op * Fix CI conflict * Also remove included header * Final Fix try * update * fix bug * update * update * update * fix compile bug * fix op input bug * add Get attr interface * remove usless code --------- Co-authored-by: Aurelius84 <zhangliujie@baidu.com> Co-authored-by: zhhsplendid <zhhsplendid@163.com>
jiahy0825 · Oct 26, 2023 · 6c051f8 · 6c051f8
1 parent e185bd5
commit 6c051f8
Show file tree

Hide file tree

Showing 22 changed files with 5,238 additions and 48 deletions.
diff --git a/paddle/cinn/hlir/dialect/operator/CMakeLists.txt b/paddle/cinn/hlir/dialect/operator/CMakeLists.txt
@@ -1 +1,2 @@
 add_subdirectory(ir)
+add_subdirectory(transforms)
diff --git a/paddle/cinn/hlir/dialect/operator/ir/ops.yaml b/paddle/cinn/hlir/dialect/operator/ir/ops.yaml
@@ -1,8 +1,25 @@
-- op : add
-  args : (Tensor x, Tensor y)
+- op : broadcast
+  args : (Tensor x, int64_t[] broadcast_axes,  int64_t[] out_shape)
   output : Tensor(out)
   infer_meta :
-    func : ElementwiseInferMeta
+    func : CINNBroadcastInferMeta
+    param : [x, broadcast_axes, out_shape]
   kernel :
-    func : add
-  inplace : (x -> out)
+    func : expand
+    param : [x, broadcast_axes]
+
+- op : reduce_max
+  args : (Tensor x, int64_t[] axis,  bool keep_dim)
+  output : Tensor(out)
+  infer_meta :
+    func : ReduceInferMeta
+  kernel :
+    func : frobenius_norm
+
+- op : reduce_sum
+  args : (Tensor x, int64_t[] axis,  bool keep_dim)
+  output : Tensor(out)
+  infer_meta :
+    func : ReduceInferMeta
+  kernel :
+    func : frobenius_norm
diff --git a/paddle/cinn/hlir/dialect/operator/transforms/CMakeLists.txt b/paddle/cinn/hlir/dialect/operator/transforms/CMakeLists.txt
@@ -0,0 +1,10 @@
+if(NOT CINN_ONLY)
+  cinn_cc_library(
+    op_with_group_merge_pass
+    SRCS
+    group_with_group_merge_pass.cc
+    op_with_group_merge_pass.cc
+    tensor_node.cc
+    DEPS
+    pd_op_dialect)
+endif()