refactor: move ChainRulesCore to an extension

Author: AayushSabharwal

Project.toml

@@ -5,13 +5,24 @@ repo = "https://github.com/JuliaAlgebra/MultivariatePolynomials.jl"
 version = "0.5.12"
 
 [deps]
-ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
 DataStructures = "864edb3b-99cc-5e75-8d2d-829cb0a9cfe8"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 MutableArithmetics = "d8a4904e-b15c-11e9-3269-09a3773c0cb0"
 
+[weakdeps]
+ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
+
+[extensions]
+MultivariatePolynomialsChainRulesCoreExt = "ChainRulesCore"
+
 [compat]
 ChainRulesCore = "1"
 DataStructures = "0.19"
 MutableArithmetics = "0.3, 1"
 julia = "1.10"
+
+[extras]
+ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
+
+[targets]
+test = ["ChainRulesCore"]

ext/MultivariatePolynomialsChainRulesCoreExt.jl

@@ -0,0 +1,38 @@
+module MultivariatePolynomialsChainRulesCoreExt
+
+import ChainRulesCore
+using MultivariatePolynomials
+using MultivariatePolynomials: _APL
+
+ChainRulesCore.@scalar_rule +(x::_APL) true
+ChainRulesCore.@scalar_rule -(x::_APL) -1
+
+ChainRulesCore.@scalar_rule +(x::_APL, y::_APL) (true, true)
+ChainRulesCore.@scalar_rule -(x::_APL, y::_APL) (true, -1)
+
+function ChainRulesCore.frule((_, Δp, Δq), ::typeof(*), p::_APL, q::_APL)
+    return p * q, MA.add_mul!!(p * Δq, q, Δp)
+end
+function ChainRulesCore.rrule(::typeof(*), p::_APL, q::_APL)
+    function times_pullback2(ΔΩ̇)
+        #ΔΩ = ChainRulesCore.unthunk(Ω̇)
+        #return (ChainRulesCore.NoTangent(), ChainRulesCore.ProjectTo(p)(ΔΩ * q'), ChainRulesCore.ProjectTo(q)(p' * ΔΩ))
+        return (ChainRulesCore.NoTangent(), ΔΩ̇ * q', p' * ΔΩ̇)
+    end
+    return p * q, times_pullback2
+end
+
+function ChainRulesCore.frule((_, Δp, _), ::typeof(differentiate), p, x)
+    return differentiate(p, x), differentiate(Δp, x)
+end
+function pullback(Δdpdx, x)
+    return ChainRulesCore.NoTangent(),
+    x * differentiate(x * Δdpdx, x),
+    ChainRulesCore.NoTangent()
+end
+function ChainRulesCore.rrule(::typeof(differentiate), p, x)
+    dpdx = differentiate(p, x)
+    return dpdx, Base.Fix2(pullback, x)
+end
+
+end

src/chain_rules.jl

@@ -1,32 +0,0 @@
-import ChainRulesCore
-
-ChainRulesCore.@scalar_rule +(x::_APL) true
-ChainRulesCore.@scalar_rule -(x::_APL) -1
-
-ChainRulesCore.@scalar_rule +(x::_APL, y::_APL) (true, true)
-ChainRulesCore.@scalar_rule -(x::_APL, y::_APL) (true, -1)
-
-function ChainRulesCore.frule((_, Δp, Δq), ::typeof(*), p::_APL, q::_APL)
-    return p * q, MA.add_mul!!(p * Δq, q, Δp)
-end
-function ChainRulesCore.rrule(::typeof(*), p::_APL, q::_APL)
-    function times_pullback2(ΔΩ̇)
-        #ΔΩ = ChainRulesCore.unthunk(Ω̇)
-        #return (ChainRulesCore.NoTangent(), ChainRulesCore.ProjectTo(p)(ΔΩ * q'), ChainRulesCore.ProjectTo(q)(p' * ΔΩ))
-        return (ChainRulesCore.NoTangent(), ΔΩ̇ * q', p' * ΔΩ̇)
-    end
-    return p * q, times_pullback2
-end
-
-function ChainRulesCore.frule((_, Δp, _), ::typeof(differentiate), p, x)
-    return differentiate(p, x), differentiate(Δp, x)
-end
-function pullback(Δdpdx, x)
-    return ChainRulesCore.NoTangent(),
-    x * differentiate(x * Δdpdx, x),
-    ChainRulesCore.NoTangent()
-end
-function ChainRulesCore.rrule(::typeof(differentiate), p, x)
-    dpdx = differentiate(p, x)
-    return dpdx, Base.Fix2(pullback, x)
-end