Skip to content

Commit 31e635e

Browse files
committed
Fix typo
1 parent b629a0b commit 31e635e

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

src/utils.jl

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -712,8 +712,10 @@ function Base.iterate(it::StatefulBFS, queue = (eltype(it)[(0, it.t)]))
712712
end
713713

714714
function jacobian_wrt_vars(pf::F, p, input_idxs, chunk::C) where {F, C}
715-
dualtype = ForwardDiff.Dual{ForwardDiff.Tag{F, eltype(p)},
716-
eltype(p), ForwardDiff.chunksize(chunk)}
715+
E = eltype(p)
716+
tag = ForwardDiff.Tag(pf, E)
717+
T = typeof(tag)
718+
dualtype = ForwardDiff.Dual{T, E, ForwardDiff.chunksize(chunk)}
717719
p_big = similar(p, dualtype)
718720
copyto!(p_big, p)
719721
p_closure = let pf = pf,
@@ -725,7 +727,7 @@ function jacobian_wrt_vars(pf::F, p, input_idxs, chunk::C) where {F, C}
725727
pf(p_big)
726728
end
727729
end
728-
cfg = ForwardDiff.JacobianConfig(p_closure, p_small, chunk)
729730
p_small = p[input_idxs]
730-
ForwardDiff.jacobian(p_closure, p_small, cfg)
731+
cfg = ForwardDiff.JacobianConfig(p_closure, p_small, chunk, tag)
732+
ForwardDiff.jacobian(p_closure, p_small, cfg, Val(false))
731733
end

0 commit comments

Comments
 (0)