Skip to content

Commit 52ca234

Browse files
jturner314bluss
authored andcommitted
Add benches for op with scalar and strided array
1 parent 79392bb commit 52ca234

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
@@ -431,6 +431,22 @@ fn scalar_add_2(bench: &mut test::Bencher) {
431431
bench.iter(|| n + &a);
432432
}
433433

434+
#[bench]
435+
fn scalar_add_strided_1(bench: &mut test::Bencher) {
436+
let a =
437+
Array::from_shape_fn((64, 64 * 2), |(i, j)| (i * 64 + j) as f32).slice_move(s![.., ..;2]);
438+
let n = 1.;
439+
bench.iter(|| &a + n);
440+
}
441+
442+
#[bench]
443+
fn scalar_add_strided_2(bench: &mut test::Bencher) {
444+
let a =
445+
Array::from_shape_fn((64, 64 * 2), |(i, j)| (i * 64 + j) as f32).slice_move(s![.., ..;2]);
446+
let n = 1.;
447+
bench.iter(|| n + &a);
448+
}
449+
434450
#[bench]
435451
fn scalar_sub_1(bench: &mut test::Bencher) {
436452
let a = Array::<f32, _>::zeros((64, 64));

0 commit comments

Comments
 (0)