From 4f2954f7240b2b8f14a680e55c3caae163a404a9 Mon Sep 17 00:00:00 2001
From: Michael Goin <mgoin64@gmail.com>
Date: Tue, 23 Sep 2025 18:26:10 -0400
Subject: [PATCH] Fix triton_reshape_and_cache_flash.py triton import (#25522)

Signed-off-by: mgoin <mgoin64@gmail.com>
---
 vllm/attention/ops/triton_reshape_and_cache_flash.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/vllm/attention/ops/triton_reshape_and_cache_flash.py b/vllm/attention/ops/triton_reshape_and_cache_flash.py
index 0b0c706626af3..883052cb46aab 100644
--- a/vllm/attention/ops/triton_reshape_and_cache_flash.py
+++ b/vllm/attention/ops/triton_reshape_and_cache_flash.py
@@ -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