maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

[flang][tco] Engineering option for running only CodeGen passes. 

This option allows running only CodeGen passes and then translating
FIR to LLVM IR. I am using it to fetch optimized FIR after bbc,
hand-modify it and then produce LLVM IR that can be fed to clang.
This commit is contained in:
Slava Zakharin 2022-10-03 16:54:12 -07:00
parent 4769976c49
commit 42fead6834
2 changed files with 26 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
flang/test/Driver/tco-code-gen-llvm.fir Normal file
View File

@ -0,0 +1,13 @@
// Test the tco's -code-gen-llvm option.
// RUN: tco -code-gen-llvm %s 2>&1 | FileCheck %s
// Check that FIR is translated into LLVM IR, and that
// there is no any FIR output.
// CHECK-NOT: func.func
// CHECK: define void @_QPfoo
// CHECK-NOT: func.func
func.func @_QPfoo() {
return
}

16
flang/tools/tco/tco.cpp
View File

@ -53,6 +53,11 @@ static cl::opt<std::string> targetTriple("target",
cl::desc("specify a target triple"),
cl::init("native"));
static cl::opt<bool> codeGenLLVM(
"code-gen-llvm",
cl::desc("Run only CodeGen passes and translate FIR to LLVM IR"),
cl::init(false));
#include "flang/Tools/CLOptions.inc"
static void printModuleBody(mlir::ModuleOp mod, raw_ostream &output) {
@ -112,15 +117,20 @@ compileFIR(const mlir::PassPipelineCLParser &passPipeline) {
if (mlir::failed(passPipeline.addToPipeline(pm, errorHandler)))
return mlir::failure();
} else {
// Run tco with O2 by default.
fir::createMLIRToLLVMPassPipeline(pm, llvm::OptimizationLevel::O2);
if (codeGenLLVM) {
// Run only CodeGen passes.
fir::createDefaultFIRCodeGenPassPipeline(pm);
} else {
// Run tco with O2 by default.
fir::createMLIRToLLVMPassPipeline(pm, llvm::OptimizationLevel::O2);
}
fir::addLLVMDialectToLLVMPass(pm, out.os());
}
// run the pass manager
if (mlir::succeeded(pm.run(*owningRef))) {
// passes ran successfully, so keep the output
if (emitFir || passPipeline.hasAnyOccurrences())
if ((emitFir || passPipeline.hasAnyOccurrences()) && !codeGenLLVM)
printModuleBody(*owningRef, out.os());
out.keep();
return mlir::success();