Skip to content

Commit 36d384b

Browse files
[mlir][Transforms][NFC] Dialect conversion: Simplify EraseBlockRewrite constructor (#99805)
1 parent 10d720b commit 36d384b

File tree

1 file changed

+4
-7
lines changed

1 file changed

+4
-7
lines changed

mlir/lib/Transforms/Utils/DialectConversion.cpp

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -293,10 +293,9 @@ class CreateBlockRewrite : public BlockRewrite {
293293
/// original location.
294294
class EraseBlockRewrite : public BlockRewrite {
295295
public:
296-
EraseBlockRewrite(ConversionPatternRewriterImpl &rewriterImpl, Block *block,
297-
Region *region, Block *insertBeforeBlock)
298-
: BlockRewrite(Kind::EraseBlock, rewriterImpl, block), region(region),
299-
insertBeforeBlock(insertBeforeBlock) {}
296+
EraseBlockRewrite(ConversionPatternRewriterImpl &rewriterImpl, Block *block)
297+
: BlockRewrite(Kind::EraseBlock, rewriterImpl, block),
298+
region(block->getParent()), insertBeforeBlock(block->getNextNode()) {}
300299

301300
static bool classof(const IRRewrite *rewrite) {
302301
return rewrite->getKind() == Kind::EraseBlock;
@@ -1440,9 +1439,7 @@ void ConversionPatternRewriterImpl::notifyOpReplaced(Operation *op,
14401439
}
14411440

14421441
void ConversionPatternRewriterImpl::notifyBlockIsBeingErased(Block *block) {
1443-
Region *region = block->getParent();
1444-
Block *origNextBlock = block->getNextNode();
1445-
appendRewrite<EraseBlockRewrite>(block, region, origNextBlock);
1442+
appendRewrite<EraseBlockRewrite>(block);
14461443
}
14471444

14481445
void ConversionPatternRewriterImpl::notifyBlockInserted(

0 commit comments

Comments
 (0)