[AMD] Replace MI325 with MI300 CI Runners (#31409)

This commit is contained in:
Bingxu Chen
2026-07-29 08:18:07 -07:00
committed by GitHub
parent e1f2f9d1fa
commit bfc450248e
9 changed files with 200 additions and 174 deletions
@@ -167,7 +167,7 @@ class TestMiniMaxM27EvalAMD(unittest.TestCase):
def setUpClass(cls):
cls.models = MINIMAX_M27_MODELS
cls.base_url = DEFAULT_URL_FOR_TEST
cls.num_questions = int(os.environ.get("GSM8K_NUM_QUESTIONS", "200"))
cls.num_questions = int(os.environ.get("GSM8K_NUM_QUESTIONS", "1319"))
def test_minimax_m27_accuracy(self):
"""Test MiniMax-M2.7 with GSM8K completion benchmark."""
@@ -117,7 +117,7 @@ class TestBenchServing1GPUPart1(CustomTestCase):
f"Output throughput: {res['output_throughput']:.2f} token/s\n"
)
if is_in_amd_ci():
self.assertGreater(res["output_throughput"], 3500)
self.assertGreater(res["output_throughput"], 2700)
else:
self.assertGreater(res["output_throughput"], 3700)
@@ -150,7 +150,10 @@ class TestBenchServing1GPUPart1(CustomTestCase):
f"median_e2e_latency_ms: {res['median_e2e_latency_ms']:.2f} ms\n"
f"median_ttft_ms: {res['median_ttft_ms']:.2f} ms\n"
)
self.assertLess(res["median_e2e_latency_ms"], 2400)
if is_in_amd_ci():
self.assertLess(res["median_e2e_latency_ms"], 3320)
else:
self.assertLess(res["median_e2e_latency_ms"], 2400)
# relax for mi300x (LoRA TTFT ~2x slower than mi325)
if is_in_amd_ci():
self.assertLess(res["median_ttft_ms"], 100)
@@ -166,7 +169,10 @@ class TestBenchServing1GPUPart1(CustomTestCase):
f"median_e2e_latency_ms: {res['median_e2e_latency_ms']:.2f} ms\n"
f"median_ttft_ms: {res['median_ttft_ms']:.2f} ms\n"
)
self.assertLess(res["median_e2e_latency_ms"], 4000)
if is_in_amd_ci():
self.assertLess(res["median_e2e_latency_ms"], 6000)
else:
self.assertLess(res["median_e2e_latency_ms"], 4000)
# relax for mi300x (LoRA TTFT ~2x slower than mi325)
if is_in_amd_ci():
self.assertLess(res["median_ttft_ms"], 130)