biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

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

Browse Source 
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
2 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
vllm/v1/worker/gpu/block_table.py
View File

@@ -3,10 +3,9 @@
from collections.abc import Iterable from collections.abc import Iterable
import torch import torch
import triton
import triton.language as tl
from vllm.attention.backends.utils import PAD_SLOT_ID 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.utils.math_utils import cdiv
from vllm.v1.utils import CpuGpuBuffer from vllm.v1.utils import CpuGpuBuffer

3
vllm/v1/worker/gpu/input_batch.py
View File

@@ -7,9 +7,8 @@ import numba
import numba.types as types import numba.types as types
import numpy as np import numpy as np
import torch 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 import random_uuid
from vllm.utils.math_utils import cdiv from vllm.utils.math_utils import cdiv
from vllm.v1.utils import CpuGpuBuffer from vllm.v1.utils import CpuGpuBuffer