Skip to content

Commit b3e41df

Browse files
khluuAlvant
authored andcommitted
[ci] Use different sccache bucket for CUDA 11.8 wheel build (vllm-project#6656)
Signed-off-by: kevin <kevin@anyscale.com> Signed-off-by: Alvant <alvasian@yandex.ru>
1 parent 3ec60f0 commit b3e41df

File tree

2 files changed

+8
-2
lines changed

2 files changed

+8
-2
lines changed

.buildkite/release-pipeline.yaml

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,13 +3,15 @@ steps:
33
agents:
44
queue: cpu_queue
55
commands:
6-
- "DOCKER_BUILDKIT=1 docker build --build-arg max_jobs=16 --build-arg USE_SCCACHE=1 --build-arg CUDA_VERSION={{matrix.cuda_version}} --tag vllm-ci:build-image --target build --progress plain ."
6+
- "DOCKER_BUILDKIT=1 docker build --build-arg max_jobs=16 --build-arg buildkite_commit=$BUILDKITE_COMMIT --build-arg USE_SCCACHE=1 --build-arg CUDA_VERSION={{matrix.cuda_version}} --tag vllm-ci:build-image --target build --progress plain ."
77
- "mkdir artifacts"
88
- "docker run --rm -v $(pwd)/artifacts:/artifacts_host vllm-ci:build-image bash -c 'cp -r dist /artifacts_host && chmod -R a+rw /artifacts_host'"
99
# rename the files to change linux -> manylinux1
1010
- "for f in artifacts/dist/*.whl; do mv -- \"$$f\" \"$${f/linux/manylinux1}\"; done"
1111
- "aws s3 cp --recursive artifacts/dist s3://vllm-wheels/$BUILDKITE_COMMIT/"
1212
- "aws s3 cp --recursive artifacts/dist s3://vllm-wheels/nightly/"
13+
env:
14+
DOCKER_BUILDKIT: "1"
1315
matrix:
1416
setup:
1517
cuda_version:

Dockerfile

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,11 @@ RUN --mount=type=cache,target=/root/.cache/pip \
103103
&& tar -xzf sccache.tar.gz \
104104
&& sudo mv sccache-v0.8.1-x86_64-unknown-linux-musl/sccache /usr/bin/sccache \
105105
&& rm -rf sccache.tar.gz sccache-v0.8.1-x86_64-unknown-linux-musl \
106-
&& export SCCACHE_BUCKET=vllm-build-sccache \
106+
&& if [ "$CUDA_VERSION" = "11.8.0" ]; then \
107+
export SCCACHE_BUCKET=vllm-build-sccache-2; \
108+
else \
109+
export SCCACHE_BUCKET=vllm-build-sccache; \
110+
fi \
107111
&& export SCCACHE_REGION=us-west-2 \
108112
&& export CMAKE_BUILD_TYPE=Release \
109113
&& sccache --show-stats \

0 commit comments

Comments
 (0)