[Mlir-commits] [mlir] [mlir][ArmSME] Fold transpose into xfer read to enable in-flight transpose (PR #92562)

Cullen Rhodes llvmlistbot at llvm.org
Mon May 20 00:06:24 PDT 2024


================
@@ -356,6 +356,20 @@ struct TransposeOpToArmSMELowering
       return failure();
 
     auto loc = transposeOp.getLoc();
+    Value input = transposeOp.getVector();
+
+    if (auto xferOp = input.getDefiningOp<vector::TransferReadOp>()) {
+      // Fold transpose into transfer_read to enable in-flight transpose when
+      // converting to arm_sme.tile_load.
----------------
c-rhodes wrote:

good point, we should update this lowering to use an extra tile then rather than remove.

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


More information about the Mlir-commits mailing list