@inline fixes single-alloc

src/WithAlloc.jl

@@ -5,11 +5,11 @@ export whatalloc, @withalloc1, @withalloc
 
 function whatalloc end 
 
-function _bumper_alloc(allocinfo::Tuple{Type, Vararg{Int, N}})  where {N} 
+@inline function _bumper_alloc(allocinfo::Tuple)
    (Bumper.alloc!(Bumper.default_buffer(), allocinfo... ), )
 end
 
-function _bumper_alloc(allocinfo::NTuple{N, <: Tuple}) where {N} 
+@inline function _bumper_alloc(allocinfo::NTuple{N, <: Tuple}) where {N} 
    ntuple(i -> Bumper.alloc!(Bumper.default_buffer(), allocinfo[i]...), N)
 end
 

test/_readme.jl

@@ -72,6 +72,18 @@ function alloctest1(B, C)
    end
 end 
 
+function alloctest2(B, C) 
+   @no_escape begin 
+      # allocinfo = WithAlloc.whatalloc(mymul!, B, C)
+      # A = ( (@alloc(allocinfo...)), )
+      # A = WithAlloc._bumper_alloc(allocinfo)
+      # A = (Bumper.alloc!(Bumper.default_buffer(), allocinfo... ), )
+      # s3 = sum( mymul!(A..., B, C) )
+      s3 = sum( @withalloc mymul!(B, C) )
+   end
+end 
+
+
 function alloctest(B, C) 
    @no_escape begin 
       s3 = sum( @withalloc mymul!(B, C) )
@@ -81,3 +93,4 @@ end
 B = randn(5,10); C = randn(10, 3)
 @btime alloctest($B, $C)       #   243.056 ns (2 allocations: 64 bytes)
 @btime alloctest1($B, $C)      #   106.551 ns (0 allocations: 0 bytes)
+@btime alloctest2($B, $C)

test/test1.jl

@@ -32,6 +32,19 @@ end
    s3 = sum(A3) 
 end
 @test s3 ≈ s1
+
+## allocation test 
+
+alloctest(B, C) = (@no_escape begin sum( @withalloc mymul!(B, C) ); end)
+
+nalloc = let    
+   B = randn(5,10)
+   C = randn(10, 3)
+   @allocated alloctest(B, C)
+end
+
+@test nalloc == 0
+
 
 ## 
 
@@ -66,3 +79,16 @@ end
 
 @test sb ≈ s
 
+## allocation test
+
+alloctest2(B, C, D) = 
+         (@no_escape begin sum(sum.( @withalloc mymul2!(B, C, D) )); end)
+
+nalloc2 = let    
+   B = randn(5,10)
+   C = randn(10, 3)
+   D = randn(10, 5)
+   @allocated alloctest2(B, C, D)
+end
+
+@btime alloctest2($B, $C, $D)