Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
zhaoyu6
sglang
Commits
468d23cf
"vscode:/vscode.git/clone" did not exist on "f6695521862e525177f04f44c7f706dda27b4bf3"
Unverified
Commit
468d23cf
authored
Jan 30, 2025
by
Yineng Zhang
Committed by
GitHub
Jan 30, 2025
Browse files
update setup for sgl-kernel (#3214)
parent
c38b5fb4
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
10 deletions
+3
-10
sgl-kernel/setup.py
sgl-kernel/setup.py
+3
-10
No files found.
sgl-kernel/setup.py
View file @
468d23cf
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
()
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment