Drop NNlibCUDA dependency

.buildkite/pipeline.yml

@@ -19,9 +19,11 @@ steps:
         julia:
           - "1"
           - "1.6"
-          - "1.9-nightly"
           - "nightly"
       adjustments:
+        - with:
+            julia: "1.6"
+          soft_fail: true
         - with:
             julia: "nightly"
           soft_fail: true

.github/workflows/CI.yml

@@ -20,7 +20,6 @@ jobs:
         version:
           - "1"
           - "1.6"
-          - "~1.9.0-0"
     steps:
       - uses: actions/checkout@v3
       - uses: julia-actions/setup-julia@v1
@@ -44,4 +43,3 @@ jobs:
       - uses: codecov/codecov-action@v3
         with:
           files: lcov.info
-          flags: ${{ matrix.group }}

Project.toml

@@ -1,17 +1,15 @@
 name = "LuxCUDA"
 uuid = "d0bbae9a-e099-4d5b-a835-1c6931763bda"
 authors = ["Avik Pal <[email protected]> and contributors"]
-version = "0.1.2"
+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.1"
-NNlibCUDA = "0.2"
+CUDA = "4"
 Reexport = "1"
 cuDNN = "1"
 julia = "1.6"

src/LuxCUDA.jl

@@ -2,7 +2,7 @@ module LuxCUDA
 
 using Reexport
 
-@reexport using CUDA, CUDA.CUDAKernels, NNlibCUDA, cuDNN
+@reexport using CUDA, CUDA.CUDAKernels, cuDNN
 
 const USE_CUDA_GPU = Ref{Union{Nothing, Bool}}(nothing)