squiid/rust
1
0
Fork 0
Code Issues Pull requests Projects Activity

collapse run_k into run

Browse source
This commit is contained in:
miguel raz 2021-06-01 12:26:29 -05:00
parent 56050562f1
commit f24110aa5b
1 changed files with 2 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
crates/core_simd/examples/nbody.rs
View file

@ -146,25 +146,18 @@ pub fn advance(bodies: &mut [Body; N_BODIES], dt: f64) {
}
}
pub fn run_k<K>(n: usize, k: K) -> (f64, f64)
where
K: Fn(&mut [Body; N_BODIES], f64),
{
pub fn run(n: usize) -> (f64, f64) {
let mut bodies = BODIES;
offset_momentum(&mut bodies);
let energy_before = energy(&bodies);
for _ in 0..n {
k(&mut bodies, 0.01);
advance(&mut bodies, 0.01);
}
let energy_after = energy(&bodies);
(energy_before, energy_after)
}
pub fn run(n: usize) -> (f64, f64) {
run_k(n, advance)
}
const OUTPUT: Vec<f64> = vec![-0.169075164, -0.169087605];
#[cfg(test)]
mod tests {