mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-06-06 03:29:09 +08:00
[Bugfix] Fix triton import precommit failure (#25803)
Signed-off-by: Tyler Michael Smith <tlrmchlsmth@gmail.com> Signed-off-by: yewentao256 <zhyanwentao@126.com>
This commit is contained in:
parent
c7ae7edb33
commit
1e5e5d757e
@ -7,8 +7,8 @@ from collections.abc import Callable
|
|||||||
from typing import Any, Union
|
from typing import Any, Union
|
||||||
|
|
||||||
import torch
|
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,
|
def _matmul_launch_metadata(grid: Callable[..., Any], kernel: Any,
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user