fix bench

sonos · Nov 12, 2020 · 437063d · 437063d
1 parent 332b172
commit 437063d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/linalg/benches/conv_for_wavenet_hw.rs b/linalg/benches/conv_for_wavenet_hw.rs
@@ -15,8 +15,7 @@ fn conv(c: &mut Criterion, dilation: usize, pulse: usize, ci: usize, co: usize)
         let mut conv = (tract_linalg::ops().mmm_f32)(co, kernel_offsets.len(), data_offsets.len());
         conv.c_from_data_and_strides(t as _, 1);
         let a =
-            Tensor::from_slice_align(&vec![0.0; conv.a_pack().len()], conv.a_pack().alignment())
-                .unwrap();
+            Tensor::zero_aligned::<f32>(&[conv.a_pack().len()], conv.a_pack().alignment()).unwrap();
         let input = Tensor::zero::<f32>(&[ci, t]).unwrap();
         let mut output = Tensor::zero::<f32>(&[co, t]).unwrap();
         be.iter(move || conv.run(&a.view(), &input.view(), &mut output.view_mut(), &[]).unwrap());