diff --git a/lib/LuxLib/test/others/forwarddiff_tests.jl b/lib/LuxLib/test/others/forwarddiff_tests.jl index cc029668a..3d36fd995 100644 --- a/lib/LuxLib/test/others/forwarddiff_tests.jl +++ b/lib/LuxLib/test/others/forwarddiff_tests.jl @@ -107,8 +107,6 @@ u = randn(Float32, input_dim) |> aType test_jvp_computation(x -> op(x; dims=dim), x, u, ongpu) - test_jvp_computation( - x -> op(x; dims=dim), ComponentArray(; x), u, ongpu) test_jvp_computation( x -> only(Zygote.gradient(x -> sum(op(x; dims=dim)), x)),