Skip to content

Commit e84e2cc

Browse files
tzhenghaomodularbot
authored andcommitted
[Benchmarking] Migrate serving sweep CLI from argparse to cyclopts
Replace the hand-rolled argparse parser in sweep-benchmark-serving.py with cyclopts, using `SweepServingBenchmarkConfig` directly as the CLI model. Adds a `collect_gpu_stats `field to SweepServingBenchmarkConfig and wires it through. MODULAR_ORIG_COMMIT_REV_ID: bb5a4de312dd81111fb27f7ce21f81578a5b92de
1 parent 2e9e11f commit e84e2cc

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

  • max/python/max/benchmark/benchmark_shared

max/python/max/benchmark/benchmark_shared/config.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1007,6 +1007,12 @@ class SweepServingBenchmarkConfig(ServingBenchmarkConfig):
10071007
},
10081008
)
10091009

1010+
collect_gpu_stats: bool = Field(
1011+
default=True,
1012+
description="Enable GPU stats collection for serving benchmarks.",
1013+
json_schema_extra={"group": "Control Flags"},
1014+
)
1015+
10101016
@classmethod
10111017
def get_default_required_fields(cls) -> set[str]:
10121018
"""Get required fields for the sweep benchmark config."""

0 commit comments

Comments
 (0)