Fix source

Author: ErikQQY

Project.toml

@@ -18,16 +18,17 @@ StatsBase = "2913bbd2-ae8a-5f71-8c99-4fb6c76f3a91"
 StatsFuns = "4c63d2b9-4356-54db-8cca-17b64c39e42c"
 VecTargets = "8a639fad-7908-4fe4-8003-906e9297f002"
 
+[sources]
+VecTargets = {url = "https://github.com/chalk-lab/VecTargets.jl", rev = "main"}
+
+
 [weakdeps]
 ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b"
 ComponentArrays = "b0b7db55-cfe3-40fc-9ded-d10e2dbeff66"
 CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
 MCMCChains = "c7f686f2-ff18-58e9-bc7b-31028e88f75d"
 OrdinaryDiffEq = "1dea7af3-3e70-54e6-95c3-0bf5283fa5ed"
 
-[sources]
-VecTargets = {rev = "main", url = "https://github.com/chalk-lab/VecTargets.jl"}
-
 [extensions]
 AdvancedHMCADTypesExt = "ADTypes"
 AdvancedHMCComponentArraysExt = "ComponentArrays"

src/riemannian/hamiltonian.jl

@@ -93,11 +93,11 @@ function ∂H∂θ_cache(
         -(∂ℓπ∂θ - 1 / 2 * diag(term_1_cached * ∂H∂θ) + 1 / 2 * diag(term_2 * ∂H∂θ))
     else
         -mapreduce(vcat, 1:d) do i
-        ∂H∂θᵢ = ∂H∂θ[:, :, i]
-        # ∂ℓπ∂θ[i] - 1 / 2 * tr(term_1_cached * ∂H∂θᵢ) + 1 / 2 * M' * (J .* (Q' * ∂H∂θᵢ * Q)) * M # (v1)
-        # NOTE Some further optimization can be done here: cache the 1st product all together
-        ∂ℓπ∂θ[i] - 1 / 2 * tr(term_1_cached * ∂H∂θᵢ) + 1 / 2 * tr(term_2_cached * ∂H∂θᵢ) # (v2) cache friendly
-    end
+            ∂H∂θᵢ = ∂H∂θ[:, :, i]
+            # ∂ℓπ∂θ[i] - 1 / 2 * tr(term_1_cached * ∂H∂θᵢ) + 1 / 2 * M' * (J .* (Q' * ∂H∂θᵢ * Q)) * M # (v1)
+            # NOTE Some further optimization can be done here: cache the 1st product all together
+            ∂ℓπ∂θ[i] - 1 / 2 * tr(term_1_cached * ∂H∂θᵢ) + 1 / 2 * tr(term_2_cached * ∂H∂θᵢ) # (v2) cache friendly
+        end
     end
 
     dv = DualValue(ℓπ, g)