youngkingdom/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 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
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
import torch
import triton
import triton.language as tl
from vllm.platforms import current_platform
from vllm.triton_utils import tl, triton
@triton.jit