Update compiler/plugins/target/AMD-AIE/iree-amd-aie/Transforms/AMDAIE…

…FuseFillIntoForall.cpp Co-authored-by: Vivian <[email protected]>
nod-ai · Dec 10, 2024 · 6b856b1 · 6b856b1
1 parent 5845c95
commit 6b856b1
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/compiler/plugins/target/AMD-AIE/iree-amd-aie/Transforms/AMDAIEFuseFillIntoForall.cpp b/compiler/plugins/target/AMD-AIE/iree-amd-aie/Transforms/AMDAIEFuseFillIntoForall.cpp
@@ -77,9 +77,7 @@ void AMDAIEFuseFillIntoForallPass::runOnOperation() {
   // fill at the beginning of the forall body.
   assert(!extractSliceOp);
   rewriter.setInsertionPointToStart(forallOp.getBody());
-  Value scalar = fillOp.value();
-  Location loc = fillOp.getLoc();
-  auto fusedFill = rewriter.create<linalg::FillOp>(loc, scalar, bbArg);
+  auto fusedFill = rewriter.create<linalg::FillOp>(fillOp.getLoc(), fillOp.value(), bbArg);
   rewriter.replaceUsesWithIf(
       bbArg, fusedFill.getResult(0), [&](OpOperand &operand) {
         Operation *owner = operand.getOwner();