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
5ce45eb5
Unverified
Commit
5ce45eb5
authored
Sep 16, 2024
by
Simon Mo
Committed by
GitHub
Sep 16, 2024
Browse files
[misc] small qol fixes for release process (#8517)
parent
5478c4b4
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
5 additions
and
1 deletion
+5
-1
Dockerfile
Dockerfile
+2
-0
setup.py
setup.py
+3
-1
No files found.
Dockerfile
View file @
5ce45eb5
...
@@ -82,6 +82,7 @@ ENV BUILDKITE_COMMIT=${buildkite_commit}
...
@@ -82,6 +82,7 @@ ENV BUILDKITE_COMMIT=${buildkite_commit}
ARG
USE_SCCACHE
ARG
USE_SCCACHE
ARG
SCCACHE_BUCKET_NAME=vllm-build-sccache
ARG
SCCACHE_BUCKET_NAME=vllm-build-sccache
ARG
SCCACHE_REGION_NAME=us-west-2
ARG
SCCACHE_REGION_NAME=us-west-2
ARG
SCCACHE_S3_NO_CREDENTIALS=0
# if USE_SCCACHE is set, use sccache to speed up compilation
# if USE_SCCACHE is set, use sccache to speed up compilation
RUN
--mount
=
type
=
cache,target
=
/root/.cache/pip
\
RUN
--mount
=
type
=
cache,target
=
/root/.cache/pip
\
if
[
"
$USE_SCCACHE
"
=
"1"
]
;
then
\
if
[
"
$USE_SCCACHE
"
=
"1"
]
;
then
\
...
@@ -92,6 +93,7 @@ RUN --mount=type=cache,target=/root/.cache/pip \
...
@@ -92,6 +93,7 @@ RUN --mount=type=cache,target=/root/.cache/pip \
&&
rm
-rf
sccache.tar.gz sccache-v0.8.1-x86_64-unknown-linux-musl
\
&&
rm
-rf
sccache.tar.gz sccache-v0.8.1-x86_64-unknown-linux-musl
\
&&
export
SCCACHE_BUCKET
=
${
SCCACHE_BUCKET_NAME
}
\
&&
export
SCCACHE_BUCKET
=
${
SCCACHE_BUCKET_NAME
}
\
&&
export
SCCACHE_REGION
=
${
SCCACHE_REGION_NAME
}
\
&&
export
SCCACHE_REGION
=
${
SCCACHE_REGION_NAME
}
\
&&
export
SCCACHE_S3_NO_CREDENTIALS
=
${
SCCACHE_S3_NO_CREDENTIALS
}
\
&&
export
SCCACHE_IDLE_TIMEOUT
=
0
\
&&
export
SCCACHE_IDLE_TIMEOUT
=
0
\
&&
export
CMAKE_BUILD_TYPE
=
Release
\
&&
export
CMAKE_BUILD_TYPE
=
Release
\
&&
sccache
--show-stats
\
&&
sccache
--show-stats
\
...
...
setup.py
View file @
5ce45eb5
...
@@ -371,7 +371,9 @@ def get_vllm_version() -> str:
...
@@ -371,7 +371,9 @@ def get_vllm_version() -> str:
cuda_version
=
str
(
get_nvcc_cuda_version
())
cuda_version
=
str
(
get_nvcc_cuda_version
())
if
cuda_version
!=
MAIN_CUDA_VERSION
:
if
cuda_version
!=
MAIN_CUDA_VERSION
:
cuda_version_str
=
cuda_version
.
replace
(
"."
,
""
)[:
3
]
cuda_version_str
=
cuda_version
.
replace
(
"."
,
""
)[:
3
]
version
+=
f
"+cu
{
cuda_version_str
}
"
# skip this for source tarball, required for pypi
if
"sdist"
not
in
sys
.
argv
:
version
+=
f
"+cu
{
cuda_version_str
}
"
elif
_is_hip
():
elif
_is_hip
():
# Get the HIP version
# Get the HIP version
hipcc_version
=
get_hipcc_rocm_version
()
hipcc_version
=
get_hipcc_rocm_version
()
...
...
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