Refactor ForStmt: having it contain a StmtBlock instead of subclassing

StmtBlock.  This is more consistent with IfStmt and also conceptually makes
more sense - a forstmt "isn't" its body, it contains its body.

This is step 1/N towards merging BasicBlock and StmtBlock.  This is required
because in the new regime StmtBlock will have a use list (just like BasicBlock
does) of operands, and ForStmt already has a use list for its induction
variable.

This is a mechanical patch, NFC.

PiperOrigin-RevId: 226684158

1 files changed, 1 insertions, 1 deletions

diff --git a/mlir/lib/Transforms/LowerVectorTransfers.cpp b/mlir/lib/Transforms/LowerVectorTransfers.cpp
index df30a779461..d4069eaa638 100644
--- a/mlir/lib/Transforms/LowerVectorTransfers.cpp
+++ b/mlir/lib/Transforms/LowerVectorTransfers.cpp
@@ -147,7 +147,7 @@ static void rewriteAsLoops(VectorTransferOpTy *transfer,
     auto *forStmt = b.createFor(transfer->getLoc(), 0, it.value());
     loops.insert(forStmt);
     // Setting the insertion point to the innermost loop achieves nesting.
-    b.setInsertionPointToStart(loops.back());
+    b.setInsertionPointToStart(loops.back()->getBody());
     if (composed == getAffineConstantExpr(0, b.getContext())) {
       transfer->emitWarning(
           "Redundant copy can be implemented as a vector broadcast");