[Feature] Enable TRITON_ATTN for Batch Invariance (#33688)

Signed-off-by: frankwang28 <frank.wbb@hotmail.com>
2026-02-03 21:27:34 -08:00
parent 5e1e0a0fbd
commit 45f8fd6f97
4 changed files with 13 additions and 4 deletions
--- a/tests/v1/determinism/utils.py
+++ b/tests/v1/determinism/utils.py
@@ -18,6 +18,7 @@ skip_unsupported = pytest.mark.skipif(

 BACKENDS: list[str] = [
    "FLASH_ATTN",
+    "TRITON_ATTN",
    "TRITON_MLA",
 ]