language-servers/roc
1
0
Fork 0
mirror of https://github.com/roc-lang/roc.git synced 2025-09-29 23:04:49 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 7

Allocate stack memory to local variables

Browse source
This commit is contained in:
Brian Carroll 2021-09-18 13:39:38 +01:00
parent 8afcc7ab4d
commit c3b5ac6c82
4 changed files with 167 additions and 58 deletions
Download patch file Download diff file Expand all files Collapse all files

34
compiler/gen_wasm/src/lib.rs
View file

@ -25,7 +25,7 @@ pub const ALIGN_8: u32 = 3;
pub const STACK_POINTER_GLOBAL_ID: u32 = 0;
#[derive(Clone, Copy, Debug)]
pub struct LocalId(u32);
pub struct LocalId(pub u32);
pub struct Env<'a> {
pub arena: &'a Bump, // not really using this much, parity_wasm works with std::vec a lot
@ -153,22 +153,42 @@ fn copy_memory(
Ok(())
}
pub fn allocate_stack_frame(instructions: &mut Vec<Instruction>, size: i32) {
pub fn allocate_stack_frame(
instructions: &mut Vec<Instruction>,
size: i32,
local_frame_pointer: Option<LocalId>,
) {
if size == 0 {
return;
}
instructions.push(GetGlobal(STACK_POINTER_GLOBAL_ID));
instructions.push(I32Const(size));
instructions.push(I32Sub);
if let Some(LocalId(local_index)) = local_frame_pointer {
instructions.push(TeeLocal(local_index));
}
instructions.push(SetGlobal(STACK_POINTER_GLOBAL_ID));
}
pub fn free_stack_frame(instructions: &mut Vec<Instruction>, size: i32) {
pub fn free_stack_frame(
instructions: &mut Vec<Instruction>,
size: i32,
local_frame_pointer: Option<LocalId>,
) {
if size == 0 {
return;
}
instructions.push(GetGlobal(STACK_POINTER_GLOBAL_ID));
instructions.push(I32Const(size));
instructions.push(I32Add);
instructions.push(SetGlobal(STACK_POINTER_GLOBAL_ID));
let get_stack_frame =
if let Some(LocalId(local_index)) = local_frame_pointer {
GetLocal(local_index)
} else {
GetGlobal(STACK_POINTER_GLOBAL_ID)
};
instructions.extend([
get_stack_frame,
I32Const(size),
I32Add,
SetGlobal(STACK_POINTER_GLOBAL_ID),
]);
}