[AMD] Add kimi mi35x nightly test, folder organization and several stability fixes (#17895)

This commit is contained in:
Michael
2026-02-04 12:03:57 -08:00
committed by GitHub
parent 36a3e78af9
commit 6fd878b41d
34 changed files with 184 additions and 14 deletions

View File

@@ -94,6 +94,7 @@ class NightlyBenchmarkRunner:
json_output_file: str,
extra_args: Optional[List[str]] = None,
server_args: Optional[List[str]] = None,
enable_profile: bool = True,
) -> List[str]:
"""Build the benchmark command with all required arguments.
@@ -106,6 +107,7 @@ class NightlyBenchmarkRunner:
json_output_file: Path to JSON output file
extra_args: Optional extra arguments to append to command
server_args: Optional server launch arguments to record in metrics
enable_profile: Whether to enable profiling (default True for NVIDIA)
Returns:
List of command arguments ready for subprocess.run()
@@ -125,15 +127,22 @@ class NightlyBenchmarkRunner:
"--output-len",
*[str(x) for x in output_lens],
"--show-report",
"--profile",
"--profile-by-stage",
"--profile-output-dir",
profile_path_prefix,
f"--pydantic-result-filename={json_output_file}",
"--no-append-to-github-summary",
"--trust-remote-code",
]
# Add profiling flags only if enabled (disabled for AMD tests)
if enable_profile and profile_path_prefix:
command.extend(
[
"--profile",
"--profile-by-stage",
"--profile-output-dir",
profile_path_prefix,
]
)
if extra_args:
command.extend(extra_args)
@@ -218,6 +227,7 @@ class NightlyBenchmarkRunner:
other_args: Optional[List[str]] = None,
variant: str = "",
extra_bench_args: Optional[List[str]] = None,
enable_profile: bool = True,
) -> Tuple[List[BenchmarkResult], bool, Optional[float]]:
"""Run a complete benchmark for a single model with server management.
@@ -236,6 +246,7 @@ class NightlyBenchmarkRunner:
other_args: Arguments to pass to server launch
variant: Optional variant suffix (e.g., "basic", "mtp")
extra_bench_args: Extra arguments for the benchmark command
enable_profile: Whether to enable profiling (default True for NVIDIA)
Returns:
Tuple of (list of BenchmarkResult objects, success_bool, avg_spec_accept_length or None)
@@ -273,6 +284,7 @@ class NightlyBenchmarkRunner:
json_output_file,
extra_args=bench_args,
server_args=other_args,
enable_profile=enable_profile,
)
result, cmd_success = self.run_benchmark_command(command, model_description)