language-servers/roc
1
0
Fork 0
mirror of https://github.com/roc-lang/roc.git synced 2025-10-01 15:51:12 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

Have cli accept an --optimize flag and use it

Browse source
This commit is contained in:
Richard Feldman 2020-04-16 20:26:37 -04:00
parent ea713023e8
commit d2e242ccbe
1 changed files with 23 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

35
cli/src/main.rs
View file

@ -10,7 +10,7 @@ use inkwell::OptimizationLevel;
use roc_collections::all::ImMap;
use roc_collections::all::MutMap;
use roc_gen::llvm::build::{
build_proc, build_proc_header, get_call_conventions, module_from_builtins,
build_proc, build_proc_header, get_call_conventions, module_from_builtins, OptLevel,
};
use roc_gen::llvm::convert::basic_type_from_layout;
use roc_load::file::{LoadedModule, LoadingProblem};
@ -33,6 +33,13 @@ fn main() -> io::Result<()> {
match argv.get(1) {
Some(filename) => {
let opt_level = match argv.get(1).map(String::as_str) {
Some("--optimize") => OptLevel::Optimize,
None => OptLevel::Normal,
Some(other) => {
panic!("Unrecognized CLI argument: {}", other);
}
};
let path = Path::new(filename);
let src_dir = path.parent().unwrap().canonicalize().unwrap();
@ -45,7 +52,7 @@ fn main() -> io::Result<()> {
.expect("Error spawning initial compiler thread."); // TODO make this error nicer.
// Spawn the root task
let loaded = rt.block_on(load_file(src_dir, path.canonicalize().unwrap()));
let loaded = rt.block_on(load_file(src_dir, path.canonicalize().unwrap(), opt_level));
loaded.expect("TODO gracefully handle LoadingProblem");
@ -59,23 +66,26 @@ fn main() -> io::Result<()> {
}
}
async fn load_file(src_dir: PathBuf, filename: PathBuf) -> Result<(), LoadingProblem> {
async fn load_file(
src_dir: PathBuf,
filename: PathBuf,
opt_level: OptLevel,
) -> Result<(), LoadingProblem> {
let compilation_start = SystemTime::now();
let arena = Bump::new();
// Step 1: compile the app and generate the .o file
let subs_by_module = MutMap::default();
let loaded = roc_load::file::load(
&roc_builtins::std::standard_stdlib(),
src_dir,
filename.clone(),
subs_by_module,
)
.await?;
// Release builds use uniqueness optimizations
let stdlib = match opt_level {
OptLevel::Normal => roc_builtins::std::standard_stdlib(),
OptLevel::Optimize => roc_builtins::unique::uniq_stdlib(),
};
let loaded = roc_load::file::load(&stdlib, src_dir, filename.clone(), subs_by_module).await?;
let dest_filename = filename.with_extension("o");
gen(&arena, loaded, filename, Triple::host(), &dest_filename);
gen(&arena, loaded, filename, Triple::host(), &dest_filename, opt_level);
let compilation_end = compilation_start.elapsed().unwrap();
@ -150,6 +160,7 @@ fn gen(
filename: PathBuf,
target: Triple,
dest_filename: &Path,
opt_level: OptLevel
) {
use roc_reporting::report::{can_problem, type_problem, RocDocAllocator, DEFAULT_PALETTE};
@ -243,7 +254,7 @@ fn gen(
let builder = context.create_builder();
let fpm = PassManager::create(&module);
roc_gen::llvm::build::add_passes(&fpm);
roc_gen::llvm::build::add_passes(&fpm, opt_level);
fpm.initialize();