From 2d3ee8eb25258eec31b4a09b82204ededd29c23a Mon Sep 17 00:00:00 2001 From: Christopher Rackauckas Date: Sun, 17 Sep 2023 01:45:47 -0400 Subject: [PATCH] Update ext/IntegralsZygoteExt.jl --- ext/IntegralsZygoteExt.jl | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/ext/IntegralsZygoteExt.jl b/ext/IntegralsZygoteExt.jl index 58d732ba..986c522d 100644 --- a/ext/IntegralsZygoteExt.jl +++ b/ext/IntegralsZygoteExt.jl @@ -124,8 +124,8 @@ function ChainRulesCore.rrule(::typeof(Integrals.__solvebp), cache, alg, senseal out, quadrature_adjoint end -# Zygote.@adjoint function Zygote.literal_getproperty(sol::SciMLBase.IntegralSolution, -# ::Val{:u}) -# sol.u, Δ -> (SciMLBase.build_solution(sol.prob, sol.alg, Δ, sol.resid),) -# end +Zygote.@adjoint function Zygote.literal_getproperty(sol::SciMLBase.IntegralSolution, + ::Val{:u}) + sol.u, Δ -> (SciMLBase.build_solution(sol.prob, sol.alg, Δ, sol.resid),) +end end