RmmPoolMemoryResource

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.

Constructor Summary

RmmPoolMemoryResource(C wrapped, long initSize, long maxSize)

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

Method Summary

close()
getHandle()
getMaxSize()
toString()

Fields inherited from class ai.rapids.cudf.RmmWrappingDeviceMemoryResource: wrapped

Methods inherited from class ai.rapids.cudf.RmmWrappingDeviceMemoryResource: getWrapped, releaseWrapped

Constructors

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

Methods

public long getMaxSize()

public long getHandle()

public void close()

public String toString()