Unverified Commit 468d23cf authored by Yineng Zhang's avatar Yineng Zhang Committed by GitHub
Browse files

update setup for sgl-kernel (#3214)

parent c38b5fb4
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 ...@@ -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( ...@@ -162,5 +157,3 @@ setup(
}, },
options={"bdist_wheel": {"py_limited_api": "cp39"}}, options={"bdist_wheel": {"py_limited_api": "cp39"}},
) )
_update_wheel_platform_tag()
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment