ai.rapids.cudf

Class NvtxRange

java.lang.Object

ai.rapids.cudf.NvtxRange

All Implemented Interfaces:

AutoCloseable

public class NvtxRange
extends Object
implements AutoCloseable

Utility class to mark an NVTX profiling range. The constructor pushes an NVTX range and the close method pops off the most recent range that was pushed. Therefore instances of this class should always be used in a try-with-resources block to guarantee that ranges are always closed in the proper order. For example:

   try (NvtxRange a = new NvtxRange("a", NvtxColor.RED)) {
     ...
     try (NvtxRange b = new NvtxRange("b", NvtxColor.BLUE)) {
       ...
     }
     ...
   }
 

Instances should be associated with a single thread to avoid pushing an NVTX range in one thread and then trying to pop the range in a different thread.

Constructor Summary

Constructors

Constructor and Description
NvtxRange(String name, int colorBits)
NvtxRange(String name, NvtxColor color)

Method Summary

Modifier and Type	Method and Description
void	close()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

NvtxRange

public NvtxRange(String name,
                 NvtxColor color)

NvtxRange

public NvtxRange(String name,
                 int colorBits)

Method Detail

close

public void close()

Specified by:

close in interface AutoCloseable