Commit 0ee10a59 authored by Nguyen, Thien Minh's avatar Nguyen, Thien Minh
Browse files

Enable opt level options for mlir tool 



Signed-off-by: default avatarThien Nguyen <nguyentm@ornl.gov>
parent 32102bba

mlir/tools/qcor-mlir-tool.cpp

+16 −1
Original line number Diff line number Diff line
@@ -39,6 +39,18 @@ cl::opt<bool> mlir_quantum_opt( 
cl::opt<std::string> mlir_specified_func_name(

    "internal-func-name", cl::desc("qcor provided function name"));



static cl::opt<bool>

    OptLevelO0("O0", cl::desc("Optimization level 0. Similar to clang -O0. "));



static cl::opt<bool>

    OptLevelO1("O1", cl::desc("Optimization level 1. Similar to clang -O1. "));



static cl::opt<bool>

    OptLevelO2("O2", cl::desc("Optimization level 2. Similar to clang -O2. "));



static cl::opt<bool>

    OptLevelO3("O3", cl::desc("Optimization level 3. Similar to clang -O3. "));



namespace {

enum Action { None, DumpMLIR, DumpMLIRLLVM, DumpLLVMIR };

}
@@ -118,7 +130,10 @@ int main(int argc, char **argv) { 
  // Optimize the LLVM IR

  llvm::InitializeNativeTarget();

  llvm::InitializeNativeTargetAsmPrinter();

  auto optPipeline = mlir::makeOptimizingTransformer(3, 0, nullptr);

  const unsigned optLevel =

      OptLevelO3 ? 3 : (OptLevelO2 ? 2 : (OptLevelO1 ? 1 : 0));

  // std::cout << "Opt level: " << optLevel << "\n";

  auto optPipeline = mlir::makeOptimizingTransformer(optLevel, 0, nullptr);

  if (auto err = optPipeline(llvmModule.get())) {

    llvm::errs() << "Failed to optimize LLVM IR " << err << "\n";

    return -1;