see if full specialize helps

Author: ChrisRackauckas

Project.toml

@@ -147,7 +147,6 @@ julia = "1.10"
 
 [extras]
 AlgebraicMultigrid = "2169fc97-5a83-5252-b627-83903c6c433c"
-AllocCheck = "9b6a8646-10ed-4001-bbdc-1d2f46dfbb1a"
 Calculus = "49dc2e85-a5d0-5ad3-a950-438e2897f1b9"
 ComponentArrays = "b0b7db55-cfe3-40fc-9ded-d10e2dbeff66"
 DiffEqCallbacks = "459566f4-90b8-5000-8ac3-15dfb0a30def"
@@ -172,4 +171,4 @@ Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 Unitful = "1986cc42-f94f-5a68-af5c-568840ba703d"
 
 [targets]
-test = ["Calculus", "AllocCheck", "ComponentArrays", "Symbolics", "AlgebraicMultigrid", "IncompleteLU", "DiffEqCallbacks", "DiffEqDevTools", "ODEProblemLibrary", "ElasticArrays", "InteractiveUtils", "ParameterizedFunctions", "PoissonRandom", "Printf", "Random", "ReverseDiff", "SafeTestsets", "SparseArrays", "Statistics", "Test", "Unitful", "ModelingToolkit", "Pkg", "NLsolve"]
+test = ["Calculus", "ComponentArrays", "Symbolics", "AlgebraicMultigrid", "IncompleteLU", "DiffEqCallbacks", "DiffEqDevTools", "ODEProblemLibrary", "ElasticArrays", "InteractiveUtils", "ParameterizedFunctions", "PoissonRandom", "Printf", "Random", "ReverseDiff", "SafeTestsets", "SparseArrays", "Statistics", "Test", "Unitful", "ModelingToolkit", "Pkg", "NLsolve"]

test/integrators/callback_allocation_tests.jl

@@ -1,5 +1,5 @@
 using OrdinaryDiffEq, Test
-using OrdinaryDiffEqCore, AllocCheck
+using OrdinaryDiffEqCore
 
 # Setup a simple ODE problem with several callbacks (to test LLVM code gen)
 # We will manually trigger the first callback and check its allocations.
@@ -32,15 +32,17 @@ cbs = CallbackSet(ContinuousCallback(cond_1, cb_affect!),
     ContinuousCallback(cond_9, cb_affect!))
 
 integrator = init(
-    ODEProblem(f!, [0.8, 1.0], (0.0, 100.0), [0, 0]), Tsit5(), callback = cbs,
+    ODEProblem{true, SciMLBase.FullSpecialize}(f!, [0.8, 1.0], 
+    (0.0, 100.0), [0, 0]), Tsit5(), callback = cbs,
     save_on = false);
 # Force a callback event to occur so we can call handle_callbacks! directly.
 # Step to a point where u[1] is still > 0.5, so we can force it below 0.5 and
 # call handle callbacks
 step!(integrator, 0.1, true)
 
-@check_allocs function handle_allocs(integrator)
+function handle_allocs(integrator)
     integrator.u[1] = 0.4
-    OrdinaryDiffEqCore.handle_callbacks!(integrator)
+    @allocations OrdinaryDiffEqCore.handle_callbacks!(integrator)
 end
-handle_allocs(integrator)
+handle_allocs(integrator)
+@test handle_allocs(integrator) == 0