exports working in 1x8

Author: vpuri3

Project.toml

@@ -12,6 +12,7 @@ FiniteDiff = "6a86dc24-6348-571c-b903-95158fe2bd41"
 ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210"
 Graphs = "86223c79-3864-5bf0-83f7-82e725a168b6"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
+Reexport = "189a3867-3050-52da-a836-e630ba90ab69"
 Requires = "ae029012-a4dd-5104-9daa-d747884805df"
 SciMLOperators = "c0aeaf25-5076-4817-a8d5-81caf7dfa961"
 SparseArrays = "2f01184e-e22b-5df5-ae63-d93ebab69eaf"
@@ -38,6 +39,7 @@ SciMLOperators = "0.1.19, 0.2"
 StaticArrays = "1"
 Tricks = "0.1.6"
 VertexSafeGraphs = "0.2"
+Zygote = "0.6"
 julia = "1.6"
 
 [extras]

ext/SparseDiffToolsZygote.jl

@@ -1,15 +1,18 @@
 module SparseDiffToolsZygote
 
-import Zygote
-
-using SparseDiffTools
-using SparseDiffTools: DeviVecTag, FwdModeAutoDiffVecProd
-
-using SparseDiffTools.ForwardDiff
-using ForwardDiff: Dual, Tag
-
-using SparseDiffTools.SciMLOperators: FunctionOperator
-using SparseDiffTools.Tricks: static_hasmethod
+if isdefined(Base, :get_extension)
+    import Zygote
+    using SparseDiffTools: SparseDiffTools, DeivVecTag, FwdModeAutoDiffVecProd
+    using ForwardDiff: ForwardDiff, Dual
+    using SciMLOperators: FunctionOperator
+    using Tricks: static_hasmethod
+else
+    import ..Zygote
+    using ..SparseDiffTools: SparseDiffTools, DeivVecTag, FwdModeAutoDiffVecProd
+    using ..ForwardDiff: ForwardDiff, Dual
+    using ..SciMLOperators: FunctionOperator
+    using ..Tricks: static_hasmethod
+end
 
 export
        numback_hesvec, numback_hesvec!,

src/SparseDiffTools.jl

@@ -20,7 +20,7 @@ using ArrayInterface: matrix_colors
 
 using SciMLOperators
 import SciMLOperators: update_coefficients, update_coefficients!
-using Tricks: static_hasmethod
+using Tricks: Tricks, static_hasmethod
 
 abstract type AbstractAutoDiffVecProd end
 
@@ -68,14 +68,14 @@ Base.@pure __parameterless_type(T) = Base.typename(T).wrapper
 parameterless_type(x) = parameterless_type(typeof(x))
 parameterless_type(x::Type) = __parameterless_type(x)
 
-if !isdefined(Base, :get_extension)
-    using Requires
-end
+import Requires
+import Reexport
 
-function __init__()
-    @static if !isdefined(Base, :get_extension)
-        @require Zygote="e88e6eb3-aa80-5325-afca-941959d7151f" begin
+@static if !isdefined(Base, :get_extension)
+    function __init__()
+        Requires.@require Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f" begin
             include("../ext/SparseDiffToolsZygote.jl")
+            Reexport.@reexport using .SparseDiffToolsZygote
         end
     end
 end