Fix overload

Author: gdalle

DifferentiationInterface/src/first_order/mixed_mode.jl

@@ -41,3 +41,10 @@ Appropriate mode type for `MixedMode` backends.
 """
 struct ForwardAndReverseMode <: ADTypes.AbstractMode end
 ADTypes.mode(::MixedMode) = ForwardAndReverseMode()
+
+function threshold_batchsize(backend::MixedMode, B::Integer)
+    return MixedMode(
+        threshold_batchsize(forward_backend(backend), B),
+        threshold_batchsize(reverse_backend(backend), B),
+    )
+end

DifferentiationInterface/src/misc/from_primitive.jl

@@ -3,14 +3,29 @@ abstract type FromPrimitive{inplace} <: AbstractADType end
 check_available(backend::FromPrimitive) = check_available(backend.backend)
 inplace_support(::FromPrimitive{true}) = InPlaceSupported()
 inplace_support(::FromPrimitive{false}) = InPlaceNotSupported()
-function inner_preparation_behavior(backend::FromPrimitive)
-    return inner_preparation_behavior(backend.backend)
+
+function pick_batchsize(backend::FromPrimitive, x_or_y::AbstractArray)
+    return pick_batchsize(backend.backend, x_or_y)
 end
 
 function pick_batchsize(backend::FromPrimitive, N::Integer)
     return pick_batchsize(backend.backend, N)
 end
 
+function inner_preparation_behavior(backend::FromPrimitive)
+    return inner_preparation_behavior(backend.backend)
+end
+
+function overloaded_input(::typeof(pushforward), f, backend::FromPrimitive, x, tx::NTuple)
+    return overloaded_input(pushforward, f, backend.backend, x, tx)
+end
+
+function overloaded_input(
+    ::typeof(pushforward), f!, y, backend::FromPrimitive, x, tx::NTuple
+)
+    return overloaded_input(pushforward, f!, y, backend.backend, x, tx)
+end
+
 """
     AutoForwardFromPrimitive(backend::AbstractADType)
 

DifferentiationInterface/src/utils/batchsize.jl

@@ -112,13 +112,6 @@ function threshold_batchsize(backend::SecondOrder, B::Integer)
     )
 end
 
-function threshold_batchsize(backend::MixedMode, B::Integer)
-    return MixedMode(
-        threshold_batchsize(forward_backend(backend), B),
-        threshold_batchsize(reverse_backend(backend), B),
-    )
-end
-
 """
     reasonable_batchsize(N::Integer, Bmax::Integer)