File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change 3232const always_valid_eltypes = (AbstractFloat, AbstractIrrational, Integer, Rational)
3333
3434""" A dictionary mapping ADTypes to the element types they use."""
35- eltypes_by_adtype = Dict (
35+ eltypes_by_adtype = Dict {Type,Tuple} (
3636 AutoForwardDiff => (ForwardDiff. Dual,),
3737 AutoReverseDiff => (
3838 ReverseDiff. TrackedArray,
@@ -203,8 +203,8 @@ if INCLUDE_MOONCAKE
203203 push! (ADTYPES, AutoMooncake (; config= nothing ))
204204end
205205if INCLUDE_ENZYME
206- push! (ADTYPES, AutoEnzyme (; mode = Enzyme. set_runtime_activity (Enzyme. Forward)))
207- push! (ADTYPES, AutoEnzyme (; mode = Enzyme. set_runtime_activity (Enzyme. Reverse)))
206+ push! (ADTYPES, AutoEnzyme (; mode= Enzyme. set_runtime_activity (Enzyme. Forward)))
207+ push! (ADTYPES, AutoEnzyme (; mode= Enzyme. set_runtime_activity (Enzyme. Reverse)))
208208end
209209
210210# Check that ADTypeCheckContext itself works as expected.
You can’t perform that action at this time.
0 commit comments