Modify test of neg_energy to avoid error from rounding differences

between dot and sum(abs2, ...) on 32 bit.
TuringLang · Nov 1, 2019 · cfcd519 · cfcd519
1 parent 734ed26
commit cfcd519
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/hamiltonian.jl b/test/hamiltonian.jl
@@ -27,7 +27,7 @@ end
         r_init = randn(D)
 
         h = Hamiltonian(UnitEuclideanMetric(D), ℓπ, ∂ℓπ∂θ)
-        @test -AdvancedHMC.neg_energy(h, r_init, θ_init) == dot(r_init, r_init) / 2
+        @test -AdvancedHMC.neg_energy(h, r_init, θ_init) == sum(abs2, r_init) / 2
         @test AdvancedHMC.∂H∂r(h, r_init) == r_init
 
         M⁻¹ = ones(D) + abs.(randn(D))