@@ -6,30 +6,27 @@ use test::Bencher;
66use ndarray:: prelude:: * ;
77
88#[ bench]
9- fn select_axis0 ( bench : & mut Bencher ) {
9+ fn select_axis0 ( bench : & mut Bencher )
10+ {
1011 let a = Array :: < f32 , _ > :: zeros ( ( 256 , 256 ) ) ;
1112 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- } ) ;
13+ bench. iter ( || a. select ( Axis ( 0 ) , & selectable) ) ;
1514}
1615
1716#[ bench]
18- fn select_axis1 ( bench : & mut Bencher ) {
17+ fn select_axis1 ( bench : & mut Bencher )
18+ {
1919 let a = Array :: < f32 , _ > :: zeros ( ( 256 , 256 ) ) ;
2020 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- } ) ;
21+ bench. iter ( || a. select ( Axis ( 1 ) , & selectable) ) ;
2422}
2523
2624#[ bench]
27- fn select_1d ( bench : & mut Bencher ) {
25+ fn select_1d ( bench : & mut Bencher )
26+ {
2827 let a = Array :: < f32 , _ > :: zeros ( 1024 ) ;
2928 let mut selectable = ( 0 ..a. len ( ) ) . step_by ( 17 ) . collect :: < Vec < _ > > ( ) ;
3029 selectable. extend ( selectable. clone ( ) . iter ( ) . rev ( ) ) ;
3130
32- bench. iter ( || {
33- a. select ( Axis ( 0 ) , & selectable)
34- } ) ;
31+ bench. iter ( || a. select ( Axis ( 0 ) , & selectable) ) ;
3532}
0 commit comments