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

…FuseFillIntoForall.cpp Co-authored-by: Vivian <[email protected]>
nod-ai · Dec 10, 2024 · 5845c95 · 5845c95
1 parent 98f610c
commit 5845c95
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
@@ -83,9 +83,7 @@ void AMDAIEFuseFillIntoForallPass::runOnOperation() {
   rewriter.replaceUsesWithIf(
       bbArg, fusedFill.getResult(0), [&](OpOperand &operand) {
         Operation *owner = operand.getOwner();
-        if (owner == fusedFill) {
-          return false;
-        } else if (isa<tensor::ParallelInsertSliceOp>(owner)) {
+        if (owner == fusedFill || isa<tensor::ParallelInsertSliceOp>(owner)) {
           return false;
         } else {
           return true;