Skip to content

Commit

Permalink
collapse run_k into run
Browse files Browse the repository at this point in the history
  • Loading branch information
miguelraz committed Jun 1, 2021
1 parent 5605056 commit f24110a
Showing 1 changed file with 2 additions and 9 deletions.
11 changes: 2 additions & 9 deletions crates/core_simd/examples/nbody.rs
Expand Up @@ -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 {
Expand Down

0 comments on commit f24110a

Please sign in to comment.