diff --git a/docs/src/reference.md b/docs/src/reference.md index edec428..01ce08b 100644 --- a/docs/src/reference.md +++ b/docs/src/reference.md @@ -10,4 +10,4 @@ explicit_unsteady_cache implicit_unsteady implicit_unsteady_cache provide_rule -``` \ No newline at end of file +``` diff --git a/src/eigenvalues.jl b/src/eigenvalues.jl index a0d70b5..99fe271 100644 --- a/src/eigenvalues.jl +++ b/src/eigenvalues.jl @@ -20,9 +20,9 @@ Make eigenvalue problems AD compatible with ForwardDiff and ReverseDiff implicit_eigval(A, B, eigsolve) = eigsolve(A, B)[1] # If no AD, just solve normally. Returns only eigenvalue. # forward cases -implicit_eigval(A::AbstractArray{<:ForwardDiff.Dual{T}}, B::AbstractArray{<:ForwardDiff.Dual{T}}, eigsolve) where {T} = eigval_fwd(A, B, eigsolve) -implicit_eigval(A, B::AbstractArray{<:ForwardDiff.Dual{T}}, eigsolve) where {T} = eigval_fwd(A, B, eigsolve) -implicit_eigval(A::AbstractArray{<:ForwardDiff.Dual{T}}, B, eigsolve) where {T} = eigval_fwd(A, B, eigsolve) +implicit_eigval(A::AbstractArray{<:ForwardDiff.Dual}, B::AbstractArray{<:ForwardDiff.Dual}, eigsolve) = eigval_fwd(A, B, eigsolve) +implicit_eigval(A, B::AbstractArray{<:ForwardDiff.Dual}, eigsolve) = eigval_fwd(A, B, eigsolve) +implicit_eigval(A::AbstractArray{<:ForwardDiff.Dual}, B, eigsolve) = eigval_fwd(A, B, eigsolve) # reverse cases implicit_eigval(A::Union{ReverseDiff.TrackedArray, AbstractArray{<:ReverseDiff.TrackedReal}}, B::Union{ReverseDiff.TrackedArray, AbstractArray{<:ReverseDiff.TrackedReal}}, eigsolve) = eigval_rev(A, B, eigsolve)