97d9f77d94
* Add `precise_output` argument to `test optimise`. Also allow optimise tests to be updated by `CRANELIFT_TEST_BLESS=1` * Move `check_precise_output` and `update_test` to `subtest`
82 lines
2.6 KiB
Rust
82 lines
2.6 KiB
Rust
//! Test command for testing the code generator pipeline
|
|
//!
|
|
//! The `compile` test command runs each function through the full code generator pipeline
|
|
|
|
use crate::subtest::{check_precise_output, run_filecheck, Context, SubTest};
|
|
use anyhow::Result;
|
|
use cranelift_codegen::ir;
|
|
use cranelift_reader::{TestCommand, TestOption};
|
|
use log::info;
|
|
use std::borrow::Cow;
|
|
|
|
struct TestCompile {
|
|
/// Flag indicating that the text expectation, comments after the function,
|
|
/// must be a precise 100% match on the compiled output of the function.
|
|
/// This test assertion is also automatically-update-able to allow tweaking
|
|
/// the code generator and easily updating all affected tests.
|
|
precise_output: bool,
|
|
}
|
|
|
|
pub fn subtest(parsed: &TestCommand) -> Result<Box<dyn SubTest>> {
|
|
assert_eq!(parsed.command, "compile");
|
|
let mut test = TestCompile {
|
|
precise_output: false,
|
|
};
|
|
for option in parsed.options.iter() {
|
|
match option {
|
|
TestOption::Flag("precise-output") => test.precise_output = true,
|
|
_ => anyhow::bail!("unknown option on {}", parsed),
|
|
}
|
|
}
|
|
Ok(Box::new(test))
|
|
}
|
|
|
|
impl SubTest for TestCompile {
|
|
fn name(&self) -> &'static str {
|
|
"compile"
|
|
}
|
|
|
|
fn is_mutating(&self) -> bool {
|
|
true
|
|
}
|
|
|
|
fn needs_isa(&self) -> bool {
|
|
true
|
|
}
|
|
|
|
fn run(&self, func: Cow<ir::Function>, context: &Context) -> Result<()> {
|
|
let isa = context.isa.expect("compile needs an ISA");
|
|
let params = func.params.clone();
|
|
let mut comp_ctx = cranelift_codegen::Context::for_function(func.into_owned());
|
|
|
|
// With `MachBackend`s, we need to explicitly request dissassembly results.
|
|
comp_ctx.set_disasm(true);
|
|
|
|
let compiled_code = comp_ctx
|
|
.compile(isa)
|
|
.map_err(|e| crate::pretty_anyhow_error(&e.func, e.inner))?;
|
|
let total_size = compiled_code.code_info().total_size;
|
|
|
|
let vcode = compiled_code.vcode.as_ref().unwrap();
|
|
|
|
info!("Generated {} bytes of code:\n{}", total_size, vcode);
|
|
|
|
if self.precise_output {
|
|
let cs = isa
|
|
.to_capstone()
|
|
.map_err(|e| anyhow::format_err!("{}", e))?;
|
|
let dis = compiled_code.disassemble(Some(¶ms), &cs)?;
|
|
|
|
let actual = Vec::from_iter(
|
|
std::iter::once("VCode:")
|
|
.chain(compiled_code.vcode.as_ref().unwrap().lines())
|
|
.chain(["", "Disassembled:"])
|
|
.chain(dis.lines()),
|
|
);
|
|
check_precise_output(&actual, context)
|
|
} else {
|
|
run_filecheck(&vcode, context)
|
|
}
|
|
}
|
|
}
|