use_cpu: Disable 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 CPU computing when calling user-defined operators.

Usage

use_cpu(ncore = 0)

Arguments

ncore

integer, number of cores used by rkeops user-defined operators to run computations in CPU mode. Default value is 0 and all available cores are used.

Details

Note 1: By default, rkeops user-defined operators run computations on CPU (even for GPU-compiled operators), thus calling the function use_gpu() is mandatory to run computations on GPU.

Note 2: By default, in CPU mode, rkeops user-defined operators run computations on all available cores for parallel computing. To control, the number of cores used by rkeops user-defined operators, you can used the input parameter ncore.

Value

None

Author(s)

Ghislain Durif

See Also

compile4cpu(), compile4gpu(), use_gpu()

Examples

library(rkeops)
use_cpu()

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