diff --git a/Project.toml b/Project.toml index 5f7b61b17..ffe617ba3 100644 --- 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.7" MLDataDevices = "1.5" MLUtils = "0.4.4" diff --git a/docs/Project.toml b/docs/Project.toml index a48d7c818..d5a5899e4 100644 --- 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.5" MLDataDevices = "1.4" diff --git a/ext/LuxReactantExt/LuxReactantExt.jl b/ext/LuxReactantExt/LuxReactantExt.jl index 23159f861..ce0e0cd06 100644 --- a/ext/LuxReactantExt/LuxReactantExt.jl +++ b/ext/LuxReactantExt/LuxReactantExt.jl @@ -7,10 +7,8 @@ using Setfield: @set! using Static: False 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 deleted file mode 100644 index 4077345b2..000000000 --- a/ext/LuxReactantExt/tracing.jl +++ /dev/null @@ -1,6 +0,0 @@ -# Avoid tracing though models since it won't contain anything useful -function Reactant.make_tracer( - seen, @nospecialize(model::AbstractLuxLayer), @nospecialize(path), mode; kwargs... -) - return model -end diff --git a/lib/LuxLib/Project.toml b/lib/LuxLib/Project.toml index ac6832f19..48aeecf56 100644 --- 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.5" Markdown = "1.10" diff --git a/test/Project.toml b/test/Project.toml index 90ee23de9..6c4e3ad51 100644 --- 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.5" MLDataDevices = "1.3"