@@ -287,20 +287,14 @@ end
287
287
288
288
struct SensitivityADPassThrough <: SciMLBase.DEAlgorithm end
289
289
290
- ZygoteRules. @adjoint function solve_up (prob,sensealg:: Union{Nothing,AbstractSensitivityAlgorithm} ,
291
- u0,p,args... ;
292
- kwargs... )
293
- _solve_adjoint (prob,sensealg,u0,p,args... ;kwargs... )
294
- end
295
-
296
290
function ChainRulesCore. frule (:: typeof (solve_up),prob,
297
291
sensealg:: Union{Nothing,AbstractSensitivityAlgorithm} ,
298
292
u0,p,args... ;
299
293
kwargs... )
300
294
_solve_forward (prob,sensealg,u0,p,args... ;kwargs... )
301
295
end
302
296
303
- function ChainRulesCore. rrule (:: typeof (solve_up),prob,
297
+ function ChainRulesCore. rrule (:: typeof (solve_up),prob:: SciMLBase.DEProblem ,
304
298
sensealg:: Union{Nothing,AbstractSensitivityAlgorithm} ,
305
299
u0,p,args... ;
306
300
kwargs... )
314
308
@deprecate concrete_solve (prob:: SciMLBase.DEProblem ,alg:: Union{SciMLBase.DEAlgorithm,Nothing} ,
315
309
u0= prob. u0,p= prob. p,args... ;kwargs... ) solve (prob,alg,args... ;u0= u0,p= p,kwargs... )
316
310
317
- ZygoteRules. @adjoint function concrete_solve (prob:: SciMLBase.DEProblem ,
318
- alg:: Union{SciMLBase.DEAlgorithm,Nothing} ,
319
- u0= prob. u0,p= prob. p,args... ;
320
- sensealg= nothing ,
321
- kwargs... )
322
- _concrete_solve_adjoint (prob,alg,sensealg,u0,p,args... ;kwargs... )
323
- end
324
-
325
311
function _solve_adjoint (prob,sensealg,u0,p,args... ;kwargs... )
326
312
if isempty (args)
327
313
_concrete_solve_adjoint (prob,nothing ,sensealg,u0,p;kwargs... )
0 commit comments