Skip to content

Commit 6d1048d

Browse files
committed
Small improvement to ThreadSafeVarInfo
1 parent 00cd304 commit 6d1048d

File tree

1 file changed

+10
-4
lines changed

1 file changed

+10
-4
lines changed

src/threadsafe.jl

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -182,10 +182,16 @@ end
182182
function resetlogp!!(vi::ThreadSafeVarInfo)
183183
vi = Accessors.@set vi.varinfo = resetlogp!!(vi.varinfo)
184184
for i in eachindex(vi.accs_by_thread)
185-
vi.accs_by_thread[i] = map_accumulator(zero, vi.accs_by_thread[i], Val(:LogPrior))
186-
vi.accs_by_thread[i] = map_accumulator(
187-
zero, vi.accs_by_thread[i], Val(:LogLikelihood)
188-
)
185+
if hasacc(vi, Val(:LogPrior))
186+
vi.accs_by_thread[i] = map_accumulator(
187+
zero, vi.accs_by_thread[i], Val(:LogPrior)
188+
)
189+
end
190+
if hasacc(vi, Val(:LogLikelihood))
191+
vi.accs_by_thread[i] = map_accumulator(
192+
zero, vi.accs_by_thread[i], Val(:LogLikelihood)
193+
)
194+
end
189195
end
190196
return vi
191197
end

0 commit comments

Comments
 (0)