[AMD] Add kimi mi35x nightly test, folder organization and several stability fixes (#17895)
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user