also import Mode

EnzymeAD · Oct 8, 2022 · b8d735e · b8d735e
1 parent cd7ca4c
commit b8d735e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/Enzyme.jl b/src/Enzyme.jl
@@ -13,7 +13,7 @@ export autodiff, jacobian, gradient, gradient!
 export markType, batch_size, onehot, chunkedonehot
 
 using LinearAlgebra
-import EnzymeCore: ReverseMode, ForwardMode, Annotation 
+import EnzymeCore: ReverseMode, ForwardMode, Annotation, Mode
 
 # Independent code, must be loaded before "compiler.jl"
 include("pmap.jl")