Consistent setup for round trip and inverse transform tests

xiph · Oct 23, 2023 · 7f01f3c · 7f01f3c
1 parent 021183f
commit 7f01f3c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/asm/shared/transform/inverse.rs b/src/asm/shared/transform/inverse.rs
@@ -149,7 +149,10 @@ pub mod test {
     for sub_h in 0..sub_h_iterations {
       let mut src_storage = [T::zero(); 64 * 64];
       let src = &mut src_storage[..tx_size.area()];
-      let mut dst = Plane::from_slice(&[T::zero(); 64 * 64], 64);
+      let mut dst = Plane::from_slice(
+        &[T::zero(); 64 * 64][..tx_size.area()],
+        tx_size.width(),
+      );
       let mut res_storage: Aligned<[MaybeUninit<i16>; 64 * 64]> =
         unsafe { Aligned::uninitialized() };
       let res = &mut res_storage.data[..tx_size.area()];

diff --git a/src/transform/mod.rs b/src/transform/mod.rs
@@ -485,9 +485,8 @@ mod test {
     let coeff_area: usize = av1_get_coded_tx_size(tx_size).area();
     let mut src_storage = [T::cast_from(0); 64 * 64];
     let src = &mut src_storage[..tx_size.area()];
-    // dynamic allocation: test
     let mut dst = Plane::from_slice(
-      &vec![T::cast_from(0); tx_size.area()],
+      &[T::zero(); 64 * 64][..tx_size.area()],
       tx_size.width(),
     );
     let mut res_storage = [0i16; 64 * 64];