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

lambda set unwrapping

Browse source
This commit is contained in:
Folkert 2023-05-02 22:19:34 +02:00
parent 0fa85fe91d
commit fb123b8990
No known key found for this signature in database
GPG key ID: 1F17F6FFD112B97C
2 changed files with 72 additions and 34 deletions
Download patch file Download diff file Expand all files Collapse all files

11
crates/compiler/gen_dev/src/generic64/mod.rs
View file

@ -3298,6 +3298,17 @@ impl<
});
}
Layout::LambdaSet(lambda_set) => {
Self::ptr_read(
buf,
storage_manager,
layout_interner,
ptr_reg,
lambda_set.runtime_representation(),
dst,
);
}
_ => todo!("unboxing of {:?}", layout_interner.dbg(element_in_layout)),
}
}

95
crates/compiler/gen_dev/src/generic64/storage.rs
View file

@ -1168,28 +1168,43 @@ impl<
{
// Claim a location for every join point parameter to be loaded at.
// Put everything on the stack for simplicity.
match *layout {
single_register_layouts!() => {
let base_offset = self.claim_stack_size(8);
self.symbol_storage_map.insert(
*symbol,
Stack(Primitive {
base_offset,
reg: None,
}),
);
self.allocation_map
.insert(*symbol, Rc::new((base_offset, 8)));
}
_ => {
let stack_size = layout_interner.stack_size(*layout);
if stack_size == 0 {
self.symbol_storage_map.insert(*symbol, NoData);
} else {
self.claim_stack_area(symbol, stack_size);
let mut layout = *layout;
'inner: loop {
match layout {
single_register_layouts!() => {
let base_offset = self.claim_stack_size(8);
self.symbol_storage_map.insert(
*symbol,
Stack(Primitive {
base_offset,
reg: None,
}),
);
self.allocation_map
.insert(*symbol, Rc::new((base_offset, 8)));
break 'inner;
}
_ => match layout_interner.get(layout) {
Layout::LambdaSet(lambda_set) => {
layout = lambda_set.runtime_representation();
continue 'inner;
}
_ => {
let stack_size = layout_interner.stack_size(layout);
if stack_size == 0 {
self.no_data(symbol);
} else {
self.claim_stack_area(symbol, stack_size);
}
break 'inner;
}
},
}
}
param_storage.push(*self.get_storage_for_sym(symbol));
}
self.join_param_map.insert(*id, param_storage);
@ -1238,22 +1253,34 @@ impl<
Stack(Primitive {
base_offset,
reg: None,
}) => match *layout {
single_register_integers!() => {
let reg = self.load_to_general_reg(buf, sym);
ASM::mov_base32_reg64(buf, *base_offset, reg);
}) => {
let mut layout = *layout;
loop {
match layout {
single_register_integers!() => {
let reg = self.load_to_general_reg(buf, sym);
ASM::mov_base32_reg64(buf, *base_offset, reg);
}
single_register_floats!() => {
let reg = self.load_to_float_reg(buf, sym);
ASM::mov_base32_freg64(buf, *base_offset, reg);
}
_ => {
if let Layout::LambdaSet(lambda_set) = layout_interner.get(layout) {
layout = lambda_set.runtime_representation();
continue;
} else {
internal_error!(
r"cannot load non-primitive layout ({:?}) to primitive stack location",
layout
)
}
}
}
break;
}
single_register_floats!() => {
let reg = self.load_to_float_reg(buf, sym);
ASM::mov_base32_freg64(buf, *base_offset, reg);
}
_ => {
internal_error!(
"cannot load non-primitive layout ({:?}) to primitive stack location",
layout
);
}
},
}
NoData => {}
Stack(Primitive { reg: Some(_), .. }) => {
internal_error!(