Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
a360ff80
"docs/vscode:/vscode.git/clone" did not exist on "b7df53cd42f3eab007b4f287c151960858e949df"
Unverified
Commit
a360ff80
authored
Jun 01, 2024
by
Daniele
Committed by
GitHub
May 31, 2024
Browse files
[CI/Build] CMakeLists: build all extensions' cmake targets at the same time (#5034)
parent
1197e021
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
10 additions
and
7 deletions
+10
-7
setup.py
setup.py
+10
-7
No files found.
setup.py
View file @
a360ff80
...
@@ -187,19 +187,22 @@ class cmake_build_ext(build_ext):
...
@@ -187,19 +187,22 @@ class cmake_build_ext(build_ext):
if
not
os
.
path
.
exists
(
self
.
build_temp
):
if
not
os
.
path
.
exists
(
self
.
build_temp
):
os
.
makedirs
(
self
.
build_temp
)
os
.
makedirs
(
self
.
build_temp
)
targets
=
[]
# Build all the extensions
# Build all the extensions
for
ext
in
self
.
extensions
:
for
ext
in
self
.
extensions
:
self
.
configure
(
ext
)
self
.
configure
(
ext
)
targets
.
append
(
remove_prefix
(
ext
.
name
,
"vllm."
))
ext_target_name
=
remove_prefix
(
ext
.
name
,
"vllm."
)
num_jobs
,
_
=
self
.
compute_num_jobs
()
num_jobs
,
_
=
self
.
compute_num_jobs
()
build_args
=
[
build_args
=
[
'--build'
,
'.'
,
'--target'
,
ext_target_name
,
'-j'
,
"--build"
,
str
(
num_jobs
)
"."
,
f
"-j=
{
num_jobs
}
"
,
*
[
f
"--target=
{
name
}
"
for
name
in
targets
],
]
]
subprocess
.
check_call
([
'
cmake
'
,
*
build_args
],
cwd
=
self
.
build_temp
)
subprocess
.
check_call
([
"
cmake
"
,
*
build_args
],
cwd
=
self
.
build_temp
)
def
_is_cuda
()
->
bool
:
def
_is_cuda
()
->
bool
:
...
...
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