Skip to content

Commit 1ac831e

Browse files
authored
Add an option for openmp opt (#195)
1 parent 805f4a2 commit 1ac831e

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

tools/mlir-clang/mlir-clang.cc

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -74,6 +74,9 @@ static cl::opt<bool> Opt3("O3", cl::init(false), cl::desc("Opt level 3"));
7474
static cl::opt<bool> SCFOpenMP("scf-openmp", cl::init(true),
7575
cl::desc("Emit llvm"));
7676

77+
static cl::opt<bool> OpenMPOpt("openmp-opt", cl::init(true),
78+
cl::desc("Turn on openmp opt"));
79+
7780
static cl::opt<bool> ShowAST("show-ast", cl::init(false), cl::desc("Show AST"));
7881

7982
static cl::opt<bool> ImmediateMLIR("immediate", cl::init(false),
@@ -599,8 +602,10 @@ int main(int argc, char **argv) {
599602
pm2.addPass(createConvertSCFToOpenMPPass());
600603
}
601604
pm2.addPass(mlir::createCanonicalizerPass());
602-
pm2.addPass(polygeist::createOpenMPOptPass());
603-
pm2.addPass(mlir::createCanonicalizerPass());
605+
if (OpenMPOpt) {
606+
pm2.addPass(polygeist::createOpenMPOptPass());
607+
pm2.addPass(mlir::createCanonicalizerPass());
608+
}
604609
pm.nest<mlir::FuncOp>().addPass(polygeist::createMem2RegPass());
605610
pm2.addPass(mlir::createCSEPass());
606611
pm2.addPass(mlir::createCanonicalizerPass());

0 commit comments

Comments
 (0)