[CI] Add scripts for bechmark collection and email distribution. (#1307)

Signed-off-by: Alicia <115451386+congw729@users.noreply.github.com>
This commit is contained in:
Alicia authored and GitHub committed 2026-03-02 23:00:33 -08:00
1 parent 3b2b5012eb
commit e7f2c88701
5 files changed
+704 -13

No files matched your search

+9 -8
View File
@@ -1,9 +1,5 @@
import os
os.environ["VLLM_WORKER_MULTIPROC_METHOD"] = "spawn"
os.environ["VLLM_TEST_CLEAN_GPU_MEMORY"] = "0"
import json
import os
import subprocess
import threading
from datetime import datetime
@@ -14,6 +10,9 @@ import pytest
from tests.conftest import OmniServer, modify_stage_config
os.environ["VLLM_WORKER_MULTIPROC_METHOD"] = "spawn"
os.environ["VLLM_TEST_CLEAN_GPU_MEMORY"] = "0"
def load_configs(config_path: str) -> list[dict[str, Any]]:
try:
@@ -123,6 +122,8 @@ def run_benchmark(args: list, test_name: str, flow, dataset_name: str, num_promp
"--endpoint",
"/v1/chat/completions",
"--save-result",
"--result-dir",
os.environ.get("BENCHMARK_DIR", "tests"),
"--result-filename",
result_filename,
]
@@ -137,9 +138,9 @@ def run_benchmark(args: list, test_name: str, flow, dataset_name: str, num_promp
for line in iter(process.stderr.readline, ""):
print(line, end=" ")
if "--result-dir" in args:
index = args.index("--result-dir")
result_dir = args[index + 1]
if "--result-dir" in command:
index = command.index("--result-dir")
result_dir = command[index + 1]
else:
result_dir = "./"