tvm
Classes | Namespaces | Macros
nvtx.h File Reference

NVTX scoped range utility (header-only). More...

#include <string>
Include dependency graph for nvtx.h:

Go to the source code of this file.

Classes

class  tvm::support::NVTXScopedRange
 A class to create a NVTX range. No-op if TVM is not built against NVTX. More...
 

Namespaces

 tvm
 An object that builds and maintains block scope and StmtSref mapping for Dependence analysis.
 
 tvm::support
 

Macros

#define TVM_NVTX_ENABLED   0
 
#define TVM_NVTX_FUNC_SCOPE()    ::tvm::support::NVTXScopedRange _nvtx_func_scope_(__PRETTY_FUNCTION__);
 

Detailed Description

NVTX scoped range utility (header-only).

Provides NVTXScopedRange: a lightweight RAII wrapper over nvtxRangePush/Pop. When TVM_NVTX_ENABLED is not defined or is 0, all methods are no-ops compiled away by the optimizer.

Macro Definition Documentation

◆ TVM_NVTX_ENABLED

#define TVM_NVTX_ENABLED   0

◆ TVM_NVTX_FUNC_SCOPE

#define TVM_NVTX_FUNC_SCOPE ( )     ::tvm::support::NVTXScopedRange _nvtx_func_scope_(__PRETTY_FUNCTION__);