[Mlir-commits] [mlir] [MLIR] Add allow Insert/extract slice option to pack/unpack op (PR #117340)

[Zhuoran Yin]

================
@@ -0,0 +1,121 @@
+// RUN: mlir-opt %s --transform-interpreter --split-input-file -canonicalize | FileCheck %s
+
+// For pack op, we use lowerPadLikeWithInsertSlice = false to ensure no insert_slice is generated.
+// This allows linalg.transpose to be fused as a producer operation. Alternatively, without this attribute
+// insert_slice will be generated and fusion blocked.
+
+module {
+  // CHECK-label: func @fuse_pack_as_producer
+  // CHECK:       scf.forall {{.*}} {
+  // CHECK:         linalg.transpose
----------------
jerryyin wrote:

I think @Max191 can potentially give you more details if you are interested. The goal here is not to have the producer fusion flow disrupted. So potentially we can chain additional ops before transpose when enabled. Follow-up PR may come up to further the progress on this direction.

https://github.com/llvm/llvm-project/pull/117340