Appease JET

Author: gdalle

DifferentiationInterface/ext/DifferentiationInterfaceForwardDiffExt/utils.jl

@@ -82,14 +82,7 @@ function mypartials!(::Type{T}, ty::NTuple{B}, ydual) where {T,B}
     return ty
 end
 
-# store preparation result with the right input eltype
-struct PrepContext{T<:DI.Prep} <: DI.Context
-    data::T
-end
-
-NotCache = Union{DI.GeneralizedConstant,PrepContext}
-
-_translate(::Type{D}, c::NotCache) where {D<:Dual} = DI.unwrap(c)
+_translate(::Type{D}, c::DI.GeneralizedConstant) where {D<:Dual} = DI.unwrap(c)
 function _translate(::Type{D}, c::DI.Cache) where {D<:Dual}
     c0 = DI.unwrap(c)
     return similar(c0, D)
@@ -102,7 +95,7 @@ function translate(::Type{D}, contexts::NTuple{C,DI.Context}) where {D<:Dual,C}
     return new_contexts
 end
 
-_translate_toprep(::Type{D}, c::NotCache) where {D<:Dual} = nothing
+_translate_toprep(::Type{D}, c::DI.GeneralizedConstant) where {D<:Dual} = nothing
 function _translate_toprep(::Type{D}, c::DI.Cache) where {D<:Dual}
     c0 = DI.unwrap(c)
     return similar(c0, D)
@@ -115,7 +108,7 @@ function translate_toprep(::Type{D}, contexts::NTuple{C,DI.Context}) where {D<:D
     return new_contexts
 end
 
-_translate_prepared(c::NotCache, _pc) = DI.unwrap(c)
+_translate_prepared(c::DI.GeneralizedConstant, _pc) = DI.unwrap(c)
 _translate_prepared(_c::DI.Cache, pc) = pc
 
 function translate_prepared(

DifferentiationInterface/src/second_order/hvp.jl

@@ -551,8 +551,7 @@ end
 
 ## Reverse over reverse
 
-struct ReverseOverReverseHVPPrep{G,PO<:PullbackPrep,PI<:Union{Nothing,PullbackPrep}} <:
-       HVPPrep
+struct ReverseOverReverseHVPPrep{G,PO,PI} <: HVPPrep
     # pullback of gradient
     grad_buffer::G
     outer_pullback_prep::PO

DifferentiationInterface/src/utils/traits.jl

@@ -181,7 +181,9 @@ struct DontPrepareInner <: InnerPreparationBehavior end
 
 inner_preparation_behavior(::AbstractADType) = DontPrepareInner()
 
-function overloaded_input end
+function overloaded_input(optype, f, backend, x, args...)
+    throw(ArgumentError("Just to appease JET"))
+end
 
 ## Conversions
 

DifferentiationInterface/test/Core/Internals/_formalities.jl

@@ -3,6 +3,7 @@
 using Aqua: Aqua
 using DifferentiationInterface
 using ExplicitImports
+using ForwardDiff: ForwardDiff
 using JET: JET
 using JuliaFormatter: JuliaFormatter
 using Test