Skip to content

Commit fbad85e

Browse files
committed
[https://nvbugs/5434352][ci] Waive test_moe.py::TestMoeFp4::test_autotune[RoutingDSv3-384-1024-1024] failed with timeout.
Signed-off-by: Simeng Liu <simengl@nvidia.com>
1 parent 04630f6 commit fbad85e

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/unittest/_torch/thop/parallel/test_moe.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1154,6 +1154,9 @@ def run_moe_fp4_test(self, num_tokens: int, hidden_size: int,
11541154
if padding >= 256:
11551155
pytest.skip("Routing kernel requires that padding be less than 256")
11561156

1157+
if intermediate_size == 384:
1158+
pytest.skip("https://nvbugs/5434352")
1159+
11571160
assert top_k <= num_experts
11581161
assert top_k <= 8
11591162
assert num_experts % 4 == 0

0 commit comments

Comments
 (0)