Skip to content

Commit 9410ce3

Browse files
authored
[https://nvbugs/5575841] [test] Move test_moe.py to serial tests to improve stability + unwaive FP4 MoE torch unit tests (#8422)
Signed-off-by: Dom Brown <3886319+DomBrown@users.noreply.github.com>
1 parent 0dac57f commit 9410ce3

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

tests/unittest/_torch/thop/parallel/test_moe.py renamed to tests/unittest/_torch/thop/serial/test_moe.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -991,7 +991,6 @@ class TestMoeFp4:
991991
the default tactic selection works. This reduces unnecessary test runs for CI
992992
"""
993993

994-
@pytest.mark.skip(reason="https://nvbugs/5575841")
995994
@pytest.mark.parametrize("num_tokens", [1, 1024])
996995
@pytest.mark.parametrize("hidden_size", [1024])
997996
@pytest.mark.parametrize("intermediate_size", [1024, 768, 384, 192])
@@ -1078,7 +1077,6 @@ def test_autotune(self, num_tokens, hidden_size, intermediate_size,
10781077
use_autotune=True,
10791078
use_topk_as_input=False)
10801079

1081-
@pytest.mark.skip(reason="https://nvbugs/5575841")
10821080
@pytest.mark.parametrize("num_tokens", [1, 150])
10831081
@pytest.mark.parametrize("hidden_size", [1024])
10841082
@pytest.mark.parametrize("intermediate_size", [1024])

0 commit comments

Comments
 (0)