Skip to content

Commit

Permalink
fix: remove ambiguity in BatchSizeSettings with PolyesterForwardDiff (#…
Browse files Browse the repository at this point in the history
…606)

* fix: remove ambiguity in BatchSizeSettings with PolyesterForwardDiff

* Deactivate Enzyme test
  • Loading branch information
gdalle authored Nov 5, 2024
1 parent 6f7407c commit 3b978b8
Show file tree
Hide file tree
Showing 3 changed files with 11 additions and 3 deletions.
2 changes: 1 addition & 1 deletion DifferentiationInterface/Project.toml
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
name = "DifferentiationInterface"
uuid = "a0c0ee7d-e4b9-4e03-894e-1c5f64a51d63"
authors = ["Guillaume Dalle", "Adrian Hill"]
version = "0.6.18"
version = "0.6.19"

[deps]
ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b"
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -28,8 +28,12 @@ end

DI.check_available(::AutoPolyesterForwardDiff) = true

function DI.BatchSizeSettings(backend::AutoPolyesterForwardDiff, x_or_N)
return DI.BatchSizeSettings(single_threaded(backend), x_or_N)
function DI.BatchSizeSettings(backend::AutoPolyesterForwardDiff, x::AbstractArray)
return DI.BatchSizeSettings(single_threaded(backend), x)
end

function DI.BatchSizeSettings(backend::AutoPolyesterForwardDiff, N::Integer)
return DI.BatchSizeSettings(single_threaded(backend), N)
end

function DI.threshold_batchsize(
Expand Down
4 changes: 4 additions & 0 deletions DifferentiationInterface/test/Back/Enzyme/test.jl
Original file line number Diff line number Diff line change
Expand Up @@ -41,12 +41,16 @@ test_differentiation(
logging=LOGGING,
);

#=
# TODO: reactivate closurified tests once Enzyme#2056 is fixed
test_differentiation(
duplicated_backends,
default_scenarios(; include_normal=false, include_closurified=true);
excluded=SECOND_ORDER,
logging=LOGGING,
);
=#

#=
# TODO: reactivate type stability tests
Expand Down

0 comments on commit 3b978b8

Please sign in to comment.