mirrors/salsa
1
0
Fork 0
mirror of https://github.com/salsa-rs/salsa.git synced 2025-01-27 07:04:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Avoid unnecessary resizes of visisted set

Browse source
This commit is contained in:
Micha Reiser 2024-12-04 09:34:35 +01:00
parent e68679b3a9
commit 8f6dcaf632
No known key found for this signature in database
1 changed files with 21 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
src/function/accumulated.rs
View file

@ -69,15 +69,31 @@ where
// output vector, we want to push in execution order, so reverse order to
// ensure the first child that was executed will be the first child popped
// from the stack.
let origin = zalsa
let Some(origin) = zalsa
.lookup_ingredient(k.ingredient_index)
.origin(db, k.key_index);
let inputs = origin.iter().flat_map(|origin| origin.inputs());
.origin(db, k.key_index)
else {
continue;
};
let estimated_inputs = match &origin {
QueryOrigin::Assigned(_) | QueryOrigin::BaseInput => 0,
QueryOrigin::Derived(edges) | QueryOrigin::DerivedUntracked(edges) => {
edges.input_outputs.len()
}
};
stack.reserve(estimated_inputs);
visited.reserve(estimated_inputs);
stack.extend(
inputs
.flat_map(|input| TryInto::<DatabaseKeyIndex>::try_into(input).into_iter())
origin
.inputs()
.filter_map(|input| TryInto::<DatabaseKeyIndex>::try_into(input).ok())
.rev(),
);
visited.reserve(stack.len());
}
output