File tree Expand file tree Collapse file tree
DifferentiationInterface/test/Back/Enzyme Expand file tree Collapse file tree Original file line number Diff line number Diff line change 1+ # see https://github.com/JuliaDiff/DifferentiationInterface.jl/issues/855
2+
13using Pkg
24Pkg. add (" Enzyme" )
35
5557 )
5658
5759 test_differentiation (
58- backends[2 ],
60+ backends[2 : 3 ],
5961 default_scenarios (;
6062 include_normal= false ,
6163 include_cachified= true ,
7577end
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- )
9593end
9694
9795@testset " Sparse" begin
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,
You can’t perform that action at this time.
0 commit comments