|
| 1 | +//! Benchmarks for the evaluation of the Helmholtz energy function |
| 2 | +//! for a given `StateHD` for different types of dual numbers. |
| 3 | +//! These should give an idea about the expected slow-down depending |
| 4 | +//! on the dual number type used without the overhead of the `State` |
| 5 | +//! creation. |
| 6 | +use criterion::{criterion_group, criterion_main, Criterion}; |
| 7 | +use feos::hard_sphere::HardSphereProperties; |
| 8 | +use feos::saftvrmie::{test_utils::test_parameters, SaftVRMie, SaftVRMieParameters}; |
| 9 | +use feos_core::si::*; |
| 10 | +use feos_core::{Derivative, Residual, State, StateHD}; |
| 11 | +use ndarray::{Array, ScalarOperand}; |
| 12 | +use num_dual::DualNum; |
| 13 | +use std::sync::Arc; |
| 14 | + |
| 15 | +/// Helper function to create a state for given parameters. |
| 16 | +/// - temperature is 80% of critical temperature, |
| 17 | +/// - volume is critical volume, |
| 18 | +/// - molefracs (or moles) for equimolar mixture. |
| 19 | +fn state_saftvrmie(parameters: &Arc<SaftVRMieParameters>) -> State<SaftVRMie> { |
| 20 | + let n = parameters.pure_records.len(); |
| 21 | + let eos = Arc::new(SaftVRMie::new(parameters.clone())); |
| 22 | + let moles = Array::from_elem(n, 1.0 / n as f64) * 10.0 * MOL; |
| 23 | + let cp = State::critical_point(&eos, Some(&moles), None, Default::default()).unwrap(); |
| 24 | + let temperature = 0.8 * cp.temperature; |
| 25 | + State::new_nvt(&eos, temperature, cp.volume, &moles).unwrap() |
| 26 | +} |
| 27 | + |
| 28 | +/// Residual Helmholtz energy given an equation of state and a StateHD. |
| 29 | +fn a_res<D: DualNum<f64> + Copy + ScalarOperand, E: Residual>(inp: (&Arc<E>, &StateHD<D>)) -> D { |
| 30 | + inp.0.residual_helmholtz_energy(inp.1) |
| 31 | +} |
| 32 | + |
| 33 | +fn d_hs<D: DualNum<f64> + Copy>(inp: (&SaftVRMieParameters, D)) -> D { |
| 34 | + inp.0.hs_diameter(inp.1)[0] |
| 35 | +} |
| 36 | + |
| 37 | +/// Benchmark for evaluation of the Helmholtz energy for different dual number types. |
| 38 | +fn bench_dual_numbers<E: Residual>( |
| 39 | + c: &mut Criterion, |
| 40 | + group_name: &str, |
| 41 | + state: State<E>, |
| 42 | + parameters: &SaftVRMieParameters, |
| 43 | +) { |
| 44 | + let mut group = c.benchmark_group(group_name); |
| 45 | + group.bench_function("d_f64", |b| { |
| 46 | + b.iter(|| d_hs((¶meters, state.derive0().temperature))) |
| 47 | + }); |
| 48 | + group.bench_function("d_dual", |b| { |
| 49 | + b.iter(|| d_hs((¶meters, state.derive1(Derivative::DT).temperature))) |
| 50 | + }); |
| 51 | + group.bench_function("d_dual2", |b| { |
| 52 | + b.iter(|| d_hs((¶meters, state.derive2(Derivative::DT).temperature))) |
| 53 | + }); |
| 54 | + group.bench_function("d_hyperdual", |b| { |
| 55 | + b.iter(|| { |
| 56 | + d_hs(( |
| 57 | + ¶meters, |
| 58 | + state |
| 59 | + .derive2_mixed(Derivative::DT, Derivative::DT) |
| 60 | + .temperature, |
| 61 | + )) |
| 62 | + }) |
| 63 | + }); |
| 64 | + group.bench_function("d_dual3", |b| { |
| 65 | + b.iter(|| d_hs((¶meters, state.derive3(Derivative::DT).temperature))) |
| 66 | + }); |
| 67 | + |
| 68 | + group.bench_function("a_f64", |b| { |
| 69 | + b.iter(|| a_res((&state.eos, &state.derive0()))) |
| 70 | + }); |
| 71 | + group.bench_function("a_dual", |b| { |
| 72 | + b.iter(|| a_res((&state.eos, &state.derive1(Derivative::DV)))) |
| 73 | + }); |
| 74 | + group.bench_function("a_dual2", |b| { |
| 75 | + b.iter(|| a_res((&state.eos, &state.derive2(Derivative::DV)))) |
| 76 | + }); |
| 77 | + group.bench_function("a_hyperdual", |b| { |
| 78 | + b.iter(|| { |
| 79 | + a_res(( |
| 80 | + &state.eos, |
| 81 | + &state.derive2_mixed(Derivative::DV, Derivative::DV), |
| 82 | + )) |
| 83 | + }) |
| 84 | + }); |
| 85 | + group.bench_function("a_dual3", |b| { |
| 86 | + b.iter(|| a_res((&state.eos, &state.derive3(Derivative::DV)))) |
| 87 | + }); |
| 88 | +} |
| 89 | + |
| 90 | +/// Benchmark for the SAFT VR Mie equation of state |
| 91 | +fn saftvrmie(c: &mut Criterion) { |
| 92 | + let parameters = Arc::new(test_parameters().remove("ethane").unwrap()); |
| 93 | + bench_dual_numbers( |
| 94 | + c, |
| 95 | + "dual_numbers_saftvrmie_ethane", |
| 96 | + state_saftvrmie(¶meters), |
| 97 | + ¶meters, |
| 98 | + ); |
| 99 | +} |
| 100 | + |
| 101 | +criterion_group!(bench, saftvrmie); |
| 102 | +criterion_main!(bench); |
0 commit comments