File tree Expand file tree Collapse file tree 2 files changed +6
-24
lines changed Expand file tree Collapse file tree 2 files changed +6
-24
lines changed Original file line number Diff line number Diff line change @@ -63,7 +63,7 @@ Moreover, each context type is supported by a specific subset of backends:
6363| ` AutoChainRules ` | ✅ | ❌ |
6464| ` AutoDiffractor ` | ❌ | ❌ |
6565| ` AutoEnzyme ` (forward) | ✅ | ✅ |
66- | ` AutoEnzyme ` (reverse) | ✅ | ❌ (soon) |
66+ | ` AutoEnzyme ` (reverse) | ✅ | ✅ |
6767| ` AutoFastDifferentiation ` | ✅ | ✅ |
6868| ` AutoFiniteDiff ` | ✅ | ✅ |
6969| ` AutoFiniteDifferences ` | ✅ | ✅ |
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
@@ -14,15 +16,6 @@ check_no_implicit_imports(DifferentiationInterface)
1416
1517LOGGING = get (ENV , " CI" , " false" ) == " false"
1618
17- function remove_matrix_inputs (scens:: Vector{<:Scenario} ) # TODO : remove
18- if VERSION < v " 1.11"
19- return scens
20- else
21- # for https://github.com/EnzymeAD/Enzyme.jl/issues/2071
22- return filter (s -> s. x isa Union{Number,AbstractVector}, scens)
23- end
24- end
25-
2619backends = [
2720 AutoEnzyme (; mode= nothing ),
2821 AutoEnzyme (; mode= Enzyme. Forward),
5548 )
5649
5750 test_differentiation (
58- backends[2 ],
51+ backends[2 : 3 ],
5952 default_scenarios (;
6053 include_normal= false ,
6154 include_cachified= true ,
8275 AutoEnzyme (; mode= Enzyme. Reverse), AutoEnzyme (; mode= Enzyme. Forward)
8376 ),
8477 ],
85- remove_matrix_inputs ( default_scenarios (; include_constantified= true ) );
78+ default_scenarios (; include_constantified= true , include_cachified = true );
8679 excluded= FIRST_ORDER,
8780 logging= LOGGING,
8881 )
89-
90- test_differentiation (
91- AutoEnzyme (; mode= Enzyme. Forward);
92- excluded= vcat (FIRST_ORDER, [:hessian , :hvp ]),
93- logging= LOGGING,
94- )
9582end
9683
9784@testset " Sparse" begin
10087 if VERSION < v " 1.11"
10188 sparse_scenarios ()
10289 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
90+ filter (s -> s. x isa AbstractVector, sparse_scenarios ())
10991 end ;
11092 sparsity= true ,
11193 logging= LOGGING,
You can’t perform that action at this time.
0 commit comments