Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
86ed7702
Unverified
Commit
86ed7702
authored
Oct 21, 2025
by
Wentao Ye
Committed by
GitHub
Oct 21, 2025
Browse files
[Feature] Batch Invariant for R1 TP 8 on Blackwell (#27229)
Signed-off-by:
yewentao256
<
zhyanwentao@126.com
>
parent
aa1356ec
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
vllm/model_executor/layers/batch_invariant.py
vllm/model_executor/layers/batch_invariant.py
+1
-1
No files found.
vllm/model_executor/layers/batch_invariant.py
View file @
86ed7702
...
...
@@ -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
=
(
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment