832666c45e
* Manually rename BasicBlock to BlockPredecessor BasicBlock is a pair of (Ebb, Inst) that is used to represent the basic block subcomponent of an Ebb that is a predecessor to an Ebb. Eventually we will be able to remove this struct, but for now it makes sense to give it a non-conflicting name so that we can start to transition Ebb to represent a basic block. I have not updated any comments that refer to BasicBlock, as eventually we will remove BlockPredecessor and replace with Block, which is a basic block, so the comments will become correct. * Manually rename SSABuilder block types to avoid conflict SSABuilder has its own Block and BlockData types. These along with associated identifier will cause conflicts in a later commit, so they are renamed to be more verbose here. * Automatically rename 'Ebb' to 'Block' in *.rs * Automatically rename 'EBB' to 'block' in *.rs * Automatically rename 'ebb' to 'block' in *.rs * Automatically rename 'extended basic block' to 'basic block' in *.rs * Automatically rename 'an basic block' to 'a basic block' in *.rs * Manually update comment for `Block` `Block`'s wikipedia article required an update. * Automatically rename 'an `Block`' to 'a `Block`' in *.rs * Automatically rename 'extended_basic_block' to 'basic_block' in *.rs * Automatically rename 'ebb' to 'block' in *.clif * Manually rename clif constant that contains 'ebb' as substring to avoid conflict * Automatically rename filecheck uses of 'EBB' to 'BB' 'regex: EBB' -> 'regex: BB' '$EBB' -> '$BB' * Automatically rename 'EBB' 'Ebb' to 'block' in *.clif * Automatically rename 'an block' to 'a block' in *.clif * Fix broken testcase when function name length increases Test function names are limited to 16 characters. This causes the new longer name to be truncated and fail a filecheck test. An outdated comment was also fixed.
80 lines
2.6 KiB
Rust
80 lines
2.6 KiB
Rust
use cranelift::prelude::*;
|
|
use cranelift_module::{default_libcall_names, Linkage, Module};
|
|
use cranelift_simplejit::{SimpleJITBackend, SimpleJITBuilder};
|
|
use std::mem;
|
|
|
|
fn main() {
|
|
let mut module: Module<SimpleJITBackend> =
|
|
Module::new(SimpleJITBuilder::new(default_libcall_names()));
|
|
let mut ctx = module.make_context();
|
|
let mut func_ctx = FunctionBuilderContext::new();
|
|
|
|
let mut sig_a = module.make_signature();
|
|
sig_a.params.push(AbiParam::new(types::I32));
|
|
sig_a.returns.push(AbiParam::new(types::I32));
|
|
|
|
let mut sig_b = module.make_signature();
|
|
sig_b.returns.push(AbiParam::new(types::I32));
|
|
|
|
let func_a = module
|
|
.declare_function("a", Linkage::Local, &sig_a)
|
|
.unwrap();
|
|
let func_b = module
|
|
.declare_function("b", Linkage::Local, &sig_b)
|
|
.unwrap();
|
|
|
|
ctx.func.signature = sig_a;
|
|
ctx.func.name = ExternalName::user(0, func_a.as_u32());
|
|
{
|
|
let mut bcx: FunctionBuilder = FunctionBuilder::new(&mut ctx.func, &mut func_ctx);
|
|
let block = bcx.create_block();
|
|
|
|
bcx.switch_to_block(block);
|
|
bcx.append_block_params_for_function_params(block);
|
|
let param = bcx.block_params(block)[0];
|
|
let cst = bcx.ins().iconst(types::I32, 37);
|
|
let add = bcx.ins().iadd(cst, param);
|
|
bcx.ins().return_(&[add]);
|
|
bcx.seal_all_blocks();
|
|
bcx.finalize();
|
|
}
|
|
module.define_function(func_a, &mut ctx).unwrap();
|
|
module.clear_context(&mut ctx);
|
|
|
|
ctx.func.signature = sig_b;
|
|
ctx.func.name = ExternalName::user(0, func_b.as_u32());
|
|
{
|
|
let mut bcx: FunctionBuilder = FunctionBuilder::new(&mut ctx.func, &mut func_ctx);
|
|
let block = bcx.create_block();
|
|
|
|
bcx.switch_to_block(block);
|
|
let local_func = module.declare_func_in_func(func_a, &mut bcx.func);
|
|
let arg = bcx.ins().iconst(types::I32, 5);
|
|
let call = bcx.ins().call(local_func, &[arg]);
|
|
let value = {
|
|
let results = bcx.inst_results(call);
|
|
assert_eq!(results.len(), 1);
|
|
results[0].clone()
|
|
};
|
|
bcx.ins().return_(&[value]);
|
|
bcx.seal_all_blocks();
|
|
bcx.finalize();
|
|
}
|
|
module.define_function(func_b, &mut ctx).unwrap();
|
|
module.clear_context(&mut ctx);
|
|
|
|
// Perform linking.
|
|
module.finalize_definitions();
|
|
|
|
// Get a raw pointer to the generated code.
|
|
let code_b = module.get_finalized_function(func_b);
|
|
|
|
// Cast it to a rust function pointer type.
|
|
let ptr_b = unsafe { mem::transmute::<_, fn() -> u32>(code_b) };
|
|
|
|
// Call it!
|
|
let res = ptr_b();
|
|
|
|
assert_eq!(res, 42);
|
|
}
|