Package | Description |
---|---|
ai.rapids.cudf |
Modifier and Type | Method and Description |
---|---|
static void |
Rmm.clearEventHandler()
Clears the active RMM event handler if one is set.
|
static void |
Rmm.initialize(int allocationMode,
Rmm.LogConf logConf,
long poolSize)
Initialize memory manager state and storage.
|
static boolean |
Rmm.isInitialized()
Check if RMM has been initialized already or not.
|
static void |
Rmm.setEventHandler(RmmEventHandler handler)
Sets the event handler to be called on RMM events (e.g.: allocation failure).
|
static void |
Rmm.setEventHandler(RmmEventHandler handler,
boolean enableDebug)
Sets the event handler to be called on RMM events (e.g.: allocation failure) and
optionally enable debug mode (callbacks on every allocate and deallocate)
|
static void |
Rmm.shutdown()
Shut down any initialized RMM instance.
|
static void |
Rmm.shutdown(long forceGCInterval,
long maxWaitTime,
TimeUnit units)
Shut down any initialized RMM instance.
|
Copyright © 2024. All rights reserved.