diff --git a/src/alg_utils.jl b/src/alg_utils.jl index 4d22b89e..ee4b86a5 100644 --- a/src/alg_utils.jl +++ b/src/alg_utils.jl @@ -24,7 +24,7 @@ _alg_autodiff(::StochasticDiffEqNewtonAdaptiveAlgorithm{T, AD}) where {T, AD} = _alg_autodiff(::StochasticDiffEqJumpNewtonAdaptiveAlgorithm{T, AD}) where {T, AD} = Val{AD}() _alg_autodiff(::StochasticDiffEqJumpNewtonDiffusionAdaptiveAlgorithm{T, AD}) where {T, AD} = Val{AD}() -function OrdinaryDiffEqCore.alg_autodiff(alg) +function OrdinaryDiffEqCore.alg_autodiff(alg::Union{StochasticDiffEqAlgorithm, StochasticDiffEqRODEAlgorithm}) ad = _alg_autodiff(alg) if ad == Val(false) return ADTypes.AutoFiniteDiff()