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
8cfbe041
Commit
8cfbe041
authored
Nov 27, 2025
by
wujl5
Browse files
fix: 修改attetion这里的import bug
parent
4f11b099
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
1 deletion
+2
-1
vllm/v1/attention/backends/flash_attn.py
vllm/v1/attention/backends/flash_attn.py
+2
-1
No files found.
vllm/v1/attention/backends/flash_attn.py
View file @
8cfbe041
...
...
@@ -566,8 +566,8 @@ class FlashAttentionImpl(AttentionImpl):
layer
.
_v_scale
,
)
else
:
from
lightop
import
reshape_and_cache_cuda
if
envs
.
VLLM_USE_OPT_RESHAPE_AND_CACHE
and
key
.
dtype
==
value
.
dtype
==
"fp16"
:
from
lightop
import
reshape_and_cache_cuda
reshape_and_cache_cuda
(
key
,
value
,
key_cache
,
value_cache
,
...
...
@@ -576,6 +576,7 @@ class FlashAttentionImpl(AttentionImpl):
layer
.
_k_scale
,
layer
.
_v_scale
)
else
:
from
vllm.attention.utils.fa_utils
import
reshape_and_cache_cuda
reshape_and_cache_cuda
(
key
,
value
,
...
...
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