ai.rapids.cudf

Class RmmPoolMemoryResource<C extends RmmDeviceMemoryResource>

java.lang.Object

ai.rapids.cudf.RmmWrappingDeviceMemoryResource<C>

ai.rapids.cudf.RmmPoolMemoryResource<C>

All Implemented Interfaces:

RmmDeviceMemoryResource, AutoCloseable

public class RmmPoolMemoryResource<C extends RmmDeviceMemoryResource>
extends RmmWrappingDeviceMemoryResource<C>

A device memory resource that will pre-allocate a pool of resources and sub-allocate from this pool to improve memory performance.

Field Summary

Fields inherited from class ai.rapids.cudf.RmmWrappingDeviceMemoryResource

wrapped

Constructor Summary

Constructors

Constructor and Description
RmmPoolMemoryResource(C wrapped, long initSize, long maxSize) Create a new pooled memory resource taking ownership of the RmmDeviceMemoryResource that it is wrapping.

Method Summary

Modifier and Type	Method and Description
void	close()
long	getHandle() Returns a pointer to the underlying C++ class that implements rmm::mr::device_memory_resource
long	getMaxSize()
String	toString()

Methods inherited from class ai.rapids.cudf.RmmWrappingDeviceMemoryResource

getWrapped, releaseWrapped

Methods inherited from class java.lang.Object

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

Constructor Detail

RmmPoolMemoryResource

public RmmPoolMemoryResource(C wrapped,
                             long initSize,
                             long maxSize)

Create a new pooled memory resource taking ownership of the RmmDeviceMemoryResource that it is wrapping.

Parameters:

wrapped - the memory resource to use for the pool. This should not be reused.

initSize - the size of the initial pool

maxSize - the size of the maximum pool

Method Detail

getMaxSize

public long getMaxSize()

getHandle

public long getHandle()

Description copied from interface: RmmDeviceMemoryResource

Returns a pointer to the underlying C++ class that implements rmm::mr::device_memory_resource

close

public void close()

Specified by:

close in interface RmmDeviceMemoryResource

Specified by:

close in interface AutoCloseable

Overrides:

close in class RmmWrappingDeviceMemoryResource<C extends RmmDeviceMemoryResource>

toString

public String toString()

Overrides:

toString in class Object