From c2a8bf0f15a9e896e8e7e116f2fbe451b5b84e08 Mon Sep 17 00:00:00 2001 From: Michael Abbott <32575566+mcabbott@users.noreply.github.com> Date: Mon, 28 Nov 2022 14:34:33 -0500 Subject: [PATCH] add a CUDA.jl reference page --- docs/make.jl | 1 + docs/src/CUDA.md | 14 ++++++++++++++ 2 files changed, 15 insertions(+) create mode 100644 docs/src/CUDA.md diff --git a/docs/make.jl b/docs/make.jl index 1f791149a3..2540060bbe 100644 --- 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 new file mode 100644 index 0000000000..32cdffa863 --- /dev/null +++ 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 +``` \ No newline at end of file