Skip to content

Commit 1654125

Browse files
committed
TEST: Add benchmarks for .select() (based on append)
1 parent dd353b8 commit 1654125

File tree

1 file changed

+24
-0
lines changed

1 file changed

+24
-0
lines changed

benches/append.rs

+24
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,24 @@
1+
#![feature(test)]
2+
3+
extern crate test;
4+
use test::Bencher;
5+
6+
use ndarray::prelude::*;
7+
8+
#[bench]
9+
fn select_axis0(bench: &mut Bencher) {
10+
let a = Array::<f32, _>::zeros((256, 256));
11+
let selectable = vec![0, 1, 2, 0, 1, 3, 0, 4, 16, 32, 128, 147, 149, 220, 221, 255, 221, 0, 1];
12+
bench.iter(|| {
13+
a.select(Axis(0), &selectable)
14+
});
15+
}
16+
17+
#[bench]
18+
fn select_axis1(bench: &mut Bencher) {
19+
let a = Array::<f32, _>::zeros((256, 256));
20+
let selectable = vec![0, 1, 2, 0, 1, 3, 0, 4, 16, 32, 128, 147, 149, 220, 221, 255, 221, 0, 1];
21+
bench.iter(|| {
22+
a.select(Axis(1), &selectable)
23+
});
24+
}

0 commit comments

Comments
 (0)