| 1552 | |
| 1553 | |
| 1554 | def measure_warm(loader: LoaderFn, case: dict[str, Any], warmup_request: dict[str, Any], request: dict[str, Any], args: argparse.Namespace) -> dict[str, Any]: |
| 1555 | _model, run = loader(case, args) |
| 1556 | run(warmup_request, args) |
| 1557 | wall_ms: list[float] = [] |
| 1558 | duration_sec = 0.0 |
| 1559 | for _ in range(3): |
| 1560 | run_wall_ms, duration_sec = run(request, args) |
| 1561 | wall_ms.append(run_wall_ms) |
| 1562 | average_wall_ms = float(sum(wall_ms) / len(wall_ms)) |
| 1563 | return { |
| 1564 | "wall_ms": wall_ms, |
| 1565 | "average_wall_ms": average_wall_ms, |
| 1566 | "duration_sec": duration_sec, |
| 1567 | "rtf": (average_wall_ms / 1000.0 / duration_sec) if duration_sec > 0.0 else 0.0, |
| 1568 | } |
| 1569 | |
| 1570 | |
| 1571 | def main() -> int: |