[Mlir-commits] [mlir] [mlir][scf] Rewrite vector.transfer_read/write after peeling (PR #88684)
Matthias Springer
llvmlistbot at llvm.org
Sun Apr 14 23:22:27 PDT 2024
================
@@ -166,6 +167,52 @@ static LogicalResult peelForLoop(RewriterBase &b, ForOp forOp,
return success();
}
+static void rewriteVectorReadWriteToLoadStore(RewriterBase &b, Operation *op) {
+ b.setInsertionPoint(op);
+ if (auto write = dyn_cast<vector::TransferWriteOp>(op)) {
+ b.replaceOpWithNewOp<vector::StoreOp>(
+ op, write.getVector(), write.getSource(), write.getIndices());
+ } else if (auto read = dyn_cast<vector::TransferReadOp>(op)) {
+ b.replaceOpWithNewOp<vector::LoadOp>(op, read.getVectorType(),
+ read.getSource(), read.getIndices());
+ }
+}
+
+static bool hasVectorSizeEqualToStep(Operation *Op,
----------------
matthias-springer wrote:
variable names should be lower case
https://github.com/llvm/llvm-project/pull/88684
More information about the Mlir-commits
mailing list