GitLab will continue to be upgraded from 11.4.5-ce.0 on November 25th 2019 at 4.00pm (AEDT) to 5.00pm (AEDT) due to Critical Security Patch Availability. During the update, GitLab and Mattermost services will not be available.

Commit e13dc2a0 authored by qinsoon's avatar qinsoon

[wip] simple test for preserving caller saved regsiter

parent a95edaa6
......@@ -358,7 +358,7 @@ macro_rules! inst {
};
// EXPRCALL
(($vm: expr, $fv: ident) $name: ident: $res: ident = EXPRCALL ($cc: expr, is_abort: $is_abort: expr) $func: ident ($($val: ident), +)) => {
(($vm: expr, $fv: ident) $name: ident: $res: ident = EXPRCALL ($cc: expr, is_abort: $is_abort: expr) $func: ident ($($val: ident), *)) => {
let ops = vec![$func.clone(), $($val.clone()), *];
let ops_len = ops.len();
let $name = $fv.new_inst(Instruction{
......@@ -375,7 +375,7 @@ macro_rules! inst {
}
});
};
(($vm: expr, $fv: ident) $name: ident: EXPRCALL ($cc: expr, is_abort: $is_abort: expr) $func: ident ($($val: ident), +)) => {
(($vm: expr, $fv: ident) $name: ident: EXPRCALL ($cc: expr, is_abort: $is_abort: expr) $func: ident ($($val: ident), *)) => {
let ops = vec![$func.clone(), $($val.clone()), *];
let ops_len = ops.len();
let $name = $fv.new_inst(Instruction{
......
......@@ -4,6 +4,7 @@ extern crate libloading;
use mu::testutil;
use mu::testutil::aot;
use mu::utils::LinkedHashMap;
use test_compiler::test_call::gen_ccall_exit;
use test_ir::test_ir::factorial;
use self::mu::compiler::*;
use self::mu::utils::vec_utils;
......@@ -847,3 +848,211 @@ fn coalesce_branch2_moves() -> VM {
vm
}
#[test]
fn test_preserve_caller_saved() {
VM::start_logging_trace();
let vm = Arc::new(preserve_caller_saved());
let compiler = Compiler::new(CompilerPolicy::default(), &vm);
let func_foo = vm.id_of("foo");
let func_preserve_caller_saved = vm.id_of("preserve_caller_saved");
{
let funcs = vm.funcs().read().unwrap();
let func_vers = vm.func_vers().read().unwrap();
{
let func = funcs.get(&func_foo).unwrap().read().unwrap();
let mut func_ver = func_vers.get(&func.cur_ver.unwrap()).unwrap().write().unwrap();
compiler.compile(&mut func_ver);
}
{
let func = funcs.get(&func_preserve_caller_saved).unwrap().read().unwrap();
let mut func_ver = func_vers.get(&func.cur_ver.unwrap()).unwrap().write().unwrap();
compiler.compile(&mut func_ver);
}
}
vm.make_primordial_thread(func_preserve_caller_saved, true, vec![]);
backend::emit_context(&vm);
let executable = aot::link_primordial(vec![Mu("foo"), Mu("preserve_caller_saved")], "test_preserve_caller_saved", &vm);
let output = aot::execute_nocheck(executable);
// add from 0 to 9
assert!(output.status.code().is_some());
assert_eq!(output.status.code().unwrap(), 45);
}
fn preserve_caller_saved() -> VM {
let vm = VM::new_with_opts("init_mu --disable-inline");
create_empty_func_foo(&vm);
typedef! ((vm) int64 = mu_int(64));
constdef! ((vm) <int64> int64_0 = Constant::Int(0));
constdef! ((vm) <int64> int64_1 = Constant::Int(1));
constdef! ((vm) <int64> int64_2 = Constant::Int(2));
constdef! ((vm) <int64> int64_3 = Constant::Int(3));
constdef! ((vm) <int64> int64_4 = Constant::Int(4));
constdef! ((vm) <int64> int64_5 = Constant::Int(5));
constdef! ((vm) <int64> int64_6 = Constant::Int(6));
constdef! ((vm) <int64> int64_7 = Constant::Int(7));
constdef! ((vm) <int64> int64_8 = Constant::Int(8));
constdef! ((vm) <int64> int64_9 = Constant::Int(9));
funcsig! ((vm) sig = () -> ());
funcdecl! ((vm) <sig> preserve_caller_saved);
funcdef! ((vm) <sig> preserve_caller_saved VERSION preserve_caller_saved_v1);
// blk entry
block! ((vm, preserve_caller_saved_v1) blk_entry);
block! ((vm, preserve_caller_saved_v1) blk_main);
consta! ((vm, preserve_caller_saved_v1) int64_0_local = int64_0);
consta! ((vm, preserve_caller_saved_v1) int64_1_local = int64_1);
consta! ((vm, preserve_caller_saved_v1) int64_2_local = int64_2);
consta! ((vm, preserve_caller_saved_v1) int64_3_local = int64_3);
consta! ((vm, preserve_caller_saved_v1) int64_4_local = int64_4);
consta! ((vm, preserve_caller_saved_v1) int64_5_local = int64_5);
consta! ((vm, preserve_caller_saved_v1) int64_6_local = int64_6);
consta! ((vm, preserve_caller_saved_v1) int64_7_local = int64_7);
consta! ((vm, preserve_caller_saved_v1) int64_8_local = int64_8);
consta! ((vm, preserve_caller_saved_v1) int64_9_local = int64_9);
inst! ((vm, preserve_caller_saved_v1) blk_entry_branch:
BRANCH blk_main (
int64_0_local,
int64_1_local,
int64_2_local,
int64_3_local,
int64_4_local,
int64_5_local,
int64_6_local,
int64_7_local,
int64_8_local,
int64_9_local
)
);
define_block! ((vm, preserve_caller_saved_v1) blk_entry() {
blk_entry_branch
});
// blk main
ssa! ((vm, preserve_caller_saved_v1) <int64> v0);
ssa! ((vm, preserve_caller_saved_v1) <int64> v1);
ssa! ((vm, preserve_caller_saved_v1) <int64> v2);
ssa! ((vm, preserve_caller_saved_v1) <int64> v3);
ssa! ((vm, preserve_caller_saved_v1) <int64> v4);
ssa! ((vm, preserve_caller_saved_v1) <int64> v5);
ssa! ((vm, preserve_caller_saved_v1) <int64> v6);
ssa! ((vm, preserve_caller_saved_v1) <int64> v7);
ssa! ((vm, preserve_caller_saved_v1) <int64> v8);
ssa! ((vm, preserve_caller_saved_v1) <int64> v9);
let foo_sig = vm.get_func_sig(vm.id_of("foo_sig"));
let foo_id = vm.id_of("foo");
typedef! ((vm) type_funcref_foo = mu_funcref(foo_sig));
constdef! ((vm) <type_funcref_foo> const_funcref_foo = Constant::FuncRef(foo_id));
consta! ((vm, preserve_caller_saved_v1) const_funcref_foo_local = const_funcref_foo);
inst! ((vm, preserve_caller_saved_v1) blk_main_call:
EXPRCALL (CallConvention::Mu, is_abort: false) const_funcref_foo_local ()
);
ssa! ((vm, preserve_caller_saved_v1) <int64> res1);
inst! ((vm, preserve_caller_saved_v1) blk_main_add1:
res1 = BINOP (BinOp::Add) v0 v1
);
ssa! ((vm, preserve_caller_saved_v1) <int64> res2);
inst! ((vm, preserve_caller_saved_v1) blk_main_add2:
res2 = BINOP (BinOp::Add) res1 v2
);
ssa! ((vm, preserve_caller_saved_v1) <int64> res3);
inst! ((vm, preserve_caller_saved_v1) blk_main_add3:
res3 = BINOP (BinOp::Add) res2 v3
);
ssa! ((vm, preserve_caller_saved_v1) <int64> res4);
inst! ((vm, preserve_caller_saved_v1) blk_main_add4:
res4 = BINOP (BinOp::Add) res3 v4
);
ssa! ((vm, preserve_caller_saved_v1) <int64> res5);
inst! ((vm, preserve_caller_saved_v1) blk_main_add5:
res5 = BINOP (BinOp::Add) res4 v5
);
ssa! ((vm, preserve_caller_saved_v1) <int64> res6);
inst! ((vm, preserve_caller_saved_v1) blk_main_add6:
res6 = BINOP (BinOp::Add) res5 v6
);
ssa! ((vm, preserve_caller_saved_v1) <int64> res7);
inst! ((vm, preserve_caller_saved_v1) blk_main_add7:
res7 = BINOP (BinOp::Add) res6 v7
);
ssa! ((vm, preserve_caller_saved_v1) <int64> res8);
inst! ((vm, preserve_caller_saved_v1) blk_main_add8:
res8 = BINOP (BinOp::Add) res7 v8
);
ssa! ((vm, preserve_caller_saved_v1) <int64> res9);
inst! ((vm, preserve_caller_saved_v1) blk_main_add9:
res9 = BINOP (BinOp::Add) res8 v9
);
let blk_main_exit = gen_ccall_exit(res9.clone(), &mut preserve_caller_saved_v1, &vm);
inst! ((vm, preserve_caller_saved_v1) blk_main_ret:
RET
);
define_block! ((vm, preserve_caller_saved_v1) blk_main(v0, v1, v2, v3, v4, v5, v6, v7, v8, v9) {
blk_main_call,
blk_main_add1,
blk_main_add2,
blk_main_add3,
blk_main_add4,
blk_main_add5,
blk_main_add6,
blk_main_add7,
blk_main_add8,
blk_main_add9,
blk_main_exit,
blk_main_ret
});
define_func_ver!((vm) preserve_caller_saved_v1 (entry: blk_entry) {
blk_entry,
blk_main
});
vm
}
fn create_empty_func_foo(vm: &VM) {
funcsig! ((vm) foo_sig = () -> ());
funcdecl! ((vm) <foo_sig> foo);
funcdef! ((vm) <foo_sig> foo VERSION foo_v1);
block! ((vm, foo_v1) blk_entry);
inst! ((vm, foo_v1) blk_entry_ret:
RET
);
define_block! ((vm, foo_v1) blk_entry() {
blk_entry_ret
});
define_func_ver!((vm) foo_v1 (entry: blk_entry) {blk_entry});
}
\ No newline at end of file
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment