force initialization of packed tensor

sonos · Jan 30, 2024 · 3557d9c · 3557d9c
1 parent 90304fd
commit 3557d9c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/core/src/ops/matmul/pack.rs b/core/src/ops/matmul/pack.rs
@@ -34,8 +34,7 @@ impl EvalOp for MatMatMulPack {
         unsafe {
             let output_shape = self.output_shape(b.shape());
             let mut packed =
-                Tensor::uninitialized_aligned_dt(dt, &output_shape, self.packer.alignment())
-                    .unwrap();
+                Tensor::zero_aligned_dt(dt, &output_shape, self.packer.alignment()).unwrap();
             let mut bc_shape: TVec<usize> = b.shape().into();
             bc_shape[self.k_axis] = 1;
             bc_shape[self.mn_axis] = 1;