Skip to content

Commit 95e24b9

Browse files
committed
bench: Fix the multiply-add benchmark
1 parent 87d03ca commit 95e24b9

File tree

1 file changed

+2
-8
lines changed

1 file changed

+2
-8
lines changed

benches/bench1.rs

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -372,15 +372,9 @@ fn add_2d_f32_blas(bench: &mut test::Bencher)
372372
#[bench]
373373
fn muladd_2d_f32_regular(bench: &mut test::Bencher)
374374
{
375-
use rblas::Axpy;
376-
use rblas::attribute::Transpose;
377-
use ndarray::blas::AsBlas;
378-
let mut a = OwnedArray::<f32, _>::zeros((64, 64));
379-
let b = OwnedArray::<f32, _>::zeros((64, 64));
375+
let mut av = OwnedArray::<f32, _>::zeros((64, 64));
376+
let bv = OwnedArray::<f32, _>::zeros((64, 64));
380377
let scalar = 3.1415926535;
381-
let len = a.len();
382-
let mut av = a.view_mut().into_shape(len).unwrap();
383-
let bv = b.view().into_shape(len).unwrap();
384378
bench.iter(|| {
385379
av.zip_mut_with(&bv, |a, &b| *a += scalar * b);
386380
});

0 commit comments

Comments
 (0)