Skip to content

Commit 52adf89

Browse files
adityagoel14Alvant
authored andcommitted
[CI/Build] (2/2) Switching AMD CI to store images in Docker Hub (vllm-project#6350)
Signed-off-by: Alvant <alvasian@yandex.ru>
1 parent a0edf3e commit 52adf89

File tree

1 file changed

+4
-9
lines changed

1 file changed

+4
-9
lines changed

.buildkite/run-amd-test.sh

Lines changed: 4 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -45,15 +45,10 @@ while true; do
4545
fi
4646
done
4747

48-
echo "--- Building container"
49-
sha=$(git rev-parse --short HEAD)
50-
image_name=rocm_${sha}
51-
container_name=rocm_${sha}_$(tr -dc A-Za-z0-9 < /dev/urandom | head -c 10; echo)
52-
docker build \
53-
-t ${image_name} \
54-
-f Dockerfile.rocm \
55-
--progress plain \
56-
.
48+
echo "--- Pulling container"
49+
image_name="rocmshared/vllm-ci:${BUILDKITE_COMMIT}"
50+
container_name="rocm_${BUILDKITE_COMMIT}_$(tr -dc A-Za-z0-9 < /dev/urandom | head -c 10; echo)"
51+
docker pull ${image_name}
5752

5853
remove_docker_container() {
5954
docker rm -f ${container_name} || docker image rm -f ${image_name} || true

0 commit comments

Comments
 (0)