Skip to content

Commit c2a8bf0

Browse files
committed
add a CUDA.jl reference page
1 parent 46bd934 commit c2a8bf0

File tree

2 files changed

+15
-0
lines changed

2 files changed

+15
-0
lines changed

docs/make.jl

+1
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,7 @@ makedocs(
3535
"Shape Inference 📚" => "outputsize.md",
3636
"Flat vs. Nested 📚" => "destructure.md",
3737
"Callback Helpers 📚" => "training/callbacks.md",
38+
"CUDA.jl 📚 (`cu`, `CuIterator`, ...)" => "CUDA.md",
3839
"NNlib.jl 📚 (`softmax`, `conv`, ...)" => "models/nnlib.md",
3940
"Zygote.jl 📚 (`gradient`, ...)" => "training/zygote.md",
4041
"MLUtils.jl 📚 (`DataLoader`, ...)" => "data/mlutils.md",

docs/src/CUDA.md

+14
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
# CUDA.jl
2+
3+
4+
```@docs
5+
CUDA.cu
6+
CUDA.AbstractGPUArray
7+
CUDA.CuIterator
8+
```
9+
10+
```@docs
11+
CUDA.functional
12+
CUDA.allowscalar
13+
CUDA.@allowscalar
14+
```

0 commit comments

Comments
 (0)