Skip to content

Commit beb4edb

Browse files
committed
Add benchmark for add_3d strided
1 parent 623af9a commit beb4edb

File tree

1 file changed

+27
-0
lines changed

1 file changed

+27
-0
lines changed

benches/bench1.rs

Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -583,6 +583,33 @@ fn add_2d_f32_regular(bench: &mut test::Bencher)
583583
});
584584
}
585585

586+
const ADD3DSZ: usize = 16;
587+
588+
#[bench]
589+
fn add_3d_strided(bench: &mut test::Bencher)
590+
{
591+
let mut a = Array::<i32, _>::zeros((ADD3DSZ, ADD3DSZ, ADD3DSZ * 2));
592+
let mut a = a.slice_mut(s![.., .., ..;2]);
593+
let b = Array::<i32, _>::zeros(a.dim());
594+
let bv = b.view();
595+
bench.iter(|| {
596+
a += &bv;
597+
});
598+
}
599+
600+
#[bench]
601+
fn add_3d_strided_dyn(bench: &mut test::Bencher)
602+
{
603+
let mut a = Array::<i32, _>::zeros(&[ADD3DSZ, ADD3DSZ, ADD3DSZ * 2][..]);
604+
let mut a = a.slice_mut(s![.., .., ..;2]);
605+
let b = Array::<i32, _>::zeros(a.dim());
606+
let bv = b.view();
607+
bench.iter(|| {
608+
a += &bv;
609+
});
610+
}
611+
612+
586613
const ADD1D_SIZE: usize = 64 * 64;
587614

588615
#[bench]

0 commit comments

Comments
 (0)