Update

Author: odow

test/rewrite_generic.jl

@@ -480,20 +480,18 @@ function test_allocations_rewrite_unary_minus()
     MA.@rewrite(-sum(x[i] for i in 1:N), move_factors_into_sums = false)
     MA.@rewrite(sum(a * x[i] for i in 1:N), move_factors_into_sums = false)
     sum(-x[i] for i in 1:N)
-    total = @allocated sum(-x[i] for i in 1:N)
-    # actual
     value = @allocated(
         MA.@rewrite(sum(-x[i] for i in 1:N), move_factors_into_sums = false),
     )
-    @test value < total
+    @test value < @allocated sum(-x[i] for i in 1:N)
     value = @allocated(
         MA.@rewrite(-sum(x[i] for i in 1:N), move_factors_into_sums = false),
     )
-    @test value < total
+    @test value < @allocated -sum(x[i] for i in 1:N)
     value = @allocated(
         MA.@rewrite(sum(a * x[i] for i in 1:N), move_factors_into_sums = false),
     )
-    @test value < total
+    @test value < @allocated sum(a * x[i] for i in 1:N)
     return
 end
 

test/utilities.jl

@@ -10,7 +10,7 @@ include("dummy.jl")
 # v1.12.5 off by 8
 _big_int_alloc() = @allocated(BigInt(1))
 
-function alloc_test(f, expected_upper_bound)
+function alloc_test(f::F, expected_upper_bound::Int) where {F<:Function}
     f() # compile
     measured_allocations = @allocated f()
     @test measured_allocations <= expected_upper_bound