From 96eccb8f49aff58aa2a11b431bf8331f4b368606 Mon Sep 17 00:00:00 2001
From: "Kevin H. Luu" <khluu000@gmail.com>
Date: Tue, 8 Sep 2026 01:15:21 -0700
Subject: [PATCH] [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>
---
 tests/kernels/moe/conftest.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/kernels/moe/conftest.py b/tests/kernels/moe/conftest.py
index 0a210edc4a..1405991c9c 100644
--- a/tests/kernels/moe/conftest.py
+++ b/tests/kernels/moe/conftest.py
@@ -10,6 +10,7 @@ NEEDS_CLEAN_ENTRY = frozenset(
         # which they used to get only as a side effect of the preceding test
         # cleaning up after itself.
         "test_moe_layer.py",
+        "test_modular_oai_triton_moe.py",
         "test_ocp_mx_moe.py",
         "test_zero_expert_moe.py",
         # Execute no test on ROCm, where this was measured, so nothing is known