[CI/Build] Automatically retry flaky tests (#17856)

Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
2025-05-09 23:55:17 +08:00
parent 200da9a517
commit 6e5595ca39
3 changed files with 2 additions and 35 deletions
--- a/tests/kernels/moe/test_moe.py
+++ b/tests/kernels/moe/test_moe.py
@@ -286,6 +286,7 @@ def test_mixtral_moe(dtype: torch.dtype, padding: bool, use_rocm_aiter: bool,
                                   atol=mixtral_moe_tol[dtype])


+@pytest.mark.flaky(reruns=2)
@pytest.mark.parametrize("m", [1, 123, 666])
@pytest.mark.parametrize("n", [128, 1024])
@pytest.mark.parametrize("k", [256, 2048])