aitune.torch.task.profiling.events | NVIDIA AITune Documentation

Profiling result events.

Module Contents

Classes

Name	Description
`ProfilingResultEvent`	Profiling result event.

Functions

Name	Description
`get_inference_events`	Get inference events.

API

class aitune.torch.task.profiling.events.ProfilingResultEvent(
    timestamp: float,
    model_name: str,
    backend_details: str,
    batch_size: int,
    phase: str,
    measurement_id: int = 0,
    execution_time: float = 0.0
)

Dataclass

Profiling result event.

backend_details

str

batch_size

int

execution_time

float = 0.0

measurement_id

int = 0

model_name

str

phase

str

timestamp

float

aitune.torch.task.profiling.events.get_inference_events(
    events: list[aitune.torch.task.profiling.events.ProfilingResultEvent]
) -> list[aitune.torch.task.profiling.events.ProfilingResultEvent]

Get inference events.