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
02809af1
Unverified
Commit
02809af1
authored
Jan 06, 2026
by
Isotr0py
Committed by
GitHub
Jan 06, 2026
Browse files
[Bugfix]: Fix cross attention backend selection for Turing GPU (#31806)
Signed-off-by:
Isotr0py
<
mozf@mail2.sysu.edu.cn
>
parent
cbd4690a
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
9 additions
and
5 deletions
+9
-5
vllm/attention/layers/cross_attention.py
vllm/attention/layers/cross_attention.py
+9
-5
No files found.
vllm/attention/layers/cross_attention.py
View file @
02809af1
...
@@ -149,16 +149,20 @@ class CrossAttention(Attention):
...
@@ -149,16 +149,20 @@ class CrossAttention(Attention):
kv_cache_dtype
=
"auto"
kv_cache_dtype
=
"auto"
block_size
=
16
block_size
=
16
underlying_attn_backend
=
get_attn_backend
(
head_size
,
dtype
,
kv_cache_dtype
,
block_size
)
attn_backend
=
create_cross_attention_backend
(
underlying_attn_backend
)
if
attn_type
is
not
None
:
if
attn_type
is
not
None
:
assert
attn_type
==
AttentionType
.
ENCODER_DECODER
,
(
assert
attn_type
==
AttentionType
.
ENCODER_DECODER
,
(
"CrossAttention only supports AttentionType.ENCODER_DECODER"
"CrossAttention only supports AttentionType.ENCODER_DECODER"
)
)
underlying_attn_backend
=
get_attn_backend
(
head_size
,
dtype
,
kv_cache_dtype
,
block_size
,
attn_type
=
AttentionType
.
ENCODER_DECODER
,
)
attn_backend
=
create_cross_attention_backend
(
underlying_attn_backend
)
super
().
__init__
(
super
().
__init__
(
num_heads
=
num_heads
,
num_heads
=
num_heads
,
head_size
=
head_size
,
head_size
=
head_size
,
...
...
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