Merge pull request #346 from SciML/myb/frule

Intercept frule at the top level
SciML · Oct 16, 2024 · f5e6ce7 · f5e6ce7
2 parents a084b2c + 4f68b0a
commit f5e6ce7
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/ext/DataInterpolationsChainRulesCoreExt.jl b/ext/DataInterpolationsChainRulesCoreExt.jl
@@ -121,4 +121,8 @@ function ChainRulesCore.frule((_, _, Δt), ::typeof(_interpolate), A::AbstractIn
     return _interpolate(A, t), derivative(A, t) * Δt
 end
 
+function ChainRulesCore.frule((_, Δt), A::AbstractInterpolation, t::Number)
+    return A(t), derivative(A, t) * Δt
+end
+
 end # module