From e256c276d957fea855e81bb83205e9dfaeb7c388 Mon Sep 17 00:00:00 2001 From: Avik Pal Date: Sun, 14 Sep 2025 22:29:01 -0400 Subject: [PATCH 1/3] fix: pin ForwardDiff version --- Project.toml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/Project.toml b/Project.toml index fc3452d074..5a4dbc8b97 100644 --- a/Project.toml +++ b/Project.toml @@ -1,7 +1,7 @@ name = "Lux" uuid = "b2108857-7c20-44ae-9111-449ecde12c47" authors = ["Avik Pal and contributors"] -version = "1.21.1" +version = "1.21.2" [deps] ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b" @@ -91,7 +91,7 @@ Enzyme = "0.13.74" EnzymeCore = "0.8.13" FastClosures = "0.3.2" Flux = "0.16.3" -ForwardDiff = "0.10.36, 1" +ForwardDiff = "0.10.36, =1" FunctionWrappers = "1.1.3" Functors = "0.5" GPUArraysCore = "0.2" From 1b293f94d36378767a5144332a543fea63552c05 Mon Sep 17 00:00:00 2001 From: Avik Pal Date: Sun, 14 Sep 2025 22:37:46 -0400 Subject: [PATCH 2/3] fix: pin CUDA version --- lib/LuxCUDA/Project.toml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lib/LuxCUDA/Project.toml b/lib/LuxCUDA/Project.toml index 13115b7263..3c1edd7d46 100644 --- a/lib/LuxCUDA/Project.toml +++ b/lib/LuxCUDA/Project.toml @@ -1,7 +1,7 @@ name = "LuxCUDA" uuid = "d0bbae9a-e099-4d5b-a835-1c6931763bda" authors = ["Avik Pal and contributors"] -version = "0.3.4" +version = "0.3.5" [deps] CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba" @@ -9,7 +9,7 @@ Reexport = "189a3867-3050-52da-a836-e630ba90ab69" cuDNN = "02a925ec-e4fe-4b08-9a7e-0d78e3d38ccd" [compat] -CUDA = "5.8" +CUDA = "5.8 - 5.8.1" Reexport = "1" cuDNN = "1.3" julia = "1.10" From fb0a3c26858b4bd28b299dd91efd0f73269efb46 Mon Sep 17 00:00:00 2001 From: Avik Pal Date: Sun, 14 Sep 2025 22:59:27 -0400 Subject: [PATCH 3/3] fix: use special NCCL --- lib/LuxCUDA/Project.toml | 4 ++-- test/distributed/Project.toml | 1 + test/distributed/common_distributedtest.jl | 2 +- test/distributed/data_distributedtest.jl | 2 +- test/distributed/optimizer_distributedtest.jl | 2 +- test/distributed/synchronize_distributedtest.jl | 2 +- 6 files changed, 7 insertions(+), 6 deletions(-) diff --git a/lib/LuxCUDA/Project.toml b/lib/LuxCUDA/Project.toml index 3c1edd7d46..13115b7263 100644 --- a/lib/LuxCUDA/Project.toml +++ b/lib/LuxCUDA/Project.toml @@ -1,7 +1,7 @@ name = "LuxCUDA" uuid = "d0bbae9a-e099-4d5b-a835-1c6931763bda" authors = ["Avik Pal and contributors"] -version = "0.3.5" +version = "0.3.4" [deps] CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba" @@ -9,7 +9,7 @@ Reexport = "189a3867-3050-52da-a836-e630ba90ab69" cuDNN = "02a925ec-e4fe-4b08-9a7e-0d78e3d38ccd" [compat] -CUDA = "5.8 - 5.8.1" +CUDA = "5.8" Reexport = "1" cuDNN = "1.3" julia = "1.10" diff --git a/test/distributed/Project.toml b/test/distributed/Project.toml index db487a7148..8cc79f4b0c 100644 --- a/test/distributed/Project.toml +++ b/test/distributed/Project.toml @@ -14,6 +14,7 @@ Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40" Lux = {path = "../.."} LuxCUDA = {path = "../../lib/LuxCUDA"} MLDataDevices = {path = "../../lib/MLDataDevices"} +NCCL = { url = "https://github.com/LuxDL/NCCL.jl", rev = "cd85e9f7830be425b64d2eee407df7d61a6a9635" } [compat] ComponentArrays = "0.15.29" diff --git a/test/distributed/common_distributedtest.jl b/test/distributed/common_distributedtest.jl index 696405501b..d96c0b59a2 100644 --- a/test/distributed/common_distributedtest.jl +++ b/test/distributed/common_distributedtest.jl @@ -11,7 +11,7 @@ end const backend_type = input_args[2] == "nccl" ? NCCLBackend : MPIBackend -if input_args[1] == "nccl" +if input_args[2] == "nccl" using NCCL end diff --git a/test/distributed/data_distributedtest.jl b/test/distributed/data_distributedtest.jl index ab6a31e71c..0eb36c627e 100644 --- a/test/distributed/data_distributedtest.jl +++ b/test/distributed/data_distributedtest.jl @@ -17,7 +17,7 @@ end const backend_type = input_args[2] == "nccl" ? NCCLBackend : MPIBackend -if input_args[1] == "nccl" +if input_args[2] == "nccl" using NCCL end diff --git a/test/distributed/optimizer_distributedtest.jl b/test/distributed/optimizer_distributedtest.jl index 4b728ec5c9..ef0ab5ee44 100644 --- a/test/distributed/optimizer_distributedtest.jl +++ b/test/distributed/optimizer_distributedtest.jl @@ -11,7 +11,7 @@ end const backend_type = input_args[2] == "nccl" ? NCCLBackend : MPIBackend -if input_args[1] == "nccl" +if input_args[2] == "nccl" using NCCL end diff --git a/test/distributed/synchronize_distributedtest.jl b/test/distributed/synchronize_distributedtest.jl index 5218dd927c..cda7ff3737 100644 --- a/test/distributed/synchronize_distributedtest.jl +++ b/test/distributed/synchronize_distributedtest.jl @@ -11,7 +11,7 @@ end const backend_type = input_args[2] == "nccl" ? NCCLBackend : MPIBackend -if input_args[1] == "nccl" +if input_args[2] == "nccl" using NCCL end