[mlir] [llvm] [mlir][tensor] Fold linalg transpose with tensor pack (PR #74206)

lorenzo chelini via llvm-commits llvm-commits at lists.llvm.org
Mon Dec 11 04:35:35 PST 2023


================
@@ -114,3 +114,102 @@ func.func @pad_pack_different_padding_value(%src: tensor<16641x16xf32>) -> tenso
 // CHECK-LABEL: func.func @pad_pack_different_padding_value
 // CHECK:         tensor.pad
 // CHECK:         tensor.pack
+
+// -----
+
+func.func @tensor_pack_linalg_transpose_fold(%arg0: tensor<56x57x1x64xf32>) -> tensor<1x57x56x2x32xf32> {
----------------
chelini wrote:

Can we please also add a test with dynamic dimensions and tiles?

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


More information about the llvm-commits mailing list