[AMD] Add kimi mi35x nightly test, folder organization and several stability fixes (#17895)
This commit is contained in:
+1
@@ -129,6 +129,7 @@ class TestNightlyDeepseekV31Performance(unittest.TestCase):
|
||||
other_args=variant_config["other_args"],
|
||||
variant=variant_config["name"],
|
||||
extra_bench_args=["--trust-remote-code"],
|
||||
enable_profile=False, # Disable profiling for AMD tests
|
||||
)
|
||||
results = result_tuple[0]
|
||||
success = result_tuple[1]
|
||||
+1
@@ -115,6 +115,7 @@ class TestNightlyDeepseekV32BasicPerformance(unittest.TestCase):
|
||||
other_args=self.variant_config["other_args"],
|
||||
variant=self.variant_config["name"],
|
||||
extra_bench_args=["--trust-remote-code"],
|
||||
enable_profile=False, # Disable profiling for AMD tests
|
||||
)
|
||||
results = result_tuple[0]
|
||||
success = result_tuple[1]
|
||||
+1
@@ -124,6 +124,7 @@ class TestNightlyDeepseekV32MTPPerformance(unittest.TestCase):
|
||||
other_args=self.variant_config["other_args"],
|
||||
variant=self.variant_config["name"],
|
||||
extra_bench_args=["--trust-remote-code"],
|
||||
enable_profile=False, # Disable profiling for AMD tests
|
||||
)
|
||||
results = result_tuple[0]
|
||||
success = result_tuple[1]
|
||||
+1
@@ -119,6 +119,7 @@ class TestNightlyDeepseekV3Performance(unittest.TestCase):
|
||||
other_args=variant_config["other_args"],
|
||||
variant=variant_config["name"],
|
||||
extra_bench_args=["--trust-remote-code"],
|
||||
enable_profile=False, # Disable profiling for AMD tests
|
||||
)
|
||||
results = result_tuple[0]
|
||||
success = result_tuple[1]
|
||||
+1
@@ -109,6 +109,7 @@ class TestNightlyGrok1FP8Performance(unittest.TestCase):
|
||||
other_args=self.model_config["other_args"],
|
||||
variant=self.model_config["name"],
|
||||
extra_bench_args=["--trust-remote-code"],
|
||||
enable_profile=False, # Disable profiling for AMD tests
|
||||
)
|
||||
results = result_tuple[0]
|
||||
success = result_tuple[1]
|
||||
+1
@@ -119,6 +119,7 @@ class TestNightlyGrok1INT4Performance(unittest.TestCase):
|
||||
other_args=self.model_config["other_args"],
|
||||
variant=self.model_config["name"],
|
||||
extra_bench_args=["--trust-remote-code"],
|
||||
enable_profile=False, # Disable profiling for AMD tests
|
||||
)
|
||||
results = result_tuple[0]
|
||||
success = result_tuple[1]
|
||||
+1
@@ -121,6 +121,7 @@ class TestNightlyGrok2Performance(unittest.TestCase):
|
||||
other_args=self.model_config["other_args"],
|
||||
variant=self.model_config["name"],
|
||||
extra_bench_args=["--trust-remote-code"],
|
||||
enable_profile=False, # Disable profiling for AMD tests
|
||||
)
|
||||
results = result_tuple[0]
|
||||
success = result_tuple[1]
|
||||
+1
@@ -110,6 +110,7 @@ class TestNightlyTextModelsPerfAMD(unittest.TestCase):
|
||||
input_lens=self.input_lens,
|
||||
output_lens=self.output_lens,
|
||||
other_args=other_args,
|
||||
enable_profile=False, # Disable profiling for AMD tests
|
||||
)
|
||||
results = result_tuple[0]
|
||||
success = result_tuple[1]
|
||||
+1
@@ -123,6 +123,7 @@ class TestNightlyVLMsPerfAMD(unittest.TestCase):
|
||||
output_lens=self.output_lens,
|
||||
other_args=other_args,
|
||||
extra_bench_args=extra_bench_args,
|
||||
enable_profile=False, # Disable profiling for AMD tests
|
||||
)
|
||||
results = result_tuple[0]
|
||||
success = result_tuple[1]
|
||||
@@ -152,6 +152,7 @@ class TestDeepseekR1MXFP4PerfMI35x(unittest.TestCase):
|
||||
other_args=variant_config["other_args"],
|
||||
variant=variant_config["name"],
|
||||
extra_bench_args=["--trust-remote-code"],
|
||||
enable_profile=False, # Disable profiling for AMD tests
|
||||
)
|
||||
results = result_tuple[0]
|
||||
success = result_tuple[1]
|
||||
|
||||
@@ -114,6 +114,7 @@ class TestNightlyDeepseekV32BasicPerformance(unittest.TestCase):
|
||||
other_args=self.variant_config["other_args"],
|
||||
variant=self.variant_config["name"],
|
||||
extra_bench_args=["--trust-remote-code"],
|
||||
enable_profile=False, # Disable profiling for AMD tests
|
||||
)
|
||||
results = result_tuple[0]
|
||||
success = result_tuple[1]
|
||||
|
||||
@@ -96,6 +96,7 @@ def _run_benchmark_with_timeout(
|
||||
profile_path_prefix,
|
||||
json_output_file,
|
||||
extra_args=bench_args,
|
||||
enable_profile=False, # Disable profiling for AMD tests
|
||||
)
|
||||
_, cmd_success = runner.run_benchmark_command(command, model_description)
|
||||
if not cmd_success:
|
||||
|
||||
@@ -112,6 +112,7 @@ class TestGrok1INT4PerfMI35x(unittest.TestCase):
|
||||
other_args=self.model_config["other_args"],
|
||||
variant=self.model_config["name"],
|
||||
extra_bench_args=["--trust-remote-code"],
|
||||
enable_profile=False, # Disable profiling for AMD tests
|
||||
)
|
||||
results = result_tuple[0]
|
||||
success = result_tuple[1]
|
||||
|
||||
@@ -112,6 +112,7 @@ class TestGrok2PerfMI35x(unittest.TestCase):
|
||||
other_args=self.model_config["other_args"],
|
||||
variant=self.model_config["name"],
|
||||
extra_bench_args=["--trust-remote-code"],
|
||||
enable_profile=False, # Disable profiling for AMD tests
|
||||
)
|
||||
results = result_tuple[0]
|
||||
success = result_tuple[1]
|
||||
|
||||
Reference in New Issue
Block a user