Skip to content

Commit

Permalink
New config
Browse files Browse the repository at this point in the history
  • Loading branch information
skailasa committed Jan 13, 2025
1 parent d475c2a commit c037041
Show file tree
Hide file tree
Showing 4 changed files with 44 additions and 5 deletions.
14 changes: 12 additions & 2 deletions kifmm/benches/laplace_f32_amd.rs
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,10 @@ use kifmm::traits::tree::{SingleFmmTree, SingleTree};
use kifmm::tree::helpers::{points_fixture, points_fixture_oblate_spheroid, points_fixture_sphere};
use rlst::{rlst_dynamic_array2, RawAccess, RawAccessMut};

fn configure_criterion() -> Criterion {
Criterion::default().sample_size(10) // Set global sample size to 10
}

fn fft_f32(c: &mut Criterion) {
let mut group = c.benchmark_group("F32 Potentials FFT-M2L");

Expand Down Expand Up @@ -230,5 +234,11 @@ fn blas_f32(c: &mut Criterion) {
}
}

criterion_group!(laplace_p_f32, fft_f32, blas_f32);
criterion_main!(laplace_p_f32);

criterion_group! {
name = laplace_p_f32;
config = configure_criterion();
targets = fft_f32, blas_f32
}

Check warning on line 242 in kifmm/benches/laplace_f32_amd.rs

View workflow job for this annotation

GitHub Actions / Rust style checks

Diff in /home/runner/work/kifmm/kifmm/kifmm/benches/laplace_f32_amd.rs

Check warning on line 242 in kifmm/benches/laplace_f32_amd.rs

View workflow job for this annotation

GitHub Actions / Rust style checks (--features "strict")

Diff in /home/runner/work/kifmm/kifmm/kifmm/benches/laplace_f32_amd.rs

criterion_main!(laplace_p_f32);
12 changes: 11 additions & 1 deletion kifmm/benches/laplace_f32_m1.rs
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,10 @@ use kifmm::traits::tree::{SingleFmmTree, SingleTree};
use kifmm::tree::helpers::{points_fixture, points_fixture_oblate_spheroid, points_fixture_sphere};
use rlst::{rlst_dynamic_array2, RawAccess, RawAccessMut};

fn configure_criterion() -> Criterion {
Criterion::default().sample_size(10) // Set global sample size to 10
}

fn fft_f32(c: &mut Criterion) {
let mut group = c.benchmark_group("F32 Potentials FFT-M2L");

Expand Down Expand Up @@ -226,5 +230,11 @@ fn blas_f32(c: &mut Criterion) {
}
}

criterion_group!(laplace_p_f32, fft_f32, blas_f32);

criterion_group! {
name = laplace_p_f32;
config = configure_criterion();
targets = fft_f32, blas_f32
}

criterion_main!(laplace_p_f32);
11 changes: 10 additions & 1 deletion kifmm/benches/laplace_f64_amd.rs
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,10 @@ use kifmm::traits::tree::{SingleFmmTree, SingleTree};
use kifmm::tree::helpers::{points_fixture, points_fixture_oblate_spheroid, points_fixture_sphere};
use rlst::{rlst_dynamic_array2, RawAccess, RawAccessMut};

fn configure_criterion() -> Criterion {
Criterion::default().sample_size(10) // Set global sample size to 10
}

fn fft_f64(c: &mut Criterion) {
let mut group = c.benchmark_group("F64 Potentials FFT-M2L");

Expand Down Expand Up @@ -238,5 +242,10 @@ fn blas_f64(c: &mut Criterion) {
}
}

criterion_group!(laplace_p_f64, fft_f64, blas_f64);
criterion_group! {
name = laplace_p_f64;
config = configure_criterion();
targets = fft_f64, blas_f64
}

criterion_main!(laplace_p_f64);
12 changes: 11 additions & 1 deletion kifmm/benches/laplace_f64_m1.rs
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,11 @@ use kifmm::traits::tree::{SingleFmmTree, SingleTree};
use kifmm::tree::helpers::{points_fixture, points_fixture_oblate_spheroid, points_fixture_sphere};
use rlst::RawAccess;


fn configure_criterion() -> Criterion {
Criterion::default().sample_size(10) // Set global sample size to 10
}

fn fft_f64(c: &mut Criterion) {
let mut group = c.benchmark_group("F64 Potentials FFT-M2L");

Expand Down Expand Up @@ -237,5 +242,10 @@ fn blas_f64(c: &mut Criterion) {
}
}

criterion_group!(laplace_p_f64, fft_f64, blas_f64);
criterion_group! {
name = laplace_p_f64;
config = configure_criterion();
targets = fft_f64, blas_f64
}

criterion_main!(laplace_p_f64);

0 comments on commit c037041

Please sign in to comment.