Update diffeqr.R

SciML · Oct 18, 2023 · 0840ef8 · 0840ef8
1 parent 6ac30c9
commit 0840ef8
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/R/diffeqr.R b/R/diffeqr.R
@@ -117,6 +117,8 @@ jitoptimize_sde <- function (de,prob){
 #' This function initializes the DiffEqGPU package for GPU-parallelized ensembles.
 #' The first time will be long since it includes precompilation.
 #'
+#' @param backend the backend for the GPU computation. Choices are "CUDA", "AMDGPU", "Metal", or "oneAPI"
+#'
 #' @examples
 #'
 #' \dontrun{ ## diffeq_setup() is time-consuming and requires Julia+DifferentialEquations.jl