test: re-enable FlashInfer per-token NVFP4 coverage (#36985)

This commit is contained in:
Ziang Li
2026-08-30 23:44:09 -07:00
committed by GitHub
parent 3ed3326631
commit 4f997a432a
2 changed files with 0 additions and 14 deletions
@@ -77,14 +77,6 @@ class FlashinferNvFp4OnlineMoeBackendBase:
) )
# Only this class is affected, but the file runs with failfast, so leaving it
# enabled also cuts off the class that sorts after it.
@unittest.skip(
"flashinfer-ai/flashinfer#4486: on SM100/SM103 the TRTLLM_GEN tile-192 BMM "
"path returns non-finite MoE output from FlashInfer 0.6.16.post4 on, so the "
"first real prefill trips the sampler NaN assert and gsm8k scores 0.0. "
"See #34629 for the package bisect."
)
class TestFlashinferTrtllmGenMoeBackendNvFp4Online( class TestFlashinferTrtllmGenMoeBackendNvFp4Online(
FlashinferNvFp4OnlineMoeBackendBase, CustomTestCase FlashinferNvFp4OnlineMoeBackendBase, CustomTestCase
): ):
@@ -273,12 +273,6 @@ class TestFlashinferTrtllmGenMoeBackendBF16Routed(
backend = "flashinfer_trtllm_routed" backend = "flashinfer_trtllm_routed"
@unittest.skip(
"flashinfer-ai/flashinfer#4486: on SM100/SM103 the TRTLLM_GEN tile-192 BMM "
"path returns non-finite MoE output from FlashInfer 0.6.16.post4 on, so the "
"first real prefill trips the sampler NaN assert and gsm8k scores 0.0. "
"See #34629 for the package bisect."
)
class TestFlashinferTrtllmGenMoeBackendNvFp4PerTokenActivationRouted( class TestFlashinferTrtllmGenMoeBackendNvFp4PerTokenActivationRouted(
FlashinferTrtllmGenMoeBackendNVFP4Base, CustomTestCase FlashinferTrtllmGenMoeBackendNVFP4Base, CustomTestCase
): ):