[CI] Bug: Fix triton import issue (#29202)

Signed-off-by: yewentao256 <zhyanwentao@126.com>
This commit is contained in:
Wentao Ye 2025-11-21 20:14:49 -05:00 committed by GitHub
parent 9a3101b2ba
commit 1d34eb11e0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 4 deletions

View File

@ -3,10 +3,9 @@
from collections.abc import Iterable
import torch
import triton
import triton.language as tl
from vllm.attention.backends.utils import PAD_SLOT_ID
from vllm.triton_utils import tl, triton
from vllm.utils.math_utils import cdiv
from vllm.v1.utils import CpuGpuBuffer

View File

@ -7,9 +7,8 @@ import numba
import numba.types as types
import numpy as np
import torch
import triton
import triton.language as tl
from vllm.triton_utils import tl, triton
from vllm.utils import random_uuid
from vllm.utils.math_utils import cdiv
from vllm.v1.utils import CpuGpuBuffer