Last julia<1.9 version

LuxDL · Jul 4, 2023 · cd66b23 · cd66b23
1 parent ad30f8b
commit cd66b23
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/Project.toml b/Project.toml
@@ -5,11 +5,13 @@ version = "0.2.0"
 
 [deps]
 CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
+NNlibCUDA = "a00861dc-f156-4864-bf3c-e6376f28a68d"
 Reexport = "189a3867-3050-52da-a836-e630ba90ab69"
 cuDNN = "02a925ec-e4fe-4b08-9a7e-0d78e3d38ccd"
 
 [compat]
 CUDA = "4"
+NNlibCUDA = "0.2"
 Reexport = "1"
 cuDNN = "1"
 julia = "1.6"
diff --git a/src/LuxCUDA.jl b/src/LuxCUDA.jl
@@ -2,7 +2,7 @@ module LuxCUDA
 
 using Reexport
 
-@reexport using CUDA, CUDA.CUDAKernels, cuDNN
+@reexport using CUDA, CUDA.CUDAKernels, cuDNN, NNlibCUDA
 
 const USE_CUDA_GPU = Ref{Union{Nothing, Bool}}(nothing)