add a CUDA.jl reference page

FluxML · Nov 28, 2022 · c2a8bf0 · c2a8bf0
1 parent 46bd934
commit c2a8bf0
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/docs/make.jl b/docs/make.jl
@@ -35,6 +35,7 @@ makedocs(
             "Shape Inference 📚" => "outputsize.md",
             "Flat vs. Nested 📚" => "destructure.md",
             "Callback Helpers 📚" => "training/callbacks.md",
+            "CUDA.jl 📚 (`cu`, `CuIterator`, ...)" => "CUDA.md",
             "NNlib.jl 📚 (`softmax`, `conv`, ...)" => "models/nnlib.md",
             "Zygote.jl 📚 (`gradient`, ...)" => "training/zygote.md",
             "MLUtils.jl 📚 (`DataLoader`, ...)" => "data/mlutils.md",

diff --git a/docs/src/CUDA.md b/docs/src/CUDA.md
@@ -0,0 +1,14 @@
+# CUDA.jl
+
+
+```@docs
+CUDA.cu
+CUDA.AbstractGPUArray
+CUDA.CuIterator
+```
+
+```@docs
+CUDA.functional
+CUDA.allowscalar
+CUDA.@allowscalar
+```