Skip to content

Commit 7c73390

Browse files
authored
Update GPUs.md
Added "const cdev = cpu_device() const gdev = gpu_device()" that was missing and was giving error
1 parent 3e2c350 commit 7c73390

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

docs/src/examples/GPUs.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -75,6 +75,8 @@ same code works on CPUs and GPUs, dependent on `using LuxCUDA`.
7575
using Lux, Optimization, OptimizationOptimisers, Zygote, OrdinaryDiffEq, Plots, LuxCUDA,
7676
SciMLSensitivity, Random, ComponentArrays
7777
import DiffEqFlux: NeuralODE
78+
const cdev = cpu_device()
79+
const gdev = gpu_device()
7880
7981
CUDA.allowscalar(false) # Makes sure no slow operations are occurring
8082

0 commit comments

Comments
 (0)