Skip to content

Commit 77b8c4d

Browse files
committed
test: untoggle some passing Enzyme tests
1 parent 5257e7d commit 77b8c4d

1 file changed

Lines changed: 10 additions & 17 deletions

File tree

  • DifferentiationInterface/test/Back/Enzyme

DifferentiationInterface/test/Back/Enzyme/test.jl

Lines changed: 10 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
# see https://github.com/JuliaDiff/DifferentiationInterface.jl/issues/855
2+
13
using Pkg
24
Pkg.add("Enzyme")
35

@@ -55,7 +57,7 @@ end;
5557
)
5658

5759
test_differentiation(
58-
backends[2],
60+
backends[2:3],
5961
default_scenarios(;
6062
include_normal=false,
6163
include_cachified=true,
@@ -75,23 +77,19 @@ end;
7577
end
7678

7779
@testset "Second order" begin
80+
forward_enzyme = AutoEnzyme(; mode=Enzyme.Reverse)
81+
reverse_enzyme = AutoEnzyme(; mode=Enzyme.Reverse)
7882
test_differentiation(
7983
[
8084
AutoEnzyme(),
81-
SecondOrder(
82-
AutoEnzyme(; mode=Enzyme.Reverse), AutoEnzyme(; mode=Enzyme.Forward)
83-
),
85+
# SecondOrder(forward_enzyme, reverse_enzyme), # TODO: toggle
86+
SecondOrder(reverse_enzyme, forward_enzyme),
87+
# SecondOrder(forward_enzyme, forward_enzyme), # TODO: toggle
8488
],
85-
remove_matrix_inputs(default_scenarios(; include_constantified=true));
89+
default_scenarios(; include_constantified=true, include_cachified=true);
8690
excluded=FIRST_ORDER,
8791
logging=LOGGING,
8892
)
89-
90-
test_differentiation(
91-
AutoEnzyme(; mode=Enzyme.Forward);
92-
excluded=vcat(FIRST_ORDER, [:hessian, :hvp]),
93-
logging=LOGGING,
94-
)
9593
end
9694

9795
@testset "Sparse" begin
@@ -100,12 +98,7 @@ end
10098
if VERSION < v"1.11"
10199
sparse_scenarios()
102100
else
103-
filter(sparse_scenarios()) do s
104-
# for https://github.com/EnzymeAD/Enzyme.jl/issues/2168
105-
(s.x isa AbstractVector) &&
106-
(s.f != DIT.sumdiffcube) &&
107-
(s.f != DIT.sumdiffcube_mat)
108-
end
101+
filter(s -> s.x isa AbstractVector, sparse_scenarios())
109102
end;
110103
sparsity=true,
111104
logging=LOGGING,

0 commit comments

Comments
 (0)