Allow older ChainRulesCore

Author: chriselrod

Project.toml

@@ -1,11 +1,10 @@
 name = "LoopVectorization"
 uuid = "bdcacae8-1622-11e9-2a5c-532679323890"
 authors = ["Chris Elrod <[email protected]>"]
-version = "0.12.30"
+version = "0.12.31"
 
 [deps]
 ArrayInterface = "4fba245c-0d91-5ea0-9b3e-6abc04ee57a9"
-ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
 DocStringExtensions = "ffbed154-4ef7-542d-bbb7-c09d3a79fcae"
 IfElse = "615f187c-cbe4-4ef1-ba3b-2fcf58d6d173"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
@@ -21,7 +20,6 @@ VectorizationBase = "3d5dd08c-fd9d-11e8-17fa-ed2836048c2f"
 
 [compat]
 ArrayInterface = "3.1.9"
-ChainRulesCore = "0.10"
 DocStringExtensions = "0.8"
 IfElse = "0.1"
 OffsetArrays = "1.4.1"

src/simdfunctionals/vmap_grad_rrule.jl

@@ -1,17 +1,23 @@
 
 import .ChainRulesCore
 
+if isdefined(ChainRulesCore, :ZeroTangent)
+  const ChainRulesZero = ChainRulesCore.ZeroTangent
+else
+  const ChainRulesZero = ChainRulesCore.Zero
+end
+
 function ChainRulesCore.rrule(::typeof(tanh_fast), x)
     t = tanh_fast(x)
     ∂ = let t = t
-        y -> (ChainRulesCore.ZeroTangent(), mul_fast(vfnmadd_fast(t, t, one(t)), y))
+        y -> (ChainRulesZero(), mul_fast(vfnmadd_fast(t, t, one(t)), y))
     end
     t, ∂
 end
 function ChainRulesCore.rrule(::typeof(sigmoid_fast), x)
     s = sigmoid_fast(x)
     ∂ = let s = s
-        y -> (ChainRulesCore.ZeroTangent(), mul_fast(vfnmadd_fast(s, s, s), y))
+        y -> (ChainRulesZero(), mul_fast(vfnmadd_fast(s, s, s), y))
     end
     s, ∂
 end
@@ -20,7 +26,7 @@ function ChainRulesCore.rrule(::typeof(relu), v)
     cmp = v < z
     r = ifelse(cmp, z, v)
     ∂ = let cmp = cmp
-        y -> (ChainRulesCore.ZeroTangent(), ifelse(cmp, zero(y), y))
+        y -> (ChainRulesZero(), ifelse(cmp, zero(y), y))
     end
     r, ∂
 end
@@ -64,7 +70,7 @@ end
 @generated function (b::SIMDMapBack{K,T})(Δ::A) where {K,T,A}
     preloop = Expr(:block, :(jacs = b.jacs))
     loop_body = Expr(:block, :(Δᵢ = Δ[i]))
-    ret = Expr(:tuple, ChainRulesCore.ZeroTangent(), ChainRulesCore.ZeroTangent())
+    ret = Expr(:tuple, ChainRulesZero(), ChainRulesZero())
     for k ∈1:K
         jₖ = Symbol(:j_, k)
         push!(preloop.args, :($jₖ = jacs[$k]))

test/runtests.jl

@@ -14,8 +14,7 @@ const START_TIME = time()
 @time @testset "LoopVectorization.jl" begin
 
   @time if LOOPVECTORIZATION_TEST == "all" || LOOPVECTORIZATION_TEST == "part1"
-    @time Aqua.test_all(LoopVectorization, stale_deps = false)
-    @time Aqua.test_stale_deps(LoopVectorization, ignore = [:ChainRulesCore])
+    @time Aqua.test_all(LoopVectorization)
     # @test isempty(detect_unbound_args(LoopVectorization))
 
     @time include("printmethods.jl")