ai.rapids.cudf

Class DefaultHostMemoryAllocator

java.lang.Object

ai.rapids.cudf.DefaultHostMemoryAllocator

All Implemented Interfaces:

HostMemoryAllocator

public class DefaultHostMemoryAllocator
extends Object
implements HostMemoryAllocator

Constructor Summary

Constructors

Constructor and Description
DefaultHostMemoryAllocator()

Method Summary

Modifier and Type	Method and Description
HostMemoryBuffer	allocate(long bytes) Allocate memory, but be sure to close the returned buffer to avoid memory leaks.
HostMemoryBuffer	allocate(long bytes, boolean preferPinned) Allocate memory, but be sure to close the returned buffer to avoid memory leaks.
static HostMemoryAllocator	get() Retrieve current host memory allocator used by default if not passed directly to API
static void	set(HostMemoryAllocator hostMemoryAllocator) Sets a new default host memory allocator implementation by default.

Methods inherited from class java.lang.Object

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

Constructor Detail

DefaultHostMemoryAllocator

public DefaultHostMemoryAllocator()

Method Detail

get

public static HostMemoryAllocator get()

Retrieve current host memory allocator used by default if not passed directly to API

Returns:

current default HostMemoryAllocator implementation

set

public static void set(HostMemoryAllocator hostMemoryAllocator)

Sets a new default host memory allocator implementation by default.

Parameters:

hostMemoryAllocator -

allocate

public HostMemoryBuffer allocate(long bytes,
                                 boolean preferPinned)

Description copied from interface: HostMemoryAllocator

Allocate memory, but be sure to close the returned buffer to avoid memory leaks.

Specified by:

allocate in interface HostMemoryAllocator

Parameters:

bytes - size in bytes to allocate

preferPinned - If set to true, the pinned memory pool will be used if possible with a fallback to off-heap memory. If set to false, the allocation will always be from off-heap memory.

Returns:

the newly created buffer

allocate

public HostMemoryBuffer allocate(long bytes)

Description copied from interface: HostMemoryAllocator

Allocate memory, but be sure to close the returned buffer to avoid memory leaks. Pinned memory for allocations preference is up to the implementor

Specified by:

allocate in interface HostMemoryAllocator

Parameters:

bytes - size in bytes to allocate

Returns:

the newly created buffer