diff --git a/mlir/lib/Transforms/Utils/RegionUtils.cpp b/mlir/lib/Transforms/Utils/RegionUtils.cpp index dd75e283a36b8d..3c7523827699c8 100644 --- a/mlir/lib/Transforms/Utils/RegionUtils.cpp +++ b/mlir/lib/Transforms/Utils/RegionUtils.cpp @@ -1012,7 +1012,8 @@ static LogicalResult dropRedundantArguments(RewriterBase &rewriter, // Add any nested regions to the worklist. for (Block &block : *region) { - anyChanged = succeeded(dropRedundantArguments(rewriter, block)); + anyChanged = + succeeded(dropRedundantArguments(rewriter, block)) || anyChanged; for (Operation &op : block) for (Region &nestedRegion : op.getRegions())