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 index 71e416021..ff0e5dec0 100644 --- 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(loc, scalar, bbArg); + auto fusedFill = rewriter.create(fillOp.getLoc(), fillOp.value(), bbArg); rewriter.replaceUsesWithIf( bbArg, fusedFill.getResult(0), [&](OpOperand &operand) { Operation *owner = operand.getOwner();