Skip to content

Commit 4eef5d2

Browse files
committed
Remove runOn
1 parent 55614ce commit 4eef5d2

File tree

1 file changed

+5
-9
lines changed

1 file changed

+5
-9
lines changed

flang/lib/Optimizer/CodeGen/PreCGRewrite.cpp

Lines changed: 5 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -283,7 +283,9 @@ class DeclareOpConversion : public mlir::OpRewritePattern<fir::DeclareOp> {
283283

284284
class CodeGenRewrite : public fir::impl::CodeGenRewriteBase<CodeGenRewrite> {
285285
public:
286-
void runOn(mlir::Operation *op) {
286+
void runOnOperation() override final {
287+
mlir::ModuleOp mod = getOperation();
288+
287289
auto &context = getContext();
288290
mlir::ConversionTarget target(context);
289291
target.addLegalDialect<mlir::arith::ArithDialect, fir::FIROpsDialect,
@@ -300,21 +302,15 @@ class CodeGenRewrite : public fir::impl::CodeGenRewriteBase<CodeGenRewrite> {
300302
mlir::RewritePatternSet patterns(&context);
301303
fir::populatePreCGRewritePatterns(patterns);
302304
if (mlir::failed(
303-
mlir::applyPartialConversion(op, target, std::move(patterns)))) {
305+
mlir::applyPartialConversion(mod, target, std::move(patterns)))) {
304306
mlir::emitError(mlir::UnknownLoc::get(&context),
305307
"error in running the pre-codegen conversions");
306308
signalPassFailure();
307309
return;
308310
}
309311
// Erase any residual (fir.shape, fir.slice...).
310312
mlir::IRRewriter rewriter(&context);
311-
(void)mlir::runRegionDCE(rewriter, op->getRegions());
312-
}
313-
314-
void runOnOperation() override final {
315-
// Call runOn on all top level regions that may contain emboxOp/arrayCoorOp.
316-
mlir::ModuleOp mod = getOperation();
317-
runOn(mod);
313+
(void)mlir::runRegionDCE(rewriter, mod->getRegions());
318314
}
319315
};
320316

0 commit comments

Comments
 (0)