[Mlir-commits] [mlir] Fix build error from #72178 (PR #72905)

[llvmlistbot at llvm.org]

https://github.com/MaheshRavishankar updated https://github.com/llvm/llvm-project/pull/72905

>From b4b9e46d0dc47070e3e4ffd665c61c82d80b4183 Mon Sep 17 00:00:00 2001
From: MaheshRavishankar <mahesh at nod-labs.com>
Date: Mon, 20 Nov 2023 11:23:04 -0800
Subject: [PATCH] Fix build error from #72178

---
 .../SCF/Transforms/TileUsingInterface.cpp       | 17 ++++++++---------
 1 file changed, 8 insertions(+), 9 deletions(-)

diff --git a/mlir/lib/Dialect/SCF/Transforms/TileUsingInterface.cpp b/mlir/lib/Dialect/SCF/Transforms/TileUsingInterface.cpp
index b91af4d246d9919..8057b3898012d49 100644
--- a/mlir/lib/Dialect/SCF/Transforms/TileUsingInterface.cpp
+++ b/mlir/lib/Dialect/SCF/Transforms/TileUsingInterface.cpp
@@ -690,16 +690,15 @@ void mlir::scf::yieldReplacementForFusedProducer(
         rewriter.updateRootInPlace(tiledDestStyleOp, [&]() {
           tiledDestStyleOp.getDpsInitsMutable()[resultNumber].set(destSlice);
         });
-
-        Block *block = rewriter.getInsertionPoint()->getBlock();
-        rewriter.setInsertionPoint(block->getTerminator());
-        Value replacement = rewriter.create<tensor::InsertSliceOp>(
-            fusedProducerInfo.origProducer.getLoc(),
-            fusedProducerInfo.tiledAndFusedProducer,
-            loops.back().getRegionIterArgs().back(), sliceOp.getMixedOffsets(),
-            sliceOp.getMixedSizes(), sliceOp.getMixedStrides());
-        return {replacement};
       }
+      Block *block = rewriter.getInsertionPoint()->getBlock();
+      rewriter.setInsertionPoint(block->getTerminator());
+      Value replacement = rewriter.create<tensor::InsertSliceOp>(
+          fusedProducerInfo.origProducer.getLoc(),
+          fusedProducerInfo.tiledAndFusedProducer,
+          loops.back().getRegionIterArgs().back(), sliceOp.getMixedOffsets(),
+          sliceOp.getMixedSizes(), sliceOp.getMixedStrides());
+      return {replacement};
     };
 
     addInitOperandsToLoopNest(rewriter, loops,