Package ai.rapids.cudf

Class RmmLoggingResourceAdaptor<C extends RmmDeviceMemoryResource>

java.lang.Object

ai.rapids.cudf.RmmWrappingDeviceMemoryResource<C>

ai.rapids.cudf.RmmLoggingResourceAdaptor<C>

All Implemented Interfaces:

RmmDeviceMemoryResource, AutoCloseable

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

A device memory resource that will log interactions.

Field Summary

Fields inherited from class ai.rapids.cudf.RmmWrappingDeviceMemoryResource

wrapped

Constructor Summary

Constructors

Constructor	Description
RmmLoggingResourceAdaptor(C wrapped, Rmm.LogConf conf, boolean autoFlush)	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

RmmLoggingResourceAdaptor

public RmmLoggingResourceAdaptor(C wrapped,
 Rmm.LogConf conf,
 boolean autoFlush)

Create a new logging resource adaptor.

Parameters:

wrapped - the memory resource to log interactions with. This should not be reused.

conf - the config of where this should be logged to

autoFlush - should the results be flushed after each entry or not.

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