ai.rapids.cudf

Class RmmLimitingResourceAdaptor<C extends RmmDeviceMemoryResource>

java.lang.Object

ai.rapids.cudf.RmmWrappingDeviceMemoryResource<C>

ai.rapids.cudf.RmmLimitingResourceAdaptor<C>

All Implemented Interfaces:

RmmDeviceMemoryResource, AutoCloseable

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

A device memory resource that will limit the maximum amount allocated.

Field Summary

Fields inherited from class ai.rapids.cudf.RmmWrappingDeviceMemoryResource

wrapped

Constructor Summary

Constructors

Constructor and Description
RmmLimitingResourceAdaptor(C wrapped, long limit, long alignment) Create a new limiting resource adaptor.

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
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

RmmLimitingResourceAdaptor

public RmmLimitingResourceAdaptor(C wrapped,
                                  long limit,
                                  long alignment)

Create a new limiting resource adaptor.

Parameters:

wrapped - the memory resource to limit. This should not be reused.

limit - the allocation limit in bytes

alignment - the alignment

Method Detail

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