Skip to content

Commit 02dd8f9

Browse files
committed
Tests now work on matrices
1 parent b33ca52 commit 02dd8f9

1 file changed

Lines changed: 3 additions & 14 deletions

File tree

  • DifferentiationInterface/test/Back/Enzyme

DifferentiationInterface/test/Back/Enzyme/test.jl

Lines changed: 3 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -16,15 +16,6 @@ check_no_implicit_imports(DifferentiationInterface)
1616

1717
LOGGING = get(ENV, "CI", "false") == "false"
1818

19-
function remove_matrix_inputs(scens::Vector{<:Scenario}) # TODO: remove
20-
if VERSION < v"1.11"
21-
return scens
22-
else
23-
# for https://github.com/EnzymeAD/Enzyme.jl/issues/2071
24-
return filter(s -> s.x isa Union{Number,AbstractVector}, scens)
25-
end
26-
end
27-
2819
backends = [
2920
AutoEnzyme(; mode=nothing),
3021
AutoEnzyme(; mode=Enzyme.Forward),
@@ -77,14 +68,12 @@ end;
7768
end
7869

7970
@testset "Second order" begin
80-
forward_enzyme = AutoEnzyme(; mode=Enzyme.Reverse)
81-
reverse_enzyme = AutoEnzyme(; mode=Enzyme.Reverse)
8271
test_differentiation(
8372
[
8473
AutoEnzyme(),
85-
# SecondOrder(forward_enzyme, reverse_enzyme), # TODO: toggle
86-
SecondOrder(reverse_enzyme, forward_enzyme),
87-
# SecondOrder(forward_enzyme, forward_enzyme), # TODO: toggle
74+
SecondOrder(
75+
AutoEnzyme(; mode=Enzyme.Reverse), AutoEnzyme(; mode=Enzyme.Forward)
76+
),
8877
],
8978
default_scenarios(; include_constantified=true, include_cachified=true);
9079
excluded=FIRST_ORDER,

0 commit comments

Comments
 (0)