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
989a3fad
Commit
989a3fad
authored
Jan 28, 2026
by
zhuwenwen
Browse files
update self.kv_cache_dtype
parent
a9c755ac
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
vllm/v1/attention/backends/mla/common.py
vllm/v1/attention/backends/mla/common.py
+1
-1
No files found.
vllm/v1/attention/backends/mla/common.py
View file @
989a3fad
...
...
@@ -1308,7 +1308,7 @@ class MLACommonImpl(MLAAttentionImpl[M], Generic[M]):
decode_q_nope
=
decode_q_nope
.
transpose
(
0
,
1
)
# Multiply (N, B, P) x (N, P, L) -> (N, B, L)
# todo: bmm support
decode_ql_nope
=
torch
.
bmm
(
q_scale
,
decode_q_nope
,
self
.
W_UK_T
)
if
torch
.
cuda
.
get_device_properties
(
"cuda"
).
gcnArchName
.
split
(
':'
)[
0
]
==
"gfx938"
and
kv_cache_dtype
_str
==
"fp8_e4m3"
and
envs
.
VLLM_USE_FUSED_CACHE_QUANT_BMM_MLA
else
torch
.
bmm
(
decode_q_nope
,
self
.
W_UK_T
)
decode_ql_nope
=
torch
.
bmm
(
q_scale
,
decode_q_nope
,
self
.
W_UK_T
)
if
torch
.
cuda
.
get_device_properties
(
"cuda"
).
gcnArchName
.
split
(
':'
)[
0
]
==
"gfx938"
and
self
.
kv_cache_dtype
==
"fp8_e4m3"
and
envs
.
VLLM_USE_FUSED_CACHE_QUANT_BMM_MLA
else
torch
.
bmm
(
decode_q_nope
,
self
.
W_UK_T
)
# Convert from (N, B, L) to (B, N, L)
decode_ql_nope
=
decode_ql_nope
.
transpose
(
0
,
1
)
...
...
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