Load Zygote in extension by checking for julia <1.9 support

Vaibhavdixit02 · Jun 15, 2023 · 0b6434f · 0b6434f
1 parent 3bd40df
commit 0b6434f
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/Project.toml b/Project.toml
@@ -27,7 +27,7 @@ VertexSafeGraphs = "19fa3120-7c27-5ec5-8db8-b0b0aa330d6f"
 Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 
 [extensions]
-SparseDiffToolsZygote = "Zygote"
+SparseDiffToolsZygoteExt = "Zygote"
 
 [compat]
 ADTypes = "0.1"

diff --git a/ext/SparseDiffToolsZygote.jl → ext/SparseDiffToolsZygoteExt.jl b/ext/SparseDiffToolsZygote.jl → ext/SparseDiffToolsZygoteExt.jl
@@ -1,6 +1,6 @@
-module SparseDiffToolsZygote
+module SparseDiffToolsZygoteExt
 
-import Zygote
+isdefined(Base, :get_extension) ? (import Zygote) : (using ..Zygote)
 using ADTypes
 using LinearAlgebra
 using SparseDiffTools: SparseDiffTools, DeivVecTag, AutoDiffVJP

diff --git a/src/SparseDiffTools.jl b/src/SparseDiffTools.jl
@@ -84,8 +84,8 @@ function auto_vecjac! end
 @static if !isdefined(Base, :get_extension)
     function __init__()
         Requires.@require Zygote="e88e6eb3-aa80-5325-afca-941959d7151f" begin
-            include("../ext/SparseDiffToolsZygote.jl")
-            @reexport using .SparseDiffToolsZygote
+            include("../ext/SparseDiffToolsZygoteExt.jl")
+            @reexport using .SparseDiffToolsZygoteExt
         end
     end
 end