EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

update setup for sgl-kernel (#3214)

Browse Source
This commit is contained in:
Yineng Zhang
2025-01-30 19:47:50 +08:00
committed by GitHub
parent c38b5fb4f4
commit 468d23cff9
1 changed files with 3 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
sgl-kernel/setup.py
View File

@@ -1,5 +1,6 @@
import multiprocessing import multiprocessing
import os import os
import sys
from pathlib import Path from pathlib import Path
import torch import torch
@@ -9,14 +10,8 @@ from torch.utils.cpp_extension import BuildExtension, CUDAExtension
root = Path(__file__).parent.resolve() root = Path(__file__).parent.resolve()
def _update_wheel_platform_tag(): if "bdist_wheel" in sys.argv and "--plat-name" not in sys.argv:
wheel_dir = Path("dist") sys.argv.extend(["--plat-name", "manylinux2014_x86_64"])
if wheel_dir.exists() and wheel_dir.is_dir():
old_wheel = next(wheel_dir.glob("*.whl"))
new_wheel = wheel_dir / old_wheel.name.replace(
"linux_x86_64", "manylinux2014_x86_64"
)
old_wheel.rename(new_wheel)
def _get_cuda_version(): def _get_cuda_version():
@@ -162,5 +157,3 @@ setup(
}, },
options={"bdist_wheel": {"py_limited_api": "cp39"}}, options={"bdist_wheel": {"py_limited_api": "cp39"}},
) )
_update_wheel_platform_tag()