[CI] Append test variant name to markdown report header in nightly test (#16166)

This commit is contained in:
Baizhou Zhang
2025-12-31 00:09:24 +08:00
committed by GitHub
parent 733a0c1a37
commit f35b5da521
14 changed files with 30 additions and 6 deletions
@@ -72,7 +72,7 @@ class TestNightlyDeepseekV31Performance(unittest.TestCase):
if not success:
failed_variants.append(variant_config["name"])
self.runner.add_report(results)
self.runner.add_report(results, variant=variant_config["name"])
finally:
self.runner.write_final_report()
@@ -113,7 +113,7 @@ class TestNightlyDeepseekV32Performance(unittest.TestCase):
if not success:
failed_variants.append(variant_config["name"])
self.runner.add_report(results)
self.runner.add_report(results, variant=variant_config["name"])
finally:
self.runner.write_final_report()
@@ -48,12 +48,14 @@ class TestDeepseekV31Unified(unittest.TestCase):
DEEPSEEK_V31_MODEL_PATH,
tp_size=8,
extra_args=base_args,
variant="TP8",
),
# Variant: "mtp" - TP=8 + EAGLE speculative decoding
ModelLaunchSettings(
DEEPSEEK_V31_MODEL_PATH,
tp_size=8,
extra_args=base_args + mtp_args,
variant="TP8+MTP",
),
]
@@ -55,24 +55,28 @@ class TestDeepseekV32Unified(unittest.TestCase):
DEEPSEEK_V32_MODEL_PATH,
tp_size=8,
extra_args=BASE_ARGS + DP_ARGS,
variant="DP8",
),
# Variant: "dp+mtp" - DP + EAGLE speculative decoding
ModelLaunchSettings(
DEEPSEEK_V32_MODEL_PATH,
tp_size=8,
extra_args=BASE_ARGS + DP_ARGS + MTP_ARGS,
variant="DP8+MTP",
),
# Variant: "tp" - Pure TP=8 only
ModelLaunchSettings(
DEEPSEEK_V32_MODEL_PATH,
tp_size=8,
extra_args=BASE_ARGS + TP_ARGS,
variant="TP8",
),
# Variant: "tp+mtp" - Pure TP=8 + EAGLE speculative decoding
ModelLaunchSettings(
DEEPSEEK_V32_MODEL_PATH,
tp_size=8,
extra_args=BASE_ARGS + TP_ARGS + MTP_ARGS,
variant="TP8+MTP",
),
]
@@ -34,6 +34,7 @@ class TestGLM46Unified(unittest.TestCase):
GLM_4_6_MODEL_PATH,
tp_size=8,
extra_args=base_args,
variant="TP8",
),
]
@@ -39,11 +39,13 @@ class TestGLM46FP8Unified(unittest.TestCase):
GLM_4_6_FP8_MODEL_PATH,
tp_size=8,
extra_args=base_args,
variant="TP8",
),
ModelLaunchSettings(
GLM_4_6_FP8_MODEL_PATH,
tp_size=8,
extra_args=base_args + mtp_args,
variant="TP8+MTP",
),
]
@@ -35,6 +35,7 @@ class TestKimiK2Unified(unittest.TestCase):
KIMI_K2_THINKING_MODEL_PATH,
tp_size=8,
extra_args=base_args,
variant="TP8",
),
]
@@ -37,6 +37,7 @@ class TestMiniMaxM2Unified(unittest.TestCase):
MINIMAX_M2_MODEL_PATH,
tp_size=8,
extra_args=base_args,
variant="TP8+EP8",
),
]
@@ -63,12 +63,14 @@ class TestMistralLarge3Unified(unittest.TestCase):
MISTRAL_LARGE3_MODEL_PATH,
tp_size=8,
extra_args=base_args,
variant="TP8",
),
# Variant: "eagle" - TP=8 + trtllm_mla + EAGLE with draft model
ModelLaunchSettings(
MISTRAL_LARGE3_MODEL_PATH,
tp_size=8,
extra_args=base_args + eagle_args,
variant="TP8+MTP",
),
]
@@ -34,6 +34,7 @@ class TestQwen3235BUnified(unittest.TestCase):
QWEN3_235B_MODEL_PATH,
tp_size=8,
extra_args=base_args,
variant="TP8",
),
]