Memoize Coeffs

Author: oxinabox

src/methods.jl

@@ -156,15 +156,20 @@ function _check_p_q(p::Integer, q::Integer)
 end
 
 # Compute coefficients for the method
+
+const _COEFFS_CACHE = Dict{Tuple{AbstractVector{<:Real}, Integer, Integer}, Vector{Float64}}()
 function _coefs(grid::AbstractVector{<:Real}, p::Integer, q::Integer)
-    # For high precision on the \ we use Rational, and to prevent overflows we use Int128
-    # At the end we go to Float64 for fast floating point math (rather than rational math)
-    C = [Rational{Int128}(g^i) for i in 0:(p - 1), g in grid]
-    x = zeros(Rational{Int128}, p)
-    x[q + 1] = factorial(q)
-    return Float64.(C \ x)
+    return get!(_COEFFS_CACHE, (grid, p, q)) do
+        # For high precision on the \ we use Rational, and to prevent overflows we use Int128
+        # At the end we go to Float64 for fast floating point math (rather than rational math)
+        C = [Rational{Int128}(g^i) for i in 0:(p - 1), g in grid]
+        x = zeros(Rational{Int128}, p)
+        x[q + 1] = factorial(q)
+        return Float64.(C \ x)
+    end
 end
 
+
 # Estimate the bound on the function value and its derivatives at a point
 _estimate_bound(x, cond) = cond * maximum(abs, x) + TINY