docs: update test doc

Author: avik-pal

Project.toml

@@ -113,7 +113,7 @@ Enzyme = "0.13.120"
 EnzymeCore = "0.8.16"
 FastClosures = "0.3.2"
 Flux = "0.16.3"
-ForwardDiff = "0.10.36, 1"
+ForwardDiff = "1"
 FunctionWrappers = "1.1.3"
 Functors = "0.5"
 GPUArrays = "11"
@@ -129,7 +129,7 @@ MacroTools = "0.5.13"
 Markdown = "1.10"
 Mooncake = "0.5"
 NCCL = "0.1.2"
-NNlib = "0.9.27"
+NNlib = "0.9.33"
 Optimisers = "0.4.6"
 PrecompileTools = "1.2.1"
 Preferences = "1.4.3"

README.md

@@ -271,51 +271,44 @@ format(".")
 
 The full test of `Lux.jl` takes a long time, here's how to test a portion of the code.
 
-For each `@testitem`, there are corresponding `tags`, for example:
+Tests are organized by directories, where each directory contains test files with `@testset`
+blocks. For example, tests for `SkipConnection` are in `test/core_layers/containers_tests.jl`.
 
-```julia
-@testitem "SkipConnection" setup=[SharedTestSetup] tags=[:core_layers]
-```
+#### Running a Specific Test File
 
-For example, let's consider the tests for `SkipConnection`:
+The easiest way to run a specific test is to directly activate the test directory and
+include the test file:
 
 ```julia
-@testitem "SkipConnection" setup=[SharedTestSetup] tags=[:core_layers] begin
-    ...
-end
-```
-
-We can test the group to which `SkipConnection` belongs by testing `core_layers`.
-To do so set the `LUX_TEST_GROUP` environment variable, or rename the tag to
-further narrow the test scope:
+# From the Lux.jl root directory
+using Pkg
+Pkg.activate("test")
 
-```shell
-export LUX_TEST_GROUP="core_layers"
+# Run a specific test file
+include("test/core_layers/containers_tests.jl")
 ```
 
-Or directly modify the default test tag in `runtests.jl`:
+This approach allows you to quickly iterate on specific tests without running the entire
+test suite.
 
