1 files changed, 5 insertions, 5 deletions

diff --git a/mlir/lib/Transforms/DmaGeneration.cpp b/mlir/lib/Transforms/DmaGeneration.cpp
index dcb4828d0bf..bda98f46b61 100644
--- a/mlir/lib/Transforms/DmaGeneration.cpp
+++ b/mlir/lib/Transforms/DmaGeneration.cpp
@@ -338,7 +338,7 @@ bool DmaGeneration::generateDma(const MemRefRegion &region, Block *block,
     auto fastMemRefType = top.getMemRefType(
         fastBufferShape, memRefType.getElementType(), {}, fastMemorySpace);
 
-    // Create the fast memory space buffer just before the 'affine.for'
+    // Create the fast memory space buffer just before the 'for'
     // instruction.
     fastMemRef = prologue.create<AllocOp>(loc, fastMemRefType)->getResult();
     // Record it.
@@ -457,7 +457,7 @@ bool DmaGeneration::runOnBlock(Block *block, uint64_t consumedCapacityBytes) {
   // approach is conservative in some cases at the moment, we do a check later
   // and report an error with location info.
   // TODO(bondhugula): An 'affine.if' instruction is being treated similar to an
-  // operation instruction. 'affine.if''s could have 'affine.for's in them;
+  // operation instruction. 'affine.if''s could have 'for's in them;
   // treat them separately.
 
   // Get to the first load, store, or for op.
@@ -471,7 +471,7 @@ bool DmaGeneration::runOnBlock(Block *block, uint64_t consumedCapacityBytes) {
     if (auto forOp = it->dyn_cast<AffineForOp>()) {
       // We'll assume for now that loops with steps are tiled loops, and so DMAs
       // are not performed for that depth, but only further inside.
-      // If the memory footprint of the 'affine.for' loop is higher than fast
+      // If the memory footprint of the 'for' loop is higher than fast
       // memory capacity (when provided), we recurse to DMA at an inner level
       // until we find a depth at which footprint fits in the capacity. If the
       // footprint can't be calcuated, we assume for now it fits.
@@ -490,11 +490,11 @@ bool DmaGeneration::runOnBlock(Block *block, uint64_t consumedCapacityBytes) {
         consumedCapacityBytes += runOnBlock(/*begin=*/curBegin, /*end=*/it);
         // Recurse onto the body of this loop.
         runOnBlock(forOp->getBody(), consumedCapacityBytes);
-        // The next region starts right after the 'affine.for' instruction.
+        // The next region starts right after the 'for' instruction.
         curBegin = std::next(it);
       } else {
         // We have enough capacity, i.e., DMAs will be computed for the portion
-        // of the block until 'it', and for the 'affine.for' loop. For the
+        // of the block until 'it', and for the 'for' loop. For the
         // latter, they are placed just before this loop (for incoming DMAs) and
         // right after (for outgoing ones).
         consumedCapacityBytes += runOnBlock(/*begin=*/curBegin, /*end=*/it);