Merge pull request #1540 from SciML/static_chunksize

Make the chunksize determination static when static

Author: ChrisRackauckas

src/alg_utils.jl

@@ -171,10 +171,22 @@ function DiffEqBase.prepare_alg(alg::Union{OrdinaryDiffEqAdaptiveImplicitAlgorit
     if typeof(alg) <: OrdinaryDiffEqImplicitExtrapolationAlgorithm
       return alg # remake fails, should get fixed
     else
-      remake(alg,chunk_size=Val{ForwardDiff.pickchunksize(x)}())
+      L = ArrayInterface.known_length(typeof(u0))
+      if L === nothing # dynamic sized
+        cs = ForwardDiff.pickchunksize(x)
+        remake(alg,chunk_size=cs)
+      else # statically sized
+        cs = pick_static_chunksize(Val{L}())
+        remake(alg,chunk_size=cs)
+      end
     end
 end
 
+@generated function pick_static_chunksize(::Val{chunksize}) where chunksize
+  x = ForwardDiff.pickchunksize(chunksize)
+  :(Val{$x}())
+end
+
 function DiffEqBase.prepare_alg(alg::CompositeAlgorithm,u0,p,prob)
   algs = map(alg -> DiffEqBase.prepare_alg(alg, u0, p, prob), alg.algs)
   CompositeAlgorithm(algs, alg.choice_function)