From cba9d1155f5e082cf50866df99104fae2eb03489 Mon Sep 17 00:00:00 2001 From: Vedant Puri Date: Sun, 12 Mar 2023 16:48:10 -0400 Subject: [PATCH] Test ZygoteHesVec --- test/test_jaches_products.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/test_jaches_products.jl b/test/test_jaches_products.jl index 1efcf565..d783674f 100644 --- a/test/test_jaches_products.jl +++ b/test/test_jaches_products.jl @@ -127,7 +127,7 @@ update_coefficients!(L, v, nothing, 0.0) @test mul!(dy, L, v)≈numback_hesvec(g, v, v) rtol=1e-2 dy=rand(N);_dy=copy(dy);@test mul!(dy,L,v,a,b) ≈ a*numback_hesvec(g,x,v) + b*_dy rtol=1e-2 -L = HesVec(g, x) +L = ZygoteHesVec(g, x) @test L * x ≈ autoback_hesvec(g, x, x) @test L * v ≈ autoback_hesvec(g, x, v) @test mul!(dy, L, v)≈autoback_hesvec(g, x, v) rtol=1e-8