xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-13 21:05:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

[MISC] correct copy_blocks src_to_dists param type (#19696)

Browse Source 
Signed-off-by: Andy Xie <andy.xning@gmail.com>
This commit is contained in:
Ning Xie 2025-06-18 08:21:06 +08:00 committed by GitHub
parent dac8cc49f4
commit c53711bd63
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
vllm/attention/ops/ipex_attn.py
View File

@ -1,7 +1,7 @@
# SPDX-License-Identifier: Apache-2.0 # SPDX-License-Identifier: Apache-2.0
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project # SPDX-FileCopyrightText: Copyright contributors to the vLLM project
from typing import Dict, List, Optional, Tuple from typing import List, Optional, Tuple
try: try:
import intel_extension_for_pytorch.llm.modules as ipex_modules import intel_extension_for_pytorch.llm.modules as ipex_modules
@ -120,7 +120,7 @@ class _PagedAttention:
@staticmethod @staticmethod
def copy_blocks( def copy_blocks(
kv_caches: List[torch.Tensor], kv_caches: List[torch.Tensor],
src_to_dists: Dict[int, List[int]], src_to_dists: torch.Tensor,
*args, *args,
) -> None: ) -> None:
key_caches = [kv_cache[0] for kv_cache in kv_caches] key_caches = [kv_cache[0] for kv_cache in kv_caches]