From f3407b3c165a2c99814d3311c953cc5fd2ab46f7 Mon Sep 17 00:00:00 2001 From: Christopher Rackauckas Date: Wed, 18 Oct 2023 11:47:36 -0400 Subject: [PATCH] Update diffeqgpu_setup.Rd --- man/diffeqgpu_setup.Rd | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/man/diffeqgpu_setup.Rd b/man/diffeqgpu_setup.Rd index bd4ec04..612cca0 100644 --- a/man/diffeqgpu_setup.Rd +++ b/man/diffeqgpu_setup.Rd @@ -4,7 +4,7 @@ \alias{diffeqgpu_setup} \title{Setup DiffEqGPU} \usage{ -diffeqgpu_setup() +diffeqgpu_setup(backend="CUDA") } \description{ This function initializes the DiffEqGPU package for GPU-parallelized ensembles. @@ -12,14 +12,14 @@ The first time will be long since it includes precompilation. } \value{ Returns a degpu object which holds the module state of the Julia-side DiffEqGPU - package. The core use is to use degpu$EnsembleGPUArray() for choosing the GPU + package. The core use is to use degpu$EnsembleGPUKernel() for choosing the GPU dispatch in the solve. } \examples{ \dontrun{ ## diffeqgpu_setup() is time-consuming and requires Julia+DifferentialEquations.jl -degpu <- diffeqr::diffeqgpu_setup() +degpu <- diffeqr::diffeqgpu_setup(backend="CUDA") }