mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-24 18:45:02 +08:00
[MISC] fix import violations for re and triton modules (#26654)
Signed-off-by: Sungjae Lee <33976427+llsj14@users.noreply.github.com> Co-authored-by: Mengqing Cao <cmq0113@163.com>
This commit is contained in:
parent
ed344f4116
commit
00417f4e44
@ -14,10 +14,10 @@
|
|||||||
from typing import Literal
|
from typing import Literal
|
||||||
|
|
||||||
import torch
|
import torch
|
||||||
import triton
|
|
||||||
import triton.language as tl
|
|
||||||
from torch.library import wrap_triton
|
from torch.library import wrap_triton
|
||||||
|
|
||||||
|
from vllm.triton_utils import tl, triton
|
||||||
|
|
||||||
|
|
||||||
@triton.jit
|
@triton.jit
|
||||||
def triton_scale_swizzle(
|
def triton_scale_swizzle(
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user