language-servers/roc
1
0
Fork 0
mirror of https://github.com/roc-lang/roc.git synced 2025-10-02 16:21:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

Integrate refcount proc generator with Wasm backend

Browse source
This commit is contained in:
Brian Carroll 2021-11-27 09:13:50 +00:00
parent 61575cea7e
commit 2ad032f894
5 changed files with 60 additions and 55 deletions
Download patch file Download diff file Expand all files Collapse all files

2
compiler/gen_wasm/src/backend.rs
View file

@ -473,7 +473,7 @@ impl<'a> WasmBackend<'a> {
let (rc_stmt, new_proc_info) = self
.refcount_proc_gen
.call_refcount_proc(layout, modify, *following);
.expand_refcount_stmt_to_proc_call(layout, modify, *following);
// If we're creating a new RC procedure, we need to store its symbol data,
// so that we can correctly generate calls to it.

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

@ -36,7 +36,7 @@ pub struct Env<'a> {
}
pub fn build_module<'a>(
env: &'a mut Env<'a>,
env: &'a Env<'a>,
procedures: MutMap<(Symbol, ProcLayout<'a>), Proc<'a>>,
refcount_home: ModuleId,
) -> Result<std::vec::Vec<u8>, String> {
@ -47,7 +47,7 @@ pub fn build_module<'a>(
}
pub fn build_module_help<'a>(
env: &'a mut Env<'a>,
env: &'a Env<'a>,
procedures: MutMap<(Symbol, ProcLayout<'a>), Proc<'a>>,
refcount_home: ModuleId,
) -> Result<(WasmModule<'a>, u32), String> {
@ -100,9 +100,24 @@ pub fn build_module_help<'a>(
RefcountProcGenerator::new(env.arena, IntWidth::I32, refcount_home),
);
// Main loop: Generate the procs from the IR
for (proc, sym) in generated_procs.into_iter().zip(generated_symbols) {
backend.build_proc(proc, sym)?;
}
// The RefcountProcGenerator has now built up a vector of extra procs to generate.
// But we need to move that vector to be able to safely loop over it.
let mut procs_to_generate = Vec::with_capacity_in(0, env.arena);
std::mem::swap(&mut backend.refcount_proc_gen.procs_to_generate, &mut procs_to_generate);
// Generate the refcount helper procedures
for (layout, op, symbol) in procs_to_generate.drain(0..) {
let proc = backend
.refcount_proc_gen
.generate_refcount_proc(layout, op, symbol);
backend.build_proc(proc, symbol)?;
}
(backend.module, backend.linker_symbols)
};