()
| 31 | |
| 32 | |
| 33 | def main() -> None: |
| 34 | parser = argparse.ArgumentParser(description="Benchmark OpenQuant Python pipeline runtime.") |
| 35 | parser.add_argument("--iterations", type=int, default=30) |
| 36 | parser.add_argument("--bars", type=int, default=2048) |
| 37 | parser.add_argument("--seed", type=int, default=42) |
| 38 | args = parser.parse_args() |
| 39 | |
| 40 | out = run_benchmark(iterations=args.iterations, bars=args.bars, seed=args.seed) |
| 41 | print( |
| 42 | "pipeline_bench " |
| 43 | f"iterations={int(out['iterations'])} " |
| 44 | f"bars={int(out['bars_per_run'])} " |
| 45 | f"mean_ms={out['mean_ms']:.3f} " |
| 46 | f"p95_ms={out['p95_ms']:.3f} " |
| 47 | f"runs_per_sec={out['runs_per_sec']:.2f}" |
| 48 | ) |
| 49 | |
| 50 | |
| 51 | if __name__ == "__main__": |
no test coverage detected