R/Gpu.R

In OHDSI/Cyclops: Cyclic Coordinate Descent for Logistic, Poisson and Survival Analysis

#' @title Set GPU device
#'
#' @description
#' \code{setOpenCLDevice} set GPU device
#'
#' @param name String: Name of GPU device
#'
#' @export
setOpenCLDevice <- function(name) {
    devices <- listGPUDevices()

    if (!(name %in% devices)) {
        stop("Unable to find device.")
    }

    Sys.setenv(BOOST_COMPUTE_DEFAULT_DEVICE = name)
    # TODO set CUDA device
}