[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