test gradient allocations

Author: marius311

test/AllocationsTest.jl

@@ -37,4 +37,14 @@ convert_test_574() = convert(ForwardDiff.Dual{Nothing,ForwardDiff.Dual{Nothing,F
 
 end
 
+@testset "Test gradient allocations" begin
+    @testset "size = $(size(x))" for x in (rand(1), rand(DEFAULT_CHUNK_THRESHOLD+1))
+        y = similar(x)
+        cfg = ForwardDiff.GradientConfig(nothing, x)
+        alloc = @allocated ForwardDiff.gradient!(y, prod, x, cfg)
+        alloc = @allocated ForwardDiff.gradient!(y, prod, x, cfg)
+        @test alloc == 0
+    end
+end
+
 end

test/GradientTest.jl

@@ -158,7 +158,13 @@ end
 
 println("  ...testing GPUArray compatibility (via JLArrays)")
 
-@testset "size = $(size(x))" for x in JLArray.([rand(3), rand(13), rand(3,3), rand(13,13), rand(10,10,10)])
+@testset "size = $(size(x))" for x in JLArray.([
+    rand(1), 
+    rand(DEFAULT_CHUNK_THRESHOLD+1), 
+    rand(1,1), 
+    rand(DEFAULT_CHUNK_THRESHOLD+1,DEFAULT_CHUNK_THRESHOLD+1), 
+    rand(1,1,1)
+])
 
     @test ForwardDiff.gradient(prod, x) isa typeof(x)