updates

Author: KristofferC

DifferentiationInterface/ext/DifferentiationInterfaceHyperHessiansExt/DifferentiationInterfaceHyperHessiansExt.jl

@@ -40,6 +40,12 @@ function DI.pick_batchsize(backend::DI.AutoHyperHessians, N::Integer)
     return DI.BatchSizeSettings{B}(N)
 end
 
+function DI.threshold_batchsize(backend::DI.AutoHyperHessians, chunksize2::Integer)
+    chunksize1 = backend.chunksize
+    chunksize = isnothing(chunksize1) ? nothing : min(chunksize1, chunksize2)
+    return DI.AutoHyperHessians(; chunksize)
+end
+
 function _translate_toprep(::Type{T}, c::Union{DI.GeneralizedConstant, DI.ConstantOrCache}) where {T}
     return nothing
 end

DifferentiationInterface/test/Back/HyperHessians/test.jl

@@ -19,19 +19,17 @@ for backend in backends
     @test DI.check_inplace(backend)
 end
 
-excluded_ops = [:pushforward, :pullback, :jacobian, :derivative, :gradient]
-
-scenarios = default_scenarios(; include_constantified = true)
+scenarios = default_scenarios(; include_constantified = true, include_cachified = true)
 
 test_differentiation(
     backends, scenarios;
-    excluded = excluded_ops, logging = LOGGING,
+    excluded = FIRST_ORDER, logging = LOGGING,
 )
 
 test_differentiation(
     DI.AutoHyperHessians(), scenarios;
     correctness = false,
     type_stability = safetypestab(:prepared),
-    excluded = excluded_ops,
+    excluded = FIRST_ORDER,
     logging = LOGGING,
 )