Unverified Commit 86ed7702 authored by Wentao Ye's avatar Wentao Ye Committed by GitHub
Browse files

[Feature] Batch Invariant for R1 TP 8 on Blackwell (#27229)


Signed-off-by: default avataryewentao256 <zhyanwentao@126.com>
parent aa1356ec
......@@ -756,10 +756,10 @@ def override_envs_for_invariance():
"FLEX_ATTENTION",
"FLASHINFER",
"FLASH_ATTN_MLA",
"FLASHINFER_MLA",
"TRITON_MLA",
# Not yet supported MLA backends
# "FLASHMLA",
# "FLASHINFER_MLA",
]
if curr_attn_backend not in supported_backends:
warning = (
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment