API ReferenceOther APIsAituneUtilsMonitoring

aitune.utils.monitoring.nvtx_annotation

View as Markdown

Annotation for NVTX.

Module Contents

Classes

NameDescription
annotate_with_nvtxDecorator/context manager for nvtx annotations.

Data

NVTX_ANNOTATION_DOMAIN

API

class aitune.utils.monitoring.nvtx_annotation.annotate_with_nvtx(
    name: str | None = None,
    color: str | int | None = None
)

Decorator/context manager for nvtx annotations.

aitune.utils.monitoring.nvtx_annotation.annotate_with_nvtx.__call__(
    func
)

Wraps the function as a decorator.

aitune.utils.monitoring.nvtx_annotation.annotate_with_nvtx.__enter__()

Enters the context manager.

aitune.utils.monitoring.nvtx_annotation.annotate_with_nvtx.__exit__(
    exc_type,
    exc_value,
    traceback
)

Exits the context manager.

aitune.utils.monitoring.nvtx_annotation.annotate_with_nvtx._post_init()

Post-initialization.

aitune.utils.monitoring.nvtx_annotation.NVTX_ANNOTATION_DOMAIN = 'AITune'