fix: update default rng for reactant

LuxDL · Jan 1, 2025 · ce9f77f · ce9f77f
1 parent 1951c86
commit ce9f77f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/lib/MLDataDevices/Project.toml b/lib/MLDataDevices/Project.toml
@@ -1,7 +1,7 @@
 name = "MLDataDevices"
 uuid = "7e8f7934-dd98-4c1a-8fe8-92b47a384d40"
 authors = ["Avik Pal <[email protected]> and contributors"]
-version = "1.6.5"
+version = "1.6.6"
 
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
@@ -66,7 +66,7 @@ Metal = "1"
 OneHotArrays = "0.2.5"
 Preferences = "1.4"
 Random = "1.10"
-Reactant = "0.2.6"
+Reactant = "0.2.12"
 RecursiveArrayTools = "3.8"
 ReverseDiff = "1.15"
 SparseArrays = "1.10"

diff --git a/lib/MLDataDevices/ext/MLDataDevicesReactantExt.jl b/lib/MLDataDevices/ext/MLDataDevicesReactantExt.jl
@@ -8,9 +8,9 @@ using Reactant: Reactant, XLA, ConcreteRArray, ConcreteRNumber, TracedRArray,
 MLDataDevices.loaded(::Union{ReactantDevice, Type{<:ReactantDevice}}) = true
 MLDataDevices.functional(::Union{ReactantDevice, Type{<:ReactantDevice}}) = true
 
-# Default RNG: Forward to CPU, we will compile it
+# Default RNG
 function MLDataDevices.default_device_rng(::ReactantDevice)
-    return MLDataDevices.default_device_rng(CPUDevice())
+    return Reactant.TracedRandom.default_rng()
 end
 
 # Query Device from Array