Deprecate DiffEqBase.fastpow for FastPower.fastpower

This requires the registration to finish but will finally make it more accessible to Symbolics and other tooling.
SciML · Oct 5, 2024 · 35aa77a · 35aa77a
1 parent 6bce332
commit 35aa77a
Show file tree

Hide file tree

Showing 10 changed files with 4 additions and 126 deletions.
diff --git a/ext/DiffEqBaseEnzymeExt.jl b/ext/DiffEqBaseEnzymeExt.jl
@@ -1,7 +1,7 @@
 module DiffEqBaseEnzymeExt
 
 using DiffEqBase
-import DiffEqBase: value, fastpow
+import DiffEqBase: value
 using Enzyme
 import Enzyme: Const
 using ChainRulesCore
@@ -53,6 +53,4 @@ function Enzyme.EnzymeRules.reverse(config::Enzyme.EnzymeRules.RevConfigWidth{1}
     return ntuple(_ -> nothing, Val(length(args) + 4))
 end
 
-Enzyme.Compiler.known_ops[typeof(DiffEqBase.fastpow)] = (:pow, 2, nothing)
-
 end
diff --git a/ext/DiffEqBaseMeasurementsExt.jl b/ext/DiffEqBaseMeasurementsExt.jl
@@ -19,8 +19,6 @@ DiffEqBase.promote_u0(u0, p::AbstractArray{<:Measurements.Measurement}, t0) = el
 value(x::Type{Measurements.Measurement{T}}) where {T} = T
 value(x::Measurements.Measurement) = Measurements.value(x)
 
-@inline DiffEqBase.fastpow(x::Measurements.Measurement, y::Measurements.Measurement) = x^y
-
 # Support adaptive steps should be errorless
 @inline function DiffEqBase.ODE_DEFAULT_NORM(
         u::AbstractArray{

diff --git a/ext/DiffEqBaseMonteCarloMeasurementsExt.jl b/ext/DiffEqBaseMonteCarloMeasurementsExt.jl
@@ -27,11 +27,6 @@ end
 DiffEqBase.value(x::Type{MonteCarloMeasurements.AbstractParticles{T, N}}) where {T, N} = T
 DiffEqBase.value(x::MonteCarloMeasurements.AbstractParticles) = mean(x.particles)
 
-@inline function DiffEqBase.fastpow(x::MonteCarloMeasurements.AbstractParticles,
-        y::MonteCarloMeasurements.AbstractParticles)
-    x^y
-end
-
 # Support adaptive steps should be errorless
 @inline function DiffEqBase.ODE_DEFAULT_NORM(
         u::AbstractArray{

diff --git a/ext/DiffEqBaseTrackerExt.jl b/ext/DiffEqBaseTrackerExt.jl
@@ -31,7 +31,6 @@ end
 DiffEqBase.promote_u0(u0, p::Tracker.TrackedArray, t0) = Tracker.track(u0)
 DiffEqBase.promote_u0(u0, p::AbstractArray{<:Tracker.TrackedReal}, t0) = eltype(p).(u0)
 
-@inline DiffEqBase.fastpow(x::Tracker.TrackedReal, y::Tracker.TrackedReal) = x^y
 @inline Base.any(f::Function, x::Tracker.TrackedArray) = any(f, Tracker.data(x))
 
 # Support adaptive with non-tracked time

diff --git a/src/DiffEqBase.jl b/src/DiffEqBase.jl
@@ -6,6 +6,9 @@ end
 
 import PrecompileTools
 
+import FastPower
+@deprecate fastpow(x,y) Fastpower.fastpower(x,y)
+
 using ArrayInterface
 
 using StaticArraysCore # data arrays
@@ -128,7 +131,6 @@ abstract type DECostFunction end
 import SciMLBase: Void, unwrapped_f
 
 include("utils.jl")
-include("fastpow.jl")
 include("stats.jl")
 include("calculate_residuals.jl")
 include("tableaus.jl")

diff --git a/src/fastpow.jl b/src/fastpow.jl
diff --git a/src/forwarddiff.jl b/src/forwarddiff.jl
@@ -430,8 +430,6 @@ end
 value(x::Type{ForwardDiff.Dual{T, V, N}}) where {T, V, N} = V
 value(x::ForwardDiff.Dual) = value(ForwardDiff.value(x))
 
-@inline fastpow(x::ForwardDiff.Dual, y::ForwardDiff.Dual) = x^y
-
 sse(x::Number) = abs2(x)
 sse(x::ForwardDiff.Dual) = sse(ForwardDiff.value(x)) + sum(sse, ForwardDiff.partials(x))
 totallength(x::Number) = 1

diff --git a/test/downstream/enzyme.jl b/test/downstream/enzyme.jl
diff --git a/test/fastpow.jl b/test/fastpow.jl
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -19,7 +19,6 @@ end
 
 @time begin
     if GROUP == "All" || GROUP == "Core"
-        @time @safetestset "Fast Power" include("fastpow.jl")
         @time @safetestset "Callbacks" include("callbacks.jl")
         @time @safetestset "Internal Rootfinders" include("internal_rootfinder.jl")
         @time @safetestset "Plot Vars" include("plot_vars.jl")