Reexport NNlib

LuxDL · Apr 18, 2023 · 66bbb0b · 66bbb0b
1 parent 3dfd2e3
commit 66bbb0b
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 7 deletions.
diff --git a/Project.toml b/Project.toml
@@ -9,6 +9,7 @@ KernelAbstractions = "63c18a36-062a-441e-b654-da1e3ab1ce7c"
 Markdown = "d6f4376e-aef5-505a-96c1-9c027394607a"
 NNlib = "872c559c-99b0-510c-b3b7-b6c96a88d5cd"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
+Reexport = "189a3867-3050-52da-a836-e630ba90ab69"
 Requires = "ae029012-a4dd-5104-9daa-d747884805df"
 Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
 
@@ -25,19 +26,20 @@ LuxLibLuxCUDATrackerExt = ["LuxCUDA", "Tracker"]
 LuxLibReverseDiffExt = "ReverseDiff"
 LuxLibTrackerExt = "Tracker"
 
-[extras]
-ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210"
-LuxCUDA = "d0bbae9a-e099-4d5b-a835-1c6931763bda"
-ReverseDiff = "37e2e3b7-166d-5795-8a7a-e32c996b4267"
-Tracker = "9f7883ad-71c0-57eb-9f7f-b5c9e6d3789c"
-
 [compat]
 ChainRulesCore = "1"
 ForwardDiff = "0.10"
 KernelAbstractions = "0.9"
 LuxCUDA = "0.1"
 NNlib = "0.8"
+Reexport = "1"
 Requires = "1"
 ReverseDiff = "1"
 Tracker = "0.2"
 julia = "1.6"
+
+[extras]
+ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210"
+LuxCUDA = "d0bbae9a-e099-4d5b-a835-1c6931763bda"
+ReverseDiff = "37e2e3b7-166d-5795-8a7a-e32c996b4267"
+Tracker = "9f7883ad-71c0-57eb-9f7f-b5c9e6d3789c"
diff --git a/src/LuxLib.jl b/src/LuxLib.jl
@@ -1,8 +1,12 @@
 module LuxLib
 
-using ChainRulesCore, Markdown, NNlib, Random, Statistics
+using Reexport
+
+using ChainRulesCore, Markdown, Random, Statistics
 import ChainRulesCore as CRC
 
+@reexport using NNlib
+
 using KernelAbstractions
 import KernelAbstractions as KA