From 1e5e5d757e4f5b52d20323a60c54a7a4f7d45be4 Mon Sep 17 00:00:00 2001
From: Tyler Michael Smith <tyler@neuralmagic.com>
Date: Sat, 27 Sep 2025 10:13:11 -0400
Subject: [PATCH] [Bugfix] Fix triton import precommit failure (#25803)

Signed-off-by: Tyler Michael Smith <tlrmchlsmth@gmail.com>
Signed-off-by: yewentao256 <zhyanwentao@126.com>
---
 vllm/model_executor/layers/batch_invariant.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/vllm/model_executor/layers/batch_invariant.py b/vllm/model_executor/layers/batch_invariant.py
index ae2c842af698b..c025d509d8626 100644
--- a/vllm/model_executor/layers/batch_invariant.py
+++ b/vllm/model_executor/layers/batch_invariant.py
@@ -7,8 +7,8 @@ from collections.abc import Callable
 from typing import Any, Union
 
 import torch
-import triton
-import triton.language as tl
+
+from vllm.triton_utils import tl, triton
 
 
 def _matmul_launch_metadata(grid: Callable[..., Any], kernel: Any,