fix: remove tracing overload

LuxDL · Nov 4, 2024 · d7051ed · d7051ed
1 parent 68a8e92
commit d7051ed
Show file tree

Hide file tree

Showing 6 changed files with 4 additions and 11 deletions.
diff --git a/Project.toml b/Project.toml
@@ -86,7 +86,7 @@ Functors = "0.4.12"
 GPUArraysCore = "0.1.6, 0.2"
 LinearAlgebra = "1.10"
 LossFunctions = "0.11.1"
-LuxCore = "1"
+LuxCore = "1.1"
 LuxLib = "1.3.4"
 MLDataDevices = "1.3"
 MLUtils = "0.4.4"

diff --git a/docs/Project.toml b/docs/Project.toml
@@ -46,7 +46,7 @@ LinearAlgebra = "1.10"
 Literate = "2.18.0"
 Lux = "1"
 LuxCUDA = "0.3.2"
-LuxCore = "1"
+LuxCore = "1.1"
 LuxLib = "1.3.4"
 LuxTestUtils = "1.2"
 MLDataDevices = "1.4"

diff --git a/ext/LuxReactantExt/LuxReactantExt.jl b/ext/LuxReactantExt/LuxReactantExt.jl
@@ -10,7 +10,6 @@ using Lux: Lux, LuxOps, Training
 using LuxCore: AbstractLuxLayer
 using Lux.Training: TrainingBackendCache, ReactantBackend
 
-include("tracing.jl")
 include("training.jl")
 
 end
diff --git a/ext/LuxReactantExt/tracing.jl b/ext/LuxReactantExt/tracing.jl
diff --git a/lib/LuxLib/Project.toml b/lib/LuxLib/Project.toml
@@ -73,7 +73,7 @@ Hwloc = "3.2"
 KernelAbstractions = "0.9.27"
 LinearAlgebra = "1.10"
 LoopVectorization = "0.12.171"
-LuxCore = "1"
+LuxCore = "1.1"
 MKL = "0.7"
 MLDataDevices = "1.2"
 Markdown = "1.10"

diff --git a/test/Project.toml b/test/Project.toml
@@ -57,7 +57,7 @@ InteractiveUtils = "<0.0.1, 1"
 LinearAlgebra = "1.10"
 Logging = "1.10"
 LoopVectorization = "0.12.171"
-LuxCore = "1.0"
+LuxCore = "1.1"
 LuxLib = "1.3.4"
 LuxTestUtils = "1.3"
 MLDataDevices = "1.3"