language-servers/roc
1
0
Fork 0
mirror of https://github.com/roc-lang/roc.git synced 2025-09-26 13:29:12 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

get wasm dev backend working

Browse source
This commit is contained in:
Brendan Hansknecht 2024-07-21 19:53:36 -07:00
parent 82764dc697
commit b8fb83af86
No known key found for this signature in database
GPG key ID: 0EA784685083E75B
4 changed files with 126 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

25
crates/compiler/gen_wasm/src/backend.rs
View file

@ -1981,12 +1981,33 @@ impl<'a, 'r> WasmBackend<'a, 'r> {
self.register_helper_proc(spec_sym, spec_layout, ProcSource::Helper);
}
self.get_existing_refcount_fn_index(proc_symbol, layout, op)
self.get_existing_helper_fn_index(proc_symbol, layout, op)
}
/// Generate a copy helper procedure and return a pointer (table index) to it
/// This allows it to be indirectly called from Zig code
pub fn get_copy_fn_index(&mut self, layout: InLayout<'a>) -> u32 {
let ident_ids = self
.interns
.all_ident_ids
.get_mut(&self.env.module_id)
.unwrap();
let (proc_symbol, new_specializations) =
self.helper_proc_gen
.gen_copy_proc(ident_ids, self.layout_interner, layout);
// If any new specializations were created, register their symbol data
for (spec_sym, spec_layout) in new_specializations.into_iter() {
self.register_helper_proc(spec_sym, spec_layout, ProcSource::Helper);
}
self.get_existing_helper_fn_index(proc_symbol, layout, HelperOp::IndirectCopy)
}
/// return a pointer (table index) to a refcount helper procedure.
/// This allows it to be indirectly called from Zig code
pub fn get_existing_refcount_fn_index(
pub fn get_existing_helper_fn_index(
&mut self,
proc_symbol: Symbol,
layout: InLayout<'a>,

28
crates/compiler/gen_wasm/src/low_level.rs
View file

@ -352,7 +352,7 @@ impl<'a> LowLevelCall<'a> {
elem_layout.stack_size_and_alignment(backend.layout_interner);
let elem_refcounted = backend.layout_interner.contains_refcounted(elem_in_layout);
let dec_fn = backend.get_existing_refcount_fn_index(
let dec_fn = backend.get_existing_helper_fn_index(
dec_fn_sym,
elem_in_layout,
HelperOp::IndirectDec,
@ -476,6 +476,7 @@ impl<'a> LowLevelCall<'a> {
build_refcount_element_fn(backend, elem_in_layout, HelperOp::IndirectInc);
let dec_fn_ptr =
build_refcount_element_fn(backend, elem_in_layout, HelperOp::IndirectDec);
let copy_fn_ptr = build_copy_element_fn(backend, elem_in_layout);
// Load all the arguments for Zig
// (List return pointer) i32
@ -488,6 +489,7 @@ impl<'a> LowLevelCall<'a> {
// inc: Inc i32
// dec: Dec i32
// out_element: ?[*]u8, i32
// copy: CopyFn, i32
let code_builder = &mut backend.code_builder;
@ -517,6 +519,7 @@ impl<'a> LowLevelCall<'a> {
code_builder.i32_const((ret_offset + ret_elem_offset) as i32);
code_builder.i32_add();
}
code_builder.i32_const(copy_fn_ptr);
// There is an in-place version of this but we don't use it for dev backends. No morphic_lib analysis.
backend.call_host_fn_after_loading_args(bitcode::LIST_REPLACE);
@ -689,16 +692,18 @@ impl<'a> LowLevelCall<'a> {
let list: Symbol = self.arguments[0];
let elem: Symbol = self.arguments[1];
let elem_layout = unwrap_list_elem_layout(self.ret_layout_raw);
let elem_width = backend.layout_interner.stack_size(elem_layout);
let elem_in_layout = unwrap_list_elem_layout(self.ret_layout_raw);
let elem_width = backend.layout_interner.stack_size(elem_in_layout);
let (elem_local, elem_offset, _) =
ensure_symbol_is_in_memory(backend, elem, elem_layout, backend.env.arena);
ensure_symbol_is_in_memory(backend, elem, elem_in_layout, backend.env.arena);
let copy_fn_ptr = build_copy_element_fn(backend, elem_in_layout);
// Zig arguments Wasm types
// (return pointer) i32
// list: RocList i32
// element: Opaque i32
// element_width: usize i32
// copy: CopyFn, i32
// return pointer and list
backend.storage.load_symbols_for_call(
@ -715,6 +720,7 @@ impl<'a> LowLevelCall<'a> {
}
backend.code_builder.i32_const(elem_width as i32);
backend.code_builder.i32_const(copy_fn_ptr);
backend.call_host_fn_after_loading_args(bitcode::LIST_APPEND_UNSAFE);
}
@ -734,6 +740,7 @@ impl<'a> LowLevelCall<'a> {
let elem_refcounted = backend.layout_interner.contains_refcounted(elem_in_layout);
let inc_fn_ptr =
build_refcount_element_fn(backend, elem_in_layout, HelperOp::IndirectInc);
let copy_fn_ptr = build_copy_element_fn(backend, elem_in_layout);
// Zig arguments Wasm types
// (return pointer) i32
@ -743,6 +750,7 @@ impl<'a> LowLevelCall<'a> {
// element_width: usize i32
// element_refcounted: bool i32
// inc: Inc i32
// copy: CopyFn, i32
// return pointer and list
backend.storage.load_symbols_for_call(
@ -762,6 +770,7 @@ impl<'a> LowLevelCall<'a> {
backend.code_builder.i32_const(elem_width as i32);
backend.code_builder.i32_const(elem_refcounted as i32);
backend.code_builder.i32_const(inc_fn_ptr);
backend.code_builder.i32_const(copy_fn_ptr);
backend.call_host_fn_after_loading_args(bitcode::LIST_PREPEND);
}
@ -870,6 +879,7 @@ impl<'a> LowLevelCall<'a> {
build_refcount_element_fn(backend, elem_in_layout, HelperOp::IndirectInc);
let dec_fn_ptr =
build_refcount_element_fn(backend, elem_in_layout, HelperOp::IndirectDec);
let copy_fn_ptr = build_copy_element_fn(backend, elem_in_layout);
// Zig arguments Wasm types
// (return pointer) i32
@ -882,6 +892,7 @@ impl<'a> LowLevelCall<'a> {
// inc: Inc i32
// dec: Dec i32
// update_mode: UpdateMode, i32
// copy: CopyFn, i32
// Load the return pointer and the list
backend.storage.load_symbols_for_call(
@ -900,6 +911,7 @@ impl<'a> LowLevelCall<'a> {
backend.code_builder.i32_const(inc_fn_ptr);
backend.code_builder.i32_const(dec_fn_ptr);
backend.code_builder.i32_const(UPDATE_MODE_IMMUTABLE);
backend.code_builder.i32_const(copy_fn_ptr);
backend.call_host_fn_after_loading_args(bitcode::LIST_SWAP);
}
@ -2784,6 +2796,7 @@ pub fn call_higher_order_lowlevel<'a>(
build_refcount_element_fn(backend, elem_in_layout, HelperOp::IndirectInc);
let dec_fn_ptr =
build_refcount_element_fn(backend, elem_in_layout, HelperOp::IndirectDec);
let copy_fn_ptr = build_copy_element_fn(backend, elem_in_layout);
let cb = &mut backend.code_builder;
@ -2798,6 +2811,7 @@ pub fn call_higher_order_lowlevel<'a>(
// element_refcounted: bool i32
// inc: Inc i32
// dec: Dec i32
// copy: CopyFn i32
backend.storage.load_symbols(cb, &[return_sym, *xs]);
cb.i32_const(wrapper_fn_ptr);
@ -2817,6 +2831,7 @@ pub fn call_higher_order_lowlevel<'a>(
cb.i32_const(elem_refcounted as i32);
cb.i32_const(inc_fn_ptr);
cb.i32_const(dec_fn_ptr);
cb.i32_const(copy_fn_ptr);
backend.call_host_fn_after_loading_args(bitcode::LIST_SORT_WITH);
}
@ -2869,3 +2884,8 @@ fn build_refcount_element_fn<'a>(
let rc_fn = backend.get_refcount_fn_index(elem_layout, rc_op);
backend.get_fn_ptr(rc_fn)
}
fn build_copy_element_fn<'a>(backend: &mut WasmBackend<'a, '_>, elem_layout: InLayout<'a>) -> i32 {
let copy_fn = backend.get_copy_fn_index(elem_layout);
backend.get_fn_ptr(copy_fn)
}