Merge pull request #250 from Vaibhavdixit02/master

Load Zygote in extension by checking for julia <1.9 support

Author: ChrisRackauckas

.github/workflows/CI.yml

@@ -18,7 +18,7 @@ jobs:
         version:
           - '1' # Latest Release
           - '~1.6' # Current LTS
-          - '^1.9.0-0'
+          - '1.8'
     steps:
       - uses: actions/checkout@v3
       - uses: julia-actions/setup-julia@v1

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"

ext/SparseDiffToolsZygote.jl renamed to 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

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