-```julia
-# const LUX_TEST_GROUP = lowercase(get(ENV, "LUX_TEST_GROUP", "all"))
-const LUX_TEST_GROUP = lowercase(get(ENV, "LUX_TEST_GROUP", "core_layers"))
-```
+See [ParallelTestRunners.jl](https://github.com/JuliaTesting/ParallelTestRunner.jl) for
+details on executing specific groups of tests.
 
-But be sure to restore the default value "all" before submitting the code.
+#### Running Test Groups via CI
 
-Furthermore if you want to run a specific test based on the name of the testset, you can
-use [TestEnv.jl](https://github.com/JuliaTesting/TestEnv.jl) as follows. Start with activating the Lux environment and then run the following:
+To run a specific group of tests via the test runner, you can pass the directory name as a
+positional argument:
 
-```julia
-using TestEnv; TestEnv.activate(); using ReTestItems;
-
-# Assuming you are in the main directory of Lux
-ReTestItems.runtests("tests/"; name = "NAME OF THE TEST")
+```shell
+julia --project -e 'using Pkg; Pkg.test(test_args=["core_layers"])'
 ```
 
-For the `SkipConnection` tests that would be:
+#### Running All Tests
 
-```julia
-ReTestItems.runtests("tests/"; name = "SkipConnection")
+To run the full test suite:
+
+```shell
+julia --project -e 'using Pkg; Pkg.test()'
 ```
 
 ### 📖 Documentation

lib/LuxLib/Project.toml

@@ -1,6 +1,6 @@
 name = "LuxLib"
 uuid = "82251201-b29d-42c6-8e01-566dec8acb11"
-version = "1.15.3"
+version = "1.15.4"
 authors = ["Avik Pal <avikpal@mit.edu> and contributors"]
 
 [deps]
@@ -68,7 +68,7 @@ TrackerExt = "Tracker"
 cuDNNExt = ["CUDA", "cuDNN"]
 
 [compat]
-AMDGPU = "0.9.6, 1, 2"
+AMDGPU = "2.2"
 AppleAccelerate = "0.4, 0.5"
 ArrayInterface = "7.17.1"
 BLISBLAS = "0.1, 0.2"

lib/MLDataDevices/Project.toml

@@ -1,7 +1,7 @@
 name = "MLDataDevices"
 uuid = "7e8f7934-dd98-4c1a-8fe8-92b47a384d40"
 authors = ["Avik Pal <avikpal@mit.edu> and contributors"]
-version = "1.17.4"
+version = "1.17.5"
 
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
@@ -53,7 +53,7 @@ oneAPIExt = ["GPUArrays", "oneAPI"]
 OpenCLExt = ["GPUArrays", "OpenCL"]
 
 [compat]
-AMDGPU = "1, 2"
+AMDGPU = "2.2"
 Adapt = "4.4"
 CUDA = "5.8"
 ChainRules = "1.51"

lib/WeightInitializers/Project.toml

@@ -1,7 +1,7 @@
 name = "WeightInitializers"
 uuid = "d49dbf32-c5c2-4618-8acc-27bb2598ef2d"
 authors = ["Avik Pal <avikpal@mit.edu> and contributors"]
-version = "1.3.1"
+version = "1.3.2"
 
 [deps]
 ConcreteStructs = "2569d6c7-a4a2-43d3-a901-331e8e4be471"
@@ -26,7 +26,7 @@ GPUArraysExt = "GPUArrays"
 ReactantExt = "Reactant"
 
 [compat]
-AMDGPU = "0.9.6, 1, 2"
+AMDGPU = "2.2"
 CUDA = "5.8"
 ChainRulesCore = "1.25.1"
 ConcreteStructs = "0.2.3"

test/Project.toml

@@ -33,7 +33,6 @@ Preferences = "21216c6a-2e73-6563-6e65-726566657250"
 PythonCall = "6099a3de-0909-46bc-b1f4-468b9a2dfc0d"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 Reactant = "3c362404-f566-11ee-1572-e11a4b42c853"
-Reexport = "189a3867-3050-52da-a836-e630ba90ab69"
 ReverseDiff = "37e2e3b7-166d-5795-8a7a-e32c996b4267"
 Setfield = "efcf1570-3423-57d1-acb7-fd33fddbac46"
 StableRNGs = "860ef19b-820b-49d6-a774-d7a799459cd3"
@@ -53,6 +52,7 @@ LuxTestUtils = {path = "../lib/LuxTestUtils"}
 MLDataDevices = {path = "../lib/MLDataDevices"}
 
 [compat]
+AMDGPU = "2.2"
 ADTypes = "1.19"
 Adapt = "4"
 Aqua = "0.8.4"
@@ -63,7 +63,7 @@ DispatchDoctor = "0.4.12"
 Documenter = "1.4"
 Enzyme = "0.13.120"
 ExplicitImports = "1.14.2"
-ForwardDiff = "0.10.36, 1"
+ForwardDiff = "1"
 Functors = "0.5"
 LinearAlgebra = "1.10"
 Logging = "1.10"
@@ -73,7 +73,7 @@ LuxLib = "1.15"
 LuxTestUtils = "2"
 MLDataDevices = "1.17"
 Mooncake = "0.4.138, 0.5"
-NNlib = "0.9.27"
+NNlib = "0.9.33"
 Octavian = "0.3.28"
 OneHotArrays = "0.2.5"
 Optimisers = "0.4.6"
@@ -83,7 +83,6 @@ Preferences = "1.4.3"
 PythonCall = "0.9"
 Random = "1.10"
 Reactant = "0.2.205"
-Reexport = "1.2.2"
 ReverseDiff = "1.15.3"
 Setfield = "1.1.1"
 StableRNGs = "1.0.2"

test/runtests.jl

@@ -140,8 +140,8 @@ end
 if (isempty(parsed_args.positionals) || "others" ∈ parsed_args.positionals)
     @testset "DispatchDoctor Preferences" begin
         @testset "set_dispatch_doctor_preferences!" begin
-            @test_throws ArgumentError Lux.set_dispatch_doctor_preferences!("invalid")
-            @test_throws ArgumentError Lux.set_dispatch_doctor_preferences!(;
+            @test_throws AssertionError Lux.set_dispatch_doctor_preferences!("invalid")
+            @test_throws AssertionError Lux.set_dispatch_doctor_preferences!(;
                 luxcore="invalid"
             )