test allocations

Author: dpo

test/Project.toml

@@ -4,3 +4,8 @@ LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 NLPModels = "a4795742-8479-5a88-8948-cc11e1c8c1a6"
 NLPModelsTest = "7998695d-6960-4d3a-85c4-e1bceb8cd856"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
+
+[compat]
+CUDA = "5.8"
+NLPModels = "0.21"
+NLPModelsTest = "0.10"

test/allocs.jl

@@ -0,0 +1,15 @@
+@testset "Allocations" begin
+  @testset "Allocations for hs5" begin
+    @testset "hs5" begin
+      include("problems/hs5.jl")
+      test_zero_allocations(hs5())
+    end
+  end
+
+  @testset "Allocations for hs6" begin
+    @testset "hs6" begin
+      include("problems/hs6.jl")
+      test_zero_allocations(hs6())
+    end
+  end
+end

test/problems/hs5.jl

@@ -19,16 +19,23 @@ hs5() = hs5(Float64)
 hs5(::Type{T}) where {T <: Number} = hs5(Vector{T})
 function hs5(::Type{V}) where {V}
   T = eltype(V)
-  hprod(hv, x, v; obj_weight = one(T)) =
-    (hv .= (-sin(x[1] + x[2]) * (v[1] + v[2]) .+ 2 * V([v[1] - v[2]; v[2] - v[1]])) * obj_weight)
+  hprod(hv, x, v; obj_weight = one(T)) = begin
+    hv[1] = v[1] - v[2]
+    hv[2] = v[2] - v[1]
+    hv .*= 2 * obj_weight
+    hv .+= -sin(x[1] + x[2]) * (v[1] + v[2])
+  end
   hess_coord(vals, x; obj_weight = one(T)) = begin
     vals[1] = vals[3] = -sin(x[1] + x[2]) + 2
     vals[2] = -sin(x[1] + x[2]) - 2
     vals .*= obj_weight
   end
   f(x) = sin(x[1] + x[2]) + (x[1] - x[2])^2 - 3x[1] / 2 + 5x[2] / 2 + 1
-  grad(gx, x) =
-    (gx .= cos(x[1] + x[2]) .+ 2 * (x[1] - x[2]) * V([1; -1]) + V([-15 // 10; 25 // 10]))
+  grad(gx, x) = begin
+    gx[1] = -15 // 10 + 2 * (x[1] - x[2])
+    gx[2] = 25 // 10 - 2 * (x[1] - x[2])
+    gx .+= cos(x[1] + x[2])
+  end
   objgrad(gx, x) = f(x), grad(gx, x)
   return NLPModel(
     fill!(V(undef, 2), 0),

test/problems/hs6.jl

@@ -25,7 +25,7 @@ function hs6(::Type{V}) where {V}
   return NLPModel(
     V([-12 // 10; 1]),
     x -> (1 - x[1])^2;
-    grad = (gx, x) -> gx .= V([2 * (x[1] - 1); 0]),
+    grad = (gx, x) -> (gx[1] = 2 * (x[1] - 1); gx[2] = 0; gx),
     # objgrad explicitly not implemented
     hprod = hprod,
     hess_coord = ([1], [1], hess_coord),

test/runtests.jl

@@ -4,3 +4,4 @@ using CUDA, Test
 
 include("nlpmodelstest.jl")
 include("notimplemented.jl")
+include("allocs.jl")