chore!: remove Reexport of NNlib (will be done via Lux)

LuxDL · Aug 18, 2024 · 6813462 · 6813462
1 parent 7c8d876
commit 6813462
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 6 deletions.
diff --git a/Project.toml b/Project.toml
@@ -75,7 +75,7 @@ NNlib = "0.9.21"
 Octavian = "0.3.28"
 Polyester = "0.7.15"
 Random = "1.10"
-Reexport = "1"
+Reexport = "1.2"
 ReverseDiff = "1.15"
 SLEEFPirates = "0.6.43"
 Static = "0.8.4, 1"

diff --git a/src/LuxLib.jl b/src/LuxLib.jl
@@ -12,8 +12,6 @@ using LuxCore: LuxCore
 using MLDataDevices: get_device_type, AbstractGPUDevice
 using NNlib: NNlib, ConvDims, σ
 
-@reexport using NNlib
-
 const Optional{T} = Union{Nothing, T}
 const Numeric = Union{AbstractArray{<:T}, T} where {T <: Number}
 const ∂∅ = NoTangent()

diff --git a/test/common_ops/dense_tests.jl b/test/common_ops/dense_tests.jl
@@ -102,7 +102,7 @@ end
 end
 
 @testitem "Fused Dense: StaticArrays" tags=[:dense] begin
-    using StaticArrays
+    using StaticArrays, NNlib
 
     x = @SArray rand(2, 4)
     weight = @SArray rand(3, 2)
@@ -112,7 +112,7 @@ end
 end
 
 @testitem "Fused Dense: CPU No Scalar Indexing" tags=[:dense] begin
-    using JLArrays
+    using JLArrays, NNlib
 
     x = JLArray(rand(Float32, 2, 4))
     weight = JLArray(rand(Float32, 3, 2))

diff --git a/test/others/qa_tests.jl b/test/others/qa_tests.jl
@@ -1,5 +1,5 @@
 @testitem "Aqua: Quality Assurance" tags=[:others] begin
-    using Aqua, ChainRulesCore, EnzymeCore
+    using Aqua, ChainRulesCore, EnzymeCore, NNlib
     using EnzymeCore: EnzymeRules
 
     Aqua.test_all(