Fixes

Author: gdalle

DifferentiationInterface/ext/DifferentiationInterfacePolyesterForwardDiffExt/onearg.jl

@@ -1,8 +1,7 @@
 
 ## Pushforward
 
-struct PolyesterForwardDiffOneArgPushforwardPrep{SIG,P} <:
-       PolyesterForwardDiffOneArgPushforwardPrep{SIG}
+struct PolyesterForwardDiffOneArgPushforwardPrep{SIG,P} <: DI.PushforwardPrep{SIG}
     _sig::Val{SIG}
     single_threaded_prep::P
 end

DifferentiationInterface/ext/DifferentiationInterfaceTrackerExt/DifferentiationInterfaceTrackerExt.jl

@@ -35,7 +35,7 @@ function DI.prepare_pullback_same_point(
     ty::NTuple,
     contexts::Vararg{DI.GeneralizedConstant,C},
 ) where {C}
-    _sig = DI.signature(f, prep, backend, x, ty, contexts...; strict)
+    _sig = DI.signature(f, prep, backend, x, ty, contexts...; strict=DI.is_strict(prep))
     DI.check_prep(f, prep, backend, x, ty, contexts...)
     y, pb = forward(f, x, map(DI.unwrap, contexts)...)
     return TrackerPullbackPrepSamePoint(_sig, y, pb)

DifferentiationInterface/src/utils/prep.jl

@@ -77,13 +77,21 @@ end
 is_strict(::Prep{Nothing}) = Val(false)
 is_strict(::Prep) = Val(true)
 
-function inconsistent_signatures_error(SIG, RUNTIME_SIG)
+struct PreparationMismatchError{SIG,RUNTIME_SIG} <: Exception end
+
+function PreparationMismatchError(::Type{SIG}, ::Type{RUNTIME_SIG}) where {SIG,RUNTIME_SIG}
+    return PreparationMismatchError{SIG,RUNTIME_SIG}()
+end
+
+function Base.showerror(
+    io::IO, e::PreparationMismatchError{SIG,RUNTIME_SIG}
+) where {SIG,RUNTIME_SIG}
     msg = """
     Inconsistent signatures:
      - at preparation time: $SIG
      - at execution time: $RUNTIME_SIG
     """
-    return ArgumentError(msg)
+    return print(io, msg)
 end
 
 function signature(
@@ -138,7 +146,7 @@ function check_prep(
     if SIG !== Nothing
         RUNTIME_SIG = typeof((f, backend, x, contexts))
         if SIG != RUNTIME_SIG
-            throw(inconsistent_signatures_error(SIG, RUNTIME_SIG))
+            throw(PreparationMismatchError(SIG, RUNTIME_SIG))
         end
     end
 end
@@ -149,7 +157,7 @@ function check_prep(
     if SIG !== Nothing
         RUNTIME_SIG = typeof((f!, y, backend, x, contexts))
         if SIG != RUNTIME_SIG
-            throw(inconsistent_signatures_error(SIG, RUNTIME_SIG))
+            throw(PreparationMismatchError(SIG, RUNTIME_SIG))
         end
     end
 end
@@ -160,7 +168,7 @@ function check_prep(
     if SIG !== Nothing
         RUNTIME_SIG = typeof((f, backend, x, t, contexts))
         if SIG != RUNTIME_SIG
-            throw(inconsistent_signatures_error(SIG, RUNTIME_SIG))
+            throw(PreparationMismatchError(SIG, RUNTIME_SIG))
         end
     end
 end
@@ -171,7 +179,7 @@ function check_prep(
     if SIG !== Nothing
         RUNTIME_SIG = typeof((f!, y, backend, x, t, contexts))
         if SIG != RUNTIME_SIG
-            throw(inconsistent_signatures_error(SIG, RUNTIME_SIG))
+            throw(PreparationMismatchError(SIG, RUNTIME_SIG))
         end
     end
 end