Skip to content

Commit 9eb1949

Browse files
committed
Update tests
1 parent 64da046 commit 9eb1949

File tree

4 files changed

+21
-8
lines changed

4 files changed

+21
-8
lines changed

test/enzyme.jl

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -99,3 +99,10 @@ include("nlp/basic.jl")
9999
include("nls/basic.jl")
100100
include("nlp/nlpmodelstest.jl")
101101
include("nls/nlpmodelstest.jl")
102+
103+
const test_enzyme = true
104+
105+
include("sparse_jacobian.jl")
106+
include("sparse_jacobian_nls.jl")
107+
# include("sparse_hessian.jl")
108+
# include("sparse_hessian_nls.jl")

test/runtests.jl

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,8 @@ using ADNLPModels, ManualNLPModels, NLPModels, NLPModelsModifiers, NLPModelsTest
44
using ADNLPModels:
55
gradient, gradient!, jacobian, hessian, Jprod!, Jtprod!, directional_second_derivative, Hvprod!
66

7+
const test_enzyme = false
8+
79
@testset "Test sparsity pattern of Jacobian and Hessian" begin
810
f(x) = sum(x .^ 2)
911
c(x) = x

test/sparse_jacobian.jl

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,9 @@
1-
list_sparse_jac_backend =
2-
((ADNLPModels.SparseADJacobian, Dict()),
3-
(ADNLPModels.SparseEnzymeADJacobian, Dict()),
4-
(ADNLPModels.ForwardDiffADJacobian, Dict()))
1+
if test_enzyme
2+
list_sparse_jac_backend = ((ADNLPModels.SparseEnzymeADJacobian, Dict()),)
3+
else
4+
list_sparse_jac_backend = ((ADNLPModels.SparseADJacobian, Dict()),
5+
(ADNLPModels.ForwardDiffADJacobian, Dict()))
6+
end
57

68
dt = (Float32, Float64)
79

test/sparse_jacobian_nls.jl

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,9 @@
1-
list_sparse_jac_backend =
2-
((ADNLPModels.SparseADJacobian, Dict()),
3-
(ADNLPModels.SparseEnzymeADJacobian, Dict()),
4-
(ADNLPModels.ForwardDiffADJacobian, Dict()))
1+
if test_enzyme
2+
list_sparse_jac_backend = ((ADNLPModels.SparseEnzymeADJacobian, Dict()),)
3+
else
4+
list_sparse_jac_backend = ((ADNLPModels.SparseADJacobian, Dict()),
5+
(ADNLPModels.ForwardDiffADJacobian, Dict()))
6+
end
57

68
dt = (Float32, Float64)
79

0 commit comments

Comments
 (0)