forked from M-Labs/nac3
1
0
Fork 0

core: Switch to LLVM New Pass Manager

This commit is contained in:
David Mak 2023-09-11 13:14:35 +08:00 committed by David Mak
parent 676d07657a
commit 048fcb0a69
3 changed files with 35 additions and 37 deletions

View File

@ -8,7 +8,7 @@ use std::sync::Arc;
use inkwell::{ use inkwell::{
memory_buffer::MemoryBuffer, memory_buffer::MemoryBuffer,
module::{Linkage, Module}, module::{Linkage, Module},
passes::{PassManager, PassManagerBuilder}, passes::PassBuilderOptions,
targets::*, targets::*,
OptimizationLevel, OptimizationLevel,
}; };
@ -654,12 +654,16 @@ impl Nac3 {
global_option = global.get_next_global(); global_option = global.get_next_global();
} }
let builder = PassManagerBuilder::create(); let target_machine = self.llvm_options.target
builder.set_optimization_level(OptimizationLevel::Aggressive); .create_target_machine(self.llvm_options.opt_level)
let passes = PassManager::create(()); .expect("couldn't create target machine");
builder.set_inliner_with_threshold(255);
builder.populate_module_pass_manager(&passes); let pass_options = PassBuilderOptions::create();
passes.run_on(&main); pass_options.set_merge_functions(true);
let result = main.run_passes("default<O3>", &target_machine, pass_options);
if let Err(err) = result {
panic!("Failed to run optimization for module `main`: {}", err.to_string());
}
link_fn(&main) link_fn(&main)
} }

View File

@ -15,7 +15,7 @@ use inkwell::{
builder::Builder, builder::Builder,
context::Context, context::Context,
module::Module, module::Module,
passes::{PassManager, PassManagerBuilder}, passes::PassBuilderOptions,
targets::{CodeModel, RelocMode, Target, TargetMachine, TargetTriple}, targets::{CodeModel, RelocMode, Target, TargetMachine, TargetTriple},
types::{AnyType, BasicType, BasicTypeEnum}, types::{AnyType, BasicType, BasicTypeEnum},
values::{BasicValueEnum, FunctionValue, PhiValue, PointerValue}, values::{BasicValueEnum, FunctionValue, PhiValue, PointerValue},
@ -32,7 +32,6 @@ use std::sync::{
Arc, Arc,
}; };
use std::thread; use std::thread;
use lazy_static::lazy_static;
pub mod concrete_type; pub mod concrete_type;
pub mod expr; pub mod expr;
@ -54,12 +53,6 @@ pub struct StaticValueStore {
pub type VarValue<'ctx> = (PointerValue<'ctx>, Option<Arc<dyn StaticValue + Send + Sync>>, i64); pub type VarValue<'ctx> = (PointerValue<'ctx>, Option<Arc<dyn StaticValue + Send + Sync>>, i64);
lazy_static!(
// HACK: The Mutex is a work-around for issue
// https://git.m-labs.hk/M-Labs/nac3/issues/275
static ref PASSES_INIT_LOCK: Mutex<AtomicBool> = Mutex::new(AtomicBool::new(true));
);
/// Additional options for LLVM during codegen. /// Additional options for LLVM during codegen.
#[derive(Clone, Debug, Eq, PartialEq)] #[derive(Clone, Debug, Eq, PartialEq)]
pub struct CodeGenLLVMOptions { pub struct CodeGenLLVMOptions {
@ -305,23 +298,11 @@ impl WorkerRegistry {
context.i32_type().const_int(4, false), context.i32_type().const_int(4, false),
); );
let passes = PassManager::create(&module);
// HACK: This critical section is a work-around for issue
// https://git.m-labs.hk/M-Labs/nac3/issues/275
{
let _data = PASSES_INIT_LOCK.lock();
let pass_builder = PassManagerBuilder::create();
pass_builder.set_optimization_level(self.llvm_options.opt_level);
pass_builder.populate_function_pass_manager(&passes);
}
let mut errors = HashSet::new(); let mut errors = HashSet::new();
while let Some(task) = self.receiver.recv().unwrap() { while let Some(task) = self.receiver.recv().unwrap() {
match gen_func(&context, generator, self, builder, module, task) { match gen_func(&context, generator, self, builder, module, task) {
Ok(result) => { Ok(result) => {
builder = result.0; builder = result.0;
passes.run_on(&result.2);
module = result.1; module = result.1;
} }
Err((old_builder, e)) => { Err((old_builder, e)) => {
@ -345,9 +326,21 @@ impl WorkerRegistry {
panic!() panic!()
} }
let pass_options = PassBuilderOptions::create();
let target_machine = self.llvm_options.target.create_target_machine(
self.llvm_options.opt_level
).expect(format!("could not create target machine from properties {:?}", self.llvm_options.target).as_str());
let passes = format!("default<O{}>", self.llvm_options.opt_level as u32);
let result = module.run_passes(passes.as_str(), &target_machine, pass_options);
if let Err(err) = result {
panic!("Failed to run optimization for module `{}`: {}",
module.get_name().to_str().unwrap(),
err.to_string());
}
if self.llvm_options.emit_llvm { if self.llvm_options.emit_llvm {
println!("LLVM IR for {}", module.get_name().to_str().unwrap()); println!("LLVM IR for {}\n{}", module.get_name().to_str().unwrap(), module.to_string());
println!("{}", module.to_string());
println!(); println!();
} }

View File

@ -1,7 +1,7 @@
use clap::Parser; use clap::Parser;
use inkwell::{ use inkwell::{
memory_buffer::MemoryBuffer, memory_buffer::MemoryBuffer,
passes::{PassManager, PassManagerBuilder}, passes::PassBuilderOptions,
targets::*, targets::*,
OptimizationLevel, OptimizationLevel,
}; };
@ -325,16 +325,17 @@ fn main() {
function_iter = func.get_next_function(); function_iter = func.get_next_function();
} }
let builder = PassManagerBuilder::create();
builder.set_optimization_level(OptimizationLevel::Aggressive);
let passes = PassManager::create(());
builder.set_inliner_with_threshold(255);
builder.populate_module_pass_manager(&passes);
passes.run_on(&main);
let target_machine = llvm_options.target let target_machine = llvm_options.target
.create_target_machine(llvm_options.opt_level) .create_target_machine(llvm_options.opt_level)
.expect("couldn't create target machine"); .expect("couldn't create target machine");
let pass_options = PassBuilderOptions::create();
pass_options.set_merge_functions(true);
let result = main.run_passes("default<O3>", &target_machine, pass_options);
if let Err(err) = result {
panic!("Failed to run optimization for module `main`: {}", err.to_string());
}
target_machine target_machine
.write_to_file(&main, FileType::Object, Path::new("module.o")) .write_to_file(&main, FileType::Object, Path::new("module.o"))
.expect("couldn't write module to file"); .expect("couldn't write module to file");