File tree Expand file tree Collapse file tree 1 file changed +27
-0
lines changed Expand file tree Collapse file tree 1 file changed +27
-0
lines changed Original file line number Diff line number Diff line change @@ -583,6 +583,33 @@ fn add_2d_f32_regular(bench: &mut test::Bencher)
583
583
} ) ;
584
584
}
585
585
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
+
586
613
const ADD1D_SIZE : usize = 64 * 64 ;
587
614
588
615
#[ bench]
You can’t perform that action at this time.
0 commit comments