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
5b55c0be
Unverified
Commit
5b55c0be
authored
Jan 31, 2026
by
Francesco Fusco
Committed by
GitHub
Jan 31, 2026
Browse files
[Attention] Clarify comment explaining attn_logits +1 dimension (#33427)
Signed-off-by:
Francesco Fusco
<
ffu@zurich.ibm.com
>
parent
15e0bb9c
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
vllm/v1/attention/backends/mla/triton_mla.py
vllm/v1/attention/backends/mla/triton_mla.py
+2
-2
No files found.
vllm/v1/attention/backends/mla/triton_mla.py
View file @
5b55c0be
...
...
@@ -143,8 +143,8 @@ class TritonMLAImpl(MLACommonImpl[MLACommonMetadata]):
B
,
q_num_heads
,
num_kv_splits
,
# NOTE
(lucas) idk why
the +1
i
s he
re but sglang has it so we
#
just mirror that
# NOTE
:
the +1
store
s
t
he
LogSumExp (LSE) that the stage2
#
kernel uses to merge partial attention outputs across splits.
self
.
kv_lora_rank
+
1
,
),
dtype
=
torch
.
float32
,
...
...
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