Skip to content

Commit 558a9b8

Browse files
authored
Update fallback for mutating second-order operators (#298)
1 parent 1cf497c commit 558a9b8

File tree

3 files changed

+3
-9
lines changed

3 files changed

+3
-9
lines changed

DifferentiationInterface/src/second_order/hessian.jl

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -77,9 +77,7 @@ function hessian!(
7777
x,
7878
extras::HessianExtras=prepare_hessian(f, backend, x),
7979
) where {F}
80-
new_backend = SecondOrder(backend, backend)
81-
new_extras = prepare_hessian(f, new_backend, x)
82-
return hessian!(f, hess, new_backend, x, new_extras)
80+
return hessian!(f, hess, SecondOrder(backend, backend), x, extras)
8381
end
8482

8583
function hessian!(

DifferentiationInterface/src/second_order/hvp.jl

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -182,9 +182,7 @@ end
182182
function hvp!(
183183
f::F, p, backend::AbstractADType, x, v, extras::HVPExtras=prepare_hvp(f, backend, x, v)
184184
) where {F}
185-
new_backend = SecondOrder(backend, backend)
186-
new_extras = prepare_hvp(f, new_backend, x, v)
187-
return hvp!(f, p, new_backend, x, v, new_extras)
185+
return hvp!(f, p, SecondOrder(backend, backend), x, v, extras)
188186
end
189187

190188
function hvp!(

DifferentiationInterface/src/second_order/second_derivative.jl

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -81,9 +81,7 @@ function second_derivative!(
8181
x,
8282
extras::SecondDerivativeExtras=prepare_second_derivative(f, backend, x),
8383
) where {F}
84-
new_backend = SecondOrder(backend, backend)
85-
new_extras = prepare_second_derivative(f, new_backend, x)
86-
return second_derivative!(f, der2, new_backend, x, new_extras)
84+
return second_derivative!(f, der2, SecondOrder(backend, backend), x, extras)
8785
end
8886

8987
function second_derivative!(

0 commit comments

Comments
 (0)