Skip to content

Commit 9b0f8ae

Browse files
committed
fix: handle AD materialization
1 parent 34602e1 commit 9b0f8ae

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/core/generalized_first_order.jl

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -202,6 +202,8 @@ function SciMLBase.__init(
202202
linesearch_ad = alg.forward_ad === nothing ?
203203
(alg.reverse_ad === nothing ? alg.jacobian_ad :
204204
alg.reverse_ad) : alg.forward_ad
205+
linesearch_ad = get_concrete_forward_ad(
206+
linesearch_ad, prob, False; check_forward_mode = false)
205207
linesearch_cache = init(
206208
prob, alg.linesearch, fu, u; stats, autodiff = linesearch_ad, kwargs...)
207209
GB = :LineSearch

0 commit comments

Comments
 (0)