unroll the chunk picking

JuliaDiff · Aug 15, 2024 · 330440e · 330440e
1 parent eb6f1bd
commit 330440e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/prelude.jl b/src/prelude.jl
@@ -9,7 +9,7 @@ struct Chunk{N} end
 
 function Chunk(input_length::Integer, threshold::Integer = DEFAULT_CHUNK_THRESHOLD)
     N = pickchunksize(input_length, threshold)
-    return Chunk{N}()
+    Base.@nif 12 d->(N == d) d->(Chunk{d}()) d->(Chunk{N}())
 end
 
 function Chunk(x::AbstractArray, threshold::Integer = DEFAULT_CHUNK_THRESHOLD)