From 5845c95ae5f27ef146de695ac45a2c30056029e8 Mon Sep 17 00:00:00 2001
From: James Newling <james.newling@gmail.com>
Date: Tue, 10 Dec 2024 11:52:07 -0800
Subject: [PATCH] Update
 compiler/plugins/target/AMD-AIE/iree-amd-aie/Transforms/AMDAIEFuseFillIntoForall.cpp

Co-authored-by: Vivian <vivian.zhang@amd.com>
---
 .../iree-amd-aie/Transforms/AMDAIEFuseFillIntoForall.cpp      | 4 +---
 1 file changed, 1 insertion(+), 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
index 3a515e08f..71e416021 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
@@ -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;