[AMD] Replace MI325 with MI300 CI Runners (#31409)
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user