Merge pull request #101 from LilithHafner/lh/lbc-sizemap-int

Continue to support sizemaps that return Int in LazyBufferCache

Author: ChrisRackauckas

src/PreallocationTools.jl

@@ -209,6 +209,7 @@ struct LazyBufferCache{F <: Function}
     LazyBufferCache(f::F = identity) where {F <: Function} = new{F}(Dict(), f) # start with empty dict
 end
 
+similar_type(x::AbstractArray, s::Integer) = similar_type(x, (s,))
 function similar_type(x::AbstractArray{T}, s::NTuple{N, Integer}) where {T, N}
     # The compiler is smart enough to not allocate
     # here for simple types like Array and SubArray

test/general_lbc.jl

@@ -41,6 +41,12 @@ y = view(x, 1:900)
 @inferred cache[y]
 @test 0 == @allocated cache[y]
 
+cache_17 = LazyBufferCache(Returns(17))
+x = 1:10
+@inferred cache_17[x]
+@test 0 == @allocated cache_17[x]
+@test size(cache_17[x]) == (17,)
+
 cache = GeneralLazyBufferCache(T -> Vector{T}(undef, 1000))
 # GeneralLazyBufferCache is documented not to infer.
 # @inferred cache[Float64]