diff --git a/crates/burn-jit/src/tests/conv2d.rs b/crates/burn-jit/src/tests/conv2d.rs index 061ab54e65..8b3912d1d1 100644 --- a/crates/burn-jit/src/tests/conv2d.rs +++ b/crates/burn-jit/src/tests/conv2d.rs @@ -125,25 +125,4 @@ mod tests { into_data_sync::(output) .assert_approx_eq(&into_data_sync::(output_ref), 4); } - - #[test] - fn nchw_to_nhwc_should_match_into_contiguos() { - let test_device = Default::default(); - let input = - Tensor::::random([4, 72, 53, 56], Distribution::Default, &test_device); - - type Float = ::FloatElem; - - let output = nchw_to_nhwc::(input.clone().into_primitive().tensor()); - let output_ref = into_contiguous( - input - .clone() - .permute([0, 2, 3, 1]) - .into_primitive() - .tensor(), - ); - - into_data_sync::(output) - .assert_approx_eq(&into_data_sync::(output_ref), 1); - } }