use_gpu: Enable GPU-computing when calling user-defined operators

In rkeops: Kernel Operations on GPU or CPU, with Autodiff, without Memory Overflows

Description

Set up rkeops runtime options to use GPU computing when calling user-defined operators.

Usage

use_gpu(device = 0)

Arguments

device

integer, GPU device id to be used for computations. Default is 0. It is recommended to use default GPU and manage GPU assignment outside R by setting the environment variable CUDA_VISIBLE_DEVICES.

Details

If you have compiled GPU-compatible operators (see compile4gpu()), you can call the function use_gpu to specificly run computations on GPU.

Note: The default behavior in rkeops is to use CPU computing, thus calling the function use_gpu is mandatory to run computations on GPU.

To disable GPU computing, run use_cpu().

Value

None

Author(s)

Ghislain Durif

See Also

compile4cpu(), compile4gpu(), use_cpu()

Examples

library(rkeops)
use_gpu()

rkeops documentation built on Feb. 17, 2021, 5:08 p.m.