wehub-resource-sync
57 lines
1.8 KiB
Python
57 lines
1.8 KiB
Python
# SPDX-License-Identifier: Apache-2.0
|
|
"""``lmcache trace info`` — print a summary of a trace file."""
|
|
|
|
# Future
|
|
from __future__ import annotations
|
|
|
|
# Standard
|
|
from collections import Counter
|
|
import argparse
|
|
|
|
# First Party
|
|
from lmcache.cli.commands.base import BaseCommand
|
|
|
|
|
|
class InfoCommand(BaseCommand):
|
|
"""Print a summary of a trace file."""
|
|
|
|
def name(self) -> str:
|
|
return "info"
|
|
|
|
def help(self) -> str:
|
|
return "Print a summary of a trace file."
|
|
|
|
def add_arguments(self, parser: argparse.ArgumentParser) -> None:
|
|
parser.add_argument(
|
|
"trace_path",
|
|
metavar="FILE",
|
|
help="Path to a .lct trace file.",
|
|
)
|
|
|
|
def execute(self, args: argparse.Namespace) -> None:
|
|
# First Party
|
|
from lmcache.v1.mp_observability.trace.reader import TraceReader
|
|
|
|
with TraceReader(args.trace_path) as r:
|
|
header = r.header
|
|
counts: Counter[str] = Counter()
|
|
max_mono = 0.0
|
|
for record in r.records():
|
|
counts[record.qualname] += 1
|
|
if record.t_mono > max_mono:
|
|
max_mono = record.t_mono
|
|
|
|
print(f"Trace file: {args.trace_path}")
|
|
print(f" level: {header.level}")
|
|
print(f" format_version: {header.format_version}")
|
|
print(f" trace_schema_version: {header.trace_schema_version}")
|
|
print(f" duration: {max_mono:.3f}s")
|
|
print(f" sm_config_digest: {header.sm_config_digest or '(none)'}")
|
|
print(f" total_records: {sum(counts.values())}")
|
|
if counts:
|
|
print(" ops:")
|
|
for qn in sorted(counts):
|
|
print(f" {qn}: {counts[qn]}")
|
|
else:
|
|
print(" ops: (none)")
|