Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
c7a78b382a |
No files matched your search
@@ -653,6 +653,7 @@ def run_benchmark(
|
|||||||
random_input_len: Any | None = None,
|
random_input_len: Any | None = None,
|
||||||
random_output_len: Any | None = None,
|
random_output_len: Any | None = None,
|
||||||
resource_label: str | None = None,
|
resource_label: str | None = None,
|
||||||
|
num_warmups: int = 2,
|
||||||
) -> dict[str, Any]:
|
) -> dict[str, Any]:
|
||||||
"""Run one ``vllm bench serve --omni`` iteration and return parsed metrics.
|
"""Run one ``vllm bench serve --omni`` iteration and return parsed metrics.
|
||||||
|
|
||||||
@@ -677,7 +678,7 @@ def run_benchmark(
|
|||||||
+ args
|
+ args
|
||||||
+ [
|
+ [
|
||||||
"--num-warmups",
|
"--num-warmups",
|
||||||
"2",
|
str(num_warmups),
|
||||||
"--save-result",
|
"--save-result",
|
||||||
"--result-dir",
|
"--result-dir",
|
||||||
os.environ.get("BENCHMARK_DIR", "tests"),
|
os.environ.get("BENCHMARK_DIR", "tests"),
|
||||||
|
|||||||
@@ -279,5 +279,6 @@ def test_performance_benchmark(omni_server, benchmark_params):
|
|||||||
random_input_len=params.get("random_input_len"),
|
random_input_len=params.get("random_input_len"),
|
||||||
random_output_len=params.get("random_output_len"),
|
random_output_len=params.get("random_output_len"),
|
||||||
resource_label=resource_label,
|
resource_label=resource_label,
|
||||||
|
num_warmups=max(2, int(concurrency)),
|
||||||
)
|
)
|
||||||
assert_result(result, params, num_prompt)
|
assert_result(result, params, num_prompt)
|
||||||
Reference in new issue
Block a user