Skip to content

Commit 2e13bb2

Browse files
bench: add dot_3d_f64_contiguous and dot_3d_f64_non_contiguous benchmarks
1 parent 1145645 commit 2e13bb2

1 file changed

Lines changed: 18 additions & 0 deletions

File tree

benches/bench1.rs

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@ extern crate test;
66

77
use std::mem::MaybeUninit;
88

9+
use ndarray::linalg::Dot;
910
use ndarray::{arr0, arr1, arr2, azip, s};
1011
use ndarray::{Array, Array1, Array2, Axis, Ix, Zip};
1112
use ndarray::{Array3, Array4, ShapeBuilder};
@@ -908,6 +909,23 @@ fn equality_f32_mixorder(bench: &mut test::Bencher)
908909
bench.iter(|| a == b);
909910
}
910911

912+
#[bench]
913+
fn dot_3d_f64_contiguous(bench: &mut test::Bencher)
914+
{
915+
let a: Array3<f64> = Array::zeros((32, 32, 32));
916+
let b: Array2<f64> = Array::zeros((32, 32));
917+
bench.iter(|| a.dot(&b));
918+
}
919+
920+
#[bench]
921+
fn dot_3d_f64_non_contiguous(bench: &mut test::Bencher)
922+
{
923+
let a_base: Array3<f64> = Array::zeros((64, 32, 32));
924+
let a = a_base.slice(s![..;2, .., ..]).to_owned();
925+
let b: Array2<f64> = Array::zeros((32, 32));
926+
bench.iter(|| a.dot(&b));
927+
}
928+
911929
#[bench]
912930
fn dot_f32_16(bench: &mut test::Bencher)
913931
{

0 commit comments

Comments
 (0)