Skip to content

Commit 73ae59b

Browse files
committed
Add benches for op with scalar and strided array
1 parent c043855 commit 73ae59b

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

benches/bench1.rs

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -387,6 +387,22 @@ fn scalar_add_2(bench: &mut test::Bencher) {
387387
bench.iter(|| n + &a);
388388
}
389389

390+
#[bench]
391+
fn scalar_add_strided_1(bench: &mut test::Bencher) {
392+
let a =
393+
Array::from_shape_fn((64, 64 * 2), |(i, j)| (i * 64 + j) as f32).slice_move(s![.., ..;2]);
394+
let n = 1.;
395+
bench.iter(|| &a + n);
396+
}
397+
398+
#[bench]
399+
fn scalar_add_strided_2(bench: &mut test::Bencher) {
400+
let a =
401+
Array::from_shape_fn((64, 64 * 2), |(i, j)| (i * 64 + j) as f32).slice_move(s![.., ..;2]);
402+
let n = 1.;
403+
bench.iter(|| n + &a);
404+
}
405+
390406
#[bench]
391407
fn scalar_sub_1(bench: &mut test::Bencher) {
392408
let a = Array::<f32, _>::zeros((64, 64));

0 commit comments

Comments
 (0)