Package ai.rapids.cudf

Class RmmEventHandlerResourceAdaptor<C extends RmmDeviceMemoryResource>

java.lang.Object

ai.rapids.cudf.RmmWrappingDeviceMemoryResource<C>

ai.rapids.cudf.RmmEventHandlerResourceAdaptor<C>

All Implemented Interfaces:

RmmDeviceMemoryResource, AutoCloseable

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

A device memory resource that will give callbacks in specific situations.

Field Summary

Fields inherited from class ai.rapids.cudf.RmmWrappingDeviceMemoryResource

wrapped

Constructor Summary

Constructors

Constructor	Description
RmmEventHandlerResourceAdaptor(C wrapped, RmmTrackingResourceAdaptor<?> tracker, RmmEventHandler handler, boolean debug)	Create a new logging resource adaptor.

Method Summary

Modifier and Type	Method	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 Details

RmmEventHandlerResourceAdaptor

public RmmEventHandlerResourceAdaptor(C wrapped,
 RmmTrackingResourceAdaptor<?> tracker,
 RmmEventHandler handler,
 boolean debug)

Create a new logging resource adaptor.

Parameters:

wrapped - the memory resource to get callbacks for. This should not be reused.

handler - the handler that will get the callbacks

tracker - the tracking event handler

debug - true if you want all the callbacks, else false

Method Details

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 AutoCloseable

Specified by:

close in interface RmmDeviceMemoryResource

Overrides:

close in class RmmWrappingDeviceMemoryResource<C extends RmmDeviceMemoryResource>

toString

public String toString()

Overrides:

toString in class Object