[Mlir-commits] [mlir] [mlir][vector] Add memref reshapes to transfer flow opt (PR #110521)
Andrzej WarzyĆski
llvmlistbot at llvm.org
Tue Oct 1 11:36:19 PDT 2024
================
@@ -257,6 +257,92 @@ func.func @collapse_shape(%in_0: memref<1x20x1xi32>, %vec: vector<4xi32>) {
return
}
+// The same regression test for expand_shape.
+
+// CHECK-LABEL: func.func @expand_shape
+// CHECK: scf.for {{.*}} {
+// CHECK: vector.transfer_read
+// CHECK: vector.transfer_write
+// CHECK: vector.transfer_write
+// CHECK: vector.transfer_read
+// CHECK: vector.transfer_write
+
+func.func @expand_shape(%in_0: memref<20xi32>, %vec: vector<1x4x1xi32>) {
----------------
banach-space wrote:
[nit] For consistency with the test below
```suggestion
func.func @expand_shape(_of_source%in_0: memref<20xi32>, %vec: vector<1x4x1xi32>) {
```
https://github.com/llvm/llvm-project/pull/110521
More information about the Mlir-commits
mailing list