"tests/entrypoints/pooling/embed/test_online.py" did not exist on "316a41ac1de4e6e46933cadb39b9b7af65b01abd"
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
......@@ -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
......
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