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,
@@ -205,8 +205,8 @@ if INCLUDE_MOONCAKE
205205 push! (ADTYPES, AutoMooncake (; config= nothing ))
206206end
207207if INCLUDE_ENZYME
208- push! (ADTYPES, AutoEnzyme (; mode = Enzyme. set_runtime_activity (Enzyme. Forward)))
209- push! (ADTYPES, AutoEnzyme (; mode = Enzyme. set_runtime_activity (Enzyme. Reverse)))
208+ push! (ADTYPES, AutoEnzyme (; mode= Enzyme. set_runtime_activity (Enzyme. Forward)))
209+ push! (ADTYPES, AutoEnzyme (; mode= Enzyme. set_runtime_activity (Enzyme. Reverse)))
210210end
211211
212212# Check that ADTypeCheckContext itself works as expected.
You can’t perform that action at this time.
0 commit comments