diff --git a/mlir/lib/Transforms/Utils/DialectConversion.cpp b/mlir/lib/Transforms/Utils/DialectConversion.cpp index 9cb6f2ba1eaae..7cfe7250d02c3 100644 --- a/mlir/lib/Transforms/Utils/DialectConversion.cpp +++ b/mlir/lib/Transforms/Utils/DialectConversion.cpp @@ -2301,7 +2301,7 @@ LogicalResult OperationLegalizer::legalizePatternBlockRewrites( continue; Block *block = rewrite->getBlock(); if (isa(rewrite)) + ReplaceBlockArgRewrite, InlineBlockRewrite>(rewrite)) continue; // Only check blocks outside of the current operation. Operation *parentOp = block->getParentOp();