xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-23 19:55:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix triton_reshape_and_cache_flash.py triton import (#25522)

Browse Source 
Signed-off-by: mgoin <mgoin64@gmail.com>
Signed-off-by: yewentao256 <zhyanwentao@126.com>
This commit is contained in:
Michael Goin 2025-09-23 18:26:10 -04:00 committed by yewentao256
parent 715e24e1b3
commit 99eaeebe66
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
vllm/attention/ops/triton_reshape_and_cache_flash.py
View File

@ -2,10 +2,9 @@
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project # SPDX-FileCopyrightText: Copyright contributors to the vLLM project
import torch import torch
import triton
import triton.language as tl
from vllm.platforms import current_platform from vllm.platforms import current_platform
from vllm.triton_utils import tl, triton
@triton.jit @triton.jit