Skip to content

Commit 065eebe

Browse files
committed
Reactant: add extension to prevent stackoverflow
1 parent 34250b2 commit 065eebe

File tree

2 files changed

+10
-0
lines changed

2 files changed

+10
-0
lines changed

Project.toml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,10 +13,12 @@ Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
1313
[weakdeps]
1414
Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
1515
EnzymeCore = "f151be2c-9106-41f4-ab19-57ee4f262869"
16+
Reactant = "3c362404-f566-11ee-1572-e11a4b42c853"
1617

1718
[extensions]
1819
OptimisersAdaptExt = ["Adapt"]
1920
OptimisersEnzymeCoreExt = "EnzymeCore"
21+
OptimisersReactantExt = "EnzymeCore"
2022

2123
[compat]
2224
Adapt = "4"

ext/OptimisersReactantExt.jl

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
module OptimisersReactantExt
2+
3+
import Optimisers
4+
import Reactant
5+
6+
Optimisers._eps(T::Type{Reactant.TracedRNumber{<:AbstractFloat}}, e) = T(e)
7+
8+
end

0 commit comments

Comments
 (0)