[NVIDIA] Fix broken fp8 MoE of deepseek v3 (#13264)

Co-authored-by: Baizhou Zhang <sobereddiezhang@gmail.com>
Co-authored-by: Kangyan-Zhou <zky314343421@gmail.com>
This commit is contained in:
Kaixi Hou
2025-11-17 16:13:28 -08:00
committed by GitHub
co-authored by Baizhou Zhang Kangyan-Zhou
parent aac07bf7fd
commit e389f91dec
5 changed files with 117 additions and 4 deletions
+4 -1
View File
@@ -176,6 +176,7 @@ suites = {
# TODO: Add it back after the bug is fixed
# TestFile("test_eagle_infer_beta_dp_attention.py", 200),
],
"per-commit-8-gpu-b200": [],
"per-commit-4-gpu-gb200": [
TestFile("test_cutedsl_moe.py", 300),
TestFile("test_deepseek_v3_fp4_4gpu.py", 1800),
@@ -215,7 +216,9 @@ suites = {
TestFile("test_deepseek_v3_fp4_cutlass_moe.py", 900),
TestFile("test_fp4_moe.py", 300),
],
"nightly-8-gpu-b200": [],
"nightly-8-gpu-b200": [
TestFile("test_deepseek_r1_fp8_trtllm_backend.py", 3600),
],
"nightly-4-gpu": [
TestFile("test_qwen3_next_deterministic.py", 200),
],