Skip to content

Commit b13a31d

Browse files
committed
Fix
1 parent 0b2e62e commit b13a31d

2 files changed

Lines changed: 8 additions & 11 deletions

File tree

  • DifferentiationInterface
    • ext/DifferentiationInterfacePolyesterForwardDiffExt
    • test/Back/PolyesterForwardDiff

DifferentiationInterface/ext/DifferentiationInterfacePolyesterForwardDiffExt/onearg.jl

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -442,7 +442,7 @@ function DI.value_derivative_and_second_derivative(
442442
) where {C}
443443
DI.check_prep(f, prep, backend, x, contexts...)
444444
return DI.value_derivative_and_second_derivative(
445-
f, prep, single_threaded(backend), x, contexts...
445+
f, prep.single_threaded_prep, single_threaded(backend), x, contexts...
446446
)
447447
end
448448

@@ -457,7 +457,7 @@ function DI.value_derivative_and_second_derivative!(
457457
) where {C}
458458
DI.check_prep(f, prep, backend, x, contexts...)
459459
return DI.value_derivative_and_second_derivative!(
460-
f, der, der2, prep, single_threaded(backend), x, contexts...
460+
f, der, der2, prep.single_threaded_prep, single_threaded(backend), x, contexts...
461461
)
462462
end
463463

@@ -469,7 +469,9 @@ function DI.second_derivative(
469469
contexts::Vararg{DI.Context,C},
470470
) where {C}
471471
DI.check_prep(f, prep, backend, x, contexts...)
472-
return DI.second_derivative(f, prep, single_threaded(backend), x, contexts...)
472+
return DI.second_derivative(
473+
f, prep.single_threaded_prep, single_threaded(backend), x, contexts...
474+
)
473475
end
474476

475477
function DI.second_derivative!(
@@ -481,5 +483,7 @@ function DI.second_derivative!(
481483
contexts::Vararg{DI.Context,C},
482484
) where {C}
483485
DI.check_prep(f, prep, backend, x, contexts...)
484-
return DI.second_derivative!(f, der2, prep, single_threaded(backend), x, contexts...)
486+
return DI.second_derivative!(
487+
f, der2, prep.single_threaded_prep, single_threaded(backend), x, contexts...
488+
)
485489
end

DifferentiationInterface/test/Back/PolyesterForwardDiff/test.jl

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -30,13 +30,6 @@ test_differentiation(
3030
backends,
3131
default_scenarios(; include_constantified=true, include_cachified=true);
3232
logging=LOGGING,
33-
excluded=SECOND_ORDER,
34-
);
35-
36-
test_differentiation(
37-
SecondOrder(AutoPolyesterForwardDiff(), AutoPolyesterForwardDiff()),
38-
default_scenarios();
39-
logging=LOGGING,
4033
);
4134

4235
@testset "Batch size" begin

0 commit comments

Comments
 (0)