[CI] Restore clean GPU state before OAI Triton MoE tests (#55453)
Signed-off-by: khluu <khluu000@gmail.com> Co-authored-by: Codex <noreply@openai.com>
Kevin H. Luu committed
Sep 8, 2026 at 01:15 UTC
96eccb8f49aff58aa2a11b431bf8331f4b368606
1 file changed
+1
tests/kernels/moe/conftest.py
+1
@@ -10,6 +10,7 @@ NEEDS_CLEAN_ENTRY = frozenset(
10
# which they used to get only as a side effect of the preceding test
11
# cleaning up after itself.
12
"test_moe_layer.py",
13
+ "test_modular_oai_triton_moe.py",
14
"test_ocp_mx_moe.py",
15
"test_zero_expert_moe.py",
16
# Execute no test on ROCm, where this was measured, so nothing is known