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

make construct_closure_data more polymorphic

Browse source
This commit is contained in:
Folkert 2022-05-01 14:29:02 +02:00
parent c3209fdded
commit 10c2c5e8fd
No known key found for this signature in database
GPG key ID: 1F17F6FFD112B97C
1 changed files with 33 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

50
compiler/mono/src/ir.rs
View file

@ -4162,7 +4162,7 @@ pub fn with_hole<'a>(
layout_cache, layout_cache,
lambda_set, lambda_set,
name, name,
symbols, symbols.iter().copied(),
assigned, assigned,
hole, hole,
) )
@ -4734,17 +4734,25 @@ fn get_specialization<'a>(
} }
#[allow(clippy::too_many_arguments)] #[allow(clippy::too_many_arguments)]
fn construct_closure_data<'a>( fn construct_closure_data<'a, I>(
env: &mut Env<'a, '_>, env: &mut Env<'a, '_>,
procs: &mut Procs<'a>, procs: &mut Procs<'a>,
layout_cache: &mut LayoutCache<'a>, layout_cache: &mut LayoutCache<'a>,
lambda_set: LambdaSet<'a>, lambda_set: LambdaSet<'a>,
name: Symbol, name: Symbol,
symbols: &'a [&(Symbol, Variable)], symbols: I,
assigned: Symbol, assigned: Symbol,
hole: &'a Stmt<'a>, hole: &'a Stmt<'a>,
) -> Stmt<'a> { ) -> Stmt<'a>
where
I: IntoIterator<Item = &'a (Symbol, Variable)>,
I::IntoIter: ExactSizeIterator,
{
let lambda_set_layout = Layout::LambdaSet(lambda_set); let lambda_set_layout = Layout::LambdaSet(lambda_set);
let symbols = symbols.into_iter();
// arguments with a polymorphic type that we have to deal with
let mut polymorphic_arguments = Vec::new_in(env.arena);
let mut result = match lambda_set.layout_for_member(name) { let mut result = match lambda_set.layout_for_member(name) {
ClosureRepresentation::Union { ClosureRepresentation::Union {
@ -4755,10 +4763,14 @@ fn construct_closure_data<'a>(
} => { } => {
// captured variables are in symbol-alphabetic order, but now we want // captured variables are in symbol-alphabetic order, but now we want
// them ordered by their alignment requirements // them ordered by their alignment requirements
let mut combined = Vec::from_iter_in( let mut combined = Vec::with_capacity_in(symbols.len(), env.arena);
symbols.iter().map(|&&(s, _)| s).zip(field_layouts.iter()), for ((symbol, variable), layout) in symbols.zip(field_layouts.iter()) {
env.arena, if procs.partial_exprs.contains(*symbol) {
); polymorphic_arguments.push((*symbol, *variable));
}
combined.push((*symbol, layout))
}
let ptr_bytes = env.target_info; let ptr_bytes = env.target_info;
@ -4786,10 +4798,14 @@ fn construct_closure_data<'a>(
// captured variables are in symbol-alphabetic order, but now we want // captured variables are in symbol-alphabetic order, but now we want
// them ordered by their alignment requirements // them ordered by their alignment requirements
let mut combined = Vec::from_iter_in( let mut combined = Vec::with_capacity_in(symbols.len(), env.arena);
symbols.iter().map(|&(s, _)| s).zip(field_layouts.iter()), for ((symbol, variable), layout) in symbols.zip(field_layouts.iter()) {
env.arena, if procs.partial_exprs.contains(*symbol) {
); polymorphic_arguments.push((*symbol, *variable));
}
combined.push((*symbol, layout))
}
let ptr_bytes = env.target_info; let ptr_bytes = env.target_info;
@ -4801,7 +4817,7 @@ fn construct_closure_data<'a>(
}); });
let symbols = let symbols =
Vec::from_iter_in(combined.iter().map(|(a, _)| **a), env.arena).into_bump_slice(); Vec::from_iter_in(combined.iter().map(|(a, _)| *a), env.arena).into_bump_slice();
let field_layouts = let field_layouts =
Vec::from_iter_in(combined.iter().map(|(_, b)| **b), env.arena).into_bump_slice(); Vec::from_iter_in(combined.iter().map(|(_, b)| **b), env.arena).into_bump_slice();
@ -4842,11 +4858,9 @@ fn construct_closure_data<'a>(
// TODO: this is not quite right. What we should actually be doing is removing references to // TODO: this is not quite right. What we should actually be doing is removing references to
// polymorphic expressions from the captured symbols, and allowing the specializations of those // polymorphic expressions from the captured symbols, and allowing the specializations of those
// symbols to be inlined when specializing the closure body elsewhere. // symbols to be inlined when specializing the closure body elsewhere.
for &&(symbol, var) in symbols { for (symbol, var) in polymorphic_arguments {
if procs.partial_exprs.contains(symbol) {
result = specialize_symbol(env, procs, layout_cache, Some(var), symbol, result, symbol); result = specialize_symbol(env, procs, layout_cache, Some(var), symbol, result, symbol);
} }
}
result result
} }
@ -6937,7 +6951,7 @@ fn specialize_symbol<'a>(
layout_cache, layout_cache,
lambda_set, lambda_set,
original, original,
symbols, symbols.iter().copied(),
closure_data, closure_data,
env.arena.alloc(result), env.arena.alloc(result),
) )
@ -7696,7 +7710,7 @@ fn call_specialized_proc<'a>(
layout_cache, layout_cache,
lambda_set, lambda_set,
proc_name, proc_name,
symbols, symbols.iter().copied(),
closure_data_symbol, closure_data_symbol,
env.arena.alloc(new_hole), env.arena.alloc(new_hole),
); );