chore: rename extensions in LuxLib (#1619)

avik-pal · web-flow · commit 5c39d5966f78 · 2026-01-06T09:27:22.000-05:00
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "Lux"
 uuid = "b2108857-7c20-44ae-9111-449ecde12c47"
 authors = ["Avik Pal <avikpal@mit.edu> and contributors"]
-version = "1.29.0"
+version = "1.29.1"
 
 [deps]
 ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b"
@@ -74,6 +74,7 @@ LuxMooncakeExt = "Mooncake"
 LuxReactantExt = [
     "Enzyme",
     "Reactant",
+    "Functors",
     "ReactantCore",
     "LuxLib",
     "MLDataDevices",
@@ -108,8 +109,8 @@ GPUArrays = "11"
 GPUArraysCore = "0.2"
 LinearAlgebra = "1.10"
 LossFunctions = "0.11.1, 1"
-LuxCore = "1.4.2"
-LuxLib = "1.14"
+LuxCore = "1.5"
+LuxLib = "1.15"
 MLDataDevices = "1.15"
 MLUtils = "0.4.4"
 MPI = "0.20.19"
diff --git a/docs/Project.toml b/docs/Project.toml
@@ -59,7 +59,7 @@ LinearAlgebra = "1.10"
 Literate = "2.18.0"
 Lux = "1"
 LuxCore = "1.5"
-LuxLib = "1.3.4"
+LuxLib = "1.15"
 LuxTestUtils = "2"
 MLDataDevices = "1.6.10"
 MLUtils = "0.4.8"
diff --git a/lib/LuxLib/Project.toml b/lib/LuxLib/Project.toml
@@ -1,6 +1,6 @@
 name = "LuxLib"
 uuid = "82251201-b29d-42c6-8e01-566dec8acb11"
-version = "1.14.0"
+version = "1.15.0"
 authors = ["Avik Pal <avikpal@mit.edu> and contributors"]
 
 [deps]
@@ -50,21 +50,21 @@ path = "../LuxCore"
 path = "../MLDataDevices"
 
 [extensions]
-LuxLibAppleAccelerateExt = "AppleAccelerate"
-LuxLibBLISBLASExt = "BLISBLAS"
-LuxLibCUDAExt = "CUDA"
+AppleAccelerateExt = "AppleAccelerate"
+BLISBLASExt = "BLISBLAS"
+CUDAExt = "CUDA"
 CUDAForwardDiffExt = ["CUDA", "ForwardDiff"]
-LuxLibEnzymeExt = "Enzyme"
+EnzymeExt = "Enzyme"
 ForwardDiffExt = "ForwardDiff"
-LuxLibLoopVectorizationExt = ["LoopVectorization", "Polyester"]
-LuxLibMKLExt = "MKL"
-LuxLibOctavianExt = ["Octavian", "LoopVectorization"]
-LuxLibReactantExt = ["Reactant", "ReactantCore"]
-LuxLibReverseDiffExt = "ReverseDiff"
-LuxLibSLEEFPiratesExt = "SLEEFPirates"
-LuxLibTrackerAMDGPUExt = ["AMDGPU", "Tracker"]
-LuxLibTrackerExt = "Tracker"
-LuxLibcuDNNExt = ["CUDA", "cuDNN"]
+LoopVectorizationExt = ["LoopVectorization", "Polyester"]
+MKLExt = "MKL"
+OctavianExt = ["Octavian", "LoopVectorization"]
+ReactantExt = ["Reactant", "ReactantCore"]
+ReverseDiffExt = "ReverseDiff"
+SLEEFPiratesExt = "SLEEFPirates"
+TrackerAMDGPUExt = ["AMDGPU", "Tracker"]
+TrackerExt = "Tracker"
+cuDNNExt = ["CUDA", "cuDNN"]
 
 [compat]
 AMDGPU = "0.9.6, 1, 2"
diff --git a/lib/LuxLib/ext/AppleAccelerateExt.jl b/lib/LuxLib/ext/AppleAccelerateExt.jl
@@ -1,4 +1,4 @@
-module LuxLibAppleAccelerateExt
+module AppleAccelerateExt
 
 using LuxLib: Utils
 using Static: True
diff --git a/lib/LuxLib/ext/BLISBLASExt.jl b/lib/LuxLib/ext/BLISBLASExt.jl
@@ -1,4 +1,4 @@
-module LuxLibBLISBLASExt
+module BLISBLASExt
 
 using LuxLib: Utils
 using Static: True
diff --git a/lib/LuxLib/ext/CUDAExt/CUDAExt.jl b/lib/LuxLib/ext/CUDAExt/CUDAExt.jl
@@ -1,4 +1,4 @@
-module LuxLibCUDAExt
+module CUDAExt
 
 using CUDA: CUDA, CUBLAS, CuMatrix, CuVector, CuPtr
 using LinearAlgebra: LinearAlgebra, Transpose, Adjoint
diff --git a/lib/LuxLib/ext/CUDAExt/cublaslt.jl b/lib/LuxLib/ext/CUDAExt/cublaslt.jl
diff --git a/lib/LuxLib/ext/EnzymeExt.jl b/lib/LuxLib/ext/EnzymeExt.jl
@@ -1,4 +1,4 @@
-module LuxLibEnzymeExt
+module EnzymeExt
 
 using LuxLib: Utils
 using Static: True
diff --git a/lib/LuxLib/ext/LoopVectorizationExt.jl b/lib/LuxLib/ext/LoopVectorizationExt.jl
@@ -1,4 +1,4 @@
-module LuxLibLoopVectorizationExt
+module LoopVectorizationExt
 
 using LoopVectorization: LoopVectorization, @tturbo, @turbo, indices
 using Polyester: @batch
diff --git a/lib/LuxLib/ext/MKLExt.jl b/lib/LuxLib/ext/MKLExt.jl
@@ -1,4 +1,4 @@
-module LuxLibMKLExt
+module MKLExt
 
 using LuxLib: Utils
 using Static: True
diff --git a/lib/LuxLib/ext/OctavianExt.jl b/lib/LuxLib/ext/OctavianExt.jl
@@ -1,4 +1,4 @@
-module LuxLibOctavianExt
+module OctavianExt
 
 using Octavian: Octavian
 using Static: True
diff --git a/lib/LuxLib/ext/ReactantExt/ReactantExt.jl b/lib/LuxLib/ext/ReactantExt/ReactantExt.jl
@@ -1,4 +1,4 @@
-module LuxLibReactantExt
+module ReactantExt
 
 using Reactant: Reactant, AnyTracedRArray, AnyTracedRVector, TracedRArray, @opcall
 using ReactantCore: materialize_traced_array
diff --git a/lib/LuxLib/ext/ReactantExt/attention.jl b/lib/LuxLib/ext/ReactantExt/attention.jl
diff --git a/lib/LuxLib/ext/ReactantExt/batched_mul.jl b/lib/LuxLib/ext/ReactantExt/batched_mul.jl
diff --git a/lib/LuxLib/ext/ReactantExt/batchnorm.jl b/lib/LuxLib/ext/ReactantExt/batchnorm.jl
diff --git a/lib/LuxLib/ext/ReverseDiffExt.jl b/lib/LuxLib/ext/ReverseDiffExt.jl
@@ -1,4 +1,4 @@
-module LuxLibReverseDiffExt
+module ReverseDiffExt
 
 using ChainRulesCore: ChainRulesCore
 using LuxLib: LuxLib, Utils, Traits
diff --git a/lib/LuxLib/ext/SLEEFPiratesExt.jl b/lib/LuxLib/ext/SLEEFPiratesExt.jl
@@ -1,4 +1,4 @@
-module LuxLibSLEEFPiratesExt
+module SLEEFPiratesExt
 
 using ChainRulesCore: ChainRulesCore
 using NNlib: NNlib
diff --git a/lib/LuxLib/ext/TrackerAMDGPUExt.jl b/lib/LuxLib/ext/TrackerAMDGPUExt.jl
@@ -1,4 +1,4 @@
-module LuxLibTrackerAMDGPUExt
+module TrackerAMDGPUExt
 
 using AMDGPU: AMDGPU
 using NNlib: NNlib, PoolDims
diff --git a/lib/LuxLib/ext/TrackerExt.jl b/lib/LuxLib/ext/TrackerExt.jl
@@ -1,4 +1,4 @@
-module LuxLibTrackerExt
+module TrackerExt
 
 using FastClosures: @closure
 using LuxLib: LuxLib, Utils, Impl, Traits, GenericBroadcastOp
diff --git a/lib/LuxLib/ext/cuDNNExt/batchnorm.jl b/lib/LuxLib/ext/cuDNNExt/batchnorm.jl
diff --git a/lib/LuxLib/ext/cuDNNExt/cuDNNExt.jl b/lib/LuxLib/ext/cuDNNExt/cuDNNExt.jl
@@ -1,4 +1,4 @@
-module LuxLibcuDNNExt
+module cuDNNExt
 
 using LuxLib: LuxLib, Optional, ∂∅, Impl
 using LuxLib.Utils: Utils, safe_reshape, safe_vec, unsafe_known, recursive_unthunk
diff --git a/test/Project.toml b/test/Project.toml
@@ -65,7 +65,7 @@ LinearAlgebra = "1.10"
 Logging = "1.10"
 LoopVectorization = "0.12.171"
 LuxCore = "1.5"
-LuxLib = "1.3.4"
+LuxLib = "1.15"
 LuxTestUtils = "2"
 MLDataDevices = "1.6.10"
 Mooncake = "0.4.138"

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-module LuxLibAppleAccelerateExt`
	`1`	`+module AppleAccelerateExt`
`2`	`2`
`3`	`3`	`using LuxLib: Utils`
`4`	`4`	`using Static: True`
Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-module LuxLibBLISBLASExt`
	`1`	`+module BLISBLASExt`
`2`	`2`
`3`	`3`	`using LuxLib: Utils`
`4`	`4`	`using Static: True`
Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-module LuxLibCUDAExt`
	`1`	`+module CUDAExt`
`2`	`2`
`3`	`3`	`using CUDA: CUDA, CUBLAS, CuMatrix, CuVector, CuPtr`
`4`	`4`	`using LinearAlgebra: LinearAlgebra, Transpose, Adjoint`
Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-module LuxLibEnzymeExt`
	`1`	`+module EnzymeExt`
`2`	`2`
`3`	`3`	`using LuxLib: Utils`
`4`	`4`	`using Static: True`
Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-module LuxLibLoopVectorizationExt`
	`1`	`+module LoopVectorizationExt`
`2`	`2`
`3`	`3`	`using LoopVectorization: LoopVectorization, @tturbo, @turbo, indices`
`4`	`4`	`using Polyester: @batch`
Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-module LuxLibMKLExt`
	`1`	`+module MKLExt`
`2`	`2`
`3`	`3`	`using LuxLib: Utils`
`4`	`4`	`using Static: True`