ai.rapids.cudf

Class Rmm

java.lang.Object

ai.rapids.cudf.Rmm

public class Rmm
extends Object

This is the binding class for rmm lib.

Constructor Summary

Constructors

Constructor and Description
Rmm()

Method Summary

Modifier and Type	Method and Description
static String	getLog() If logging is enabled get the log as a String.
static void	initialize(int allocationMode, boolean enableLogging, long poolSize) ---------------------------------------------------------------------------* Initialize memory manager state and storage.
static boolean	isInitialized() Check if RMM has been initialized already or not.
static void	shutdown() Shut down any initialized rmm.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

Rmm

public Rmm()

Method Detail

initialize

public static void initialize(int allocationMode,
                              boolean enableLogging,
                              long poolSize)
                       throws RmmException

---------------------------------------------------------------------------* Initialize memory manager state and storage.

Parameters:

allocationMode - Allocation strategy to use. Bit set using RmmAllocationMode.CUDA_DEFAULT, RmmAllocationMode.POOL and RmmAllocationMode.CUDA_MANAGED_MEMORY

enableLogging - Enable logging memory manager events

poolSize - The initial pool size in bytes

Throws:

RmmException

isInitialized

public static boolean isInitialized()
                             throws RmmException

Check if RMM has been initialized already or not.

Throws:

RmmException

shutdown

public static void shutdown()
                     throws RmmException

Shut down any initialized rmm.

Throws:

RmmException

getLog

public static String getLog()
                     throws RmmException

If logging is enabled get the log as a String.

Throws:

RmmException