Skip to content

Commit 89d04e6

Browse files
Alexey Stukalovalyst
authored andcommitted
FIML: optimize Jmu
1 parent 69b16de commit 89d04e6

1 file changed

Lines changed: 4 additions & 3 deletions

File tree

src/loss/ML/FIML.jl

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -57,9 +57,10 @@ function gradient!(JΣ, Jμ, fiml::SemFIMLPattern, pat::SemObservedMissingPatter
5757
else
5858
JΣ_pat = Σ⁻¹ * (I - fiml.μ_diff * μ_diff⨉Σ⁻¹)
5959
end
60-
Jμ_pat = ((2 * n_obs(pat)) * μ_diff⨉Σ⁻¹)'
61-
vec(JΣ)[fiml.∇ind] .+= vec(JΣ_pat)
62-
Jμ[pat.obs_mask] .+= Jμ_pat
60+
@inbounds vec(JΣ)[fiml.∇ind] .+= vec(JΣ_pat)
61+
62+
lmul!(2 * n_obs(pat), μ_diff⨉Σ⁻¹)
63+
@inbounds Jμ[pat.obs_mask] .+= μ_diff⨉Σ⁻¹'
6364
return nothing
6465
end
6566

0 commit comments

Comments
 (0)