Skip to content

GitLab

Unverified Commit 4f2954f7 authored by Michael Goin's avatar Michael Goin Committed by GitHub
Browse files

Fix triton_reshape_and_cache_flash.py triton import (#25522) 


Signed-off-by: default avatarmgoin <mgoin64@gmail.com>
parent eca7be90
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
vllm/attention/ops/triton_reshape_and_cache_flash.py
View file @ 4f2954f7
...@@ -2,10 +2,9 @@ ...@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment