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

Prathamesh Tagore via llvm-commits llvm-commits at lists.llvm.org
Mon Dec 11 08:31:23 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> {
----------------
meshtag wrote:

I have added two tests with dynamic dimensions. 

Can we have dynamic tiles? If yes, can you please point me to an example. 

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


More information about the llvm-commits mailing list