Skip to content

Commit 8b6e66b

Browse files
authored
Add kernel to EnsembleGPUKernel documentaion
1 parent e9b592c commit 8b6e66b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/algorithms.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -146,7 +146,7 @@ prob = ODEProblem{false}(lorenz, u0, tspan, p)
146146
prob_func = (prob, i, repeat) -> remake(prob, p = (@SVector rand(Float32, 3)) .* p)
147147
monteprob = EnsembleProblem(prob, prob_func = prob_func, safetycopy = false)
148148
149-
@time sol = solve(monteprob, GPUTsit5(), EnsembleGPUKernel(), trajectories = 10_000,
149+
@time sol = solve(monteprob, GPUTsit5(), EnsembleGPUKernel(CUDA.CUDABackend()), trajectories = 10_000,
150150
adaptive = false, dt = 0.1f0)
151151
```
152152
"""

0 commit comments

Comments
 (0)