RmmTrackingResourceAdaptor

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

A device memory resource that will track some basic statistics about the memory usage.

Constructor Summary

RmmTrackingResourceAdaptor(C wrapped, long alignment)

Create a new tracking resource adaptor.

Method Summary

close()
getHandle()
getMaxTotalBytesAllocated()
getScopedMaxTotalBytesAllocated()
getTotalBytesAllocated()
resetScopedMaxTotalBytesAllocated(long initValue)
toString()

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

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

Constructors

public RmmTrackingResourceAdaptor(C wrapped, long alignment)

Create a new tracking resource adaptor.

Parameters:

• wrapped – the memory resource to track allocations. This should not be reused.

• alignment – the alignment to apply.

Methods

public long getHandle()

public long getTotalBytesAllocated()

public long getMaxTotalBytesAllocated()

public void resetScopedMaxTotalBytesAllocated(long initValue)

public long getScopedMaxTotalBytesAllocated()

public void close()

public String toString()