Render experiments (opendatahub-io#159)

* testing ignore it * fixing the docker * adding * commenting push plugin part * removing plugin part * adding the kfd tests * generci docker * adding * moving back older docker * Trigger Build --------- Co-authored-by: omkarkakarparthi <okakarpa> Co-authored-by: Alexei-V-Ivanov-AMD <[email protected]>
prarit · Aug 28, 2024 · 07b6b14 · 07b6b14
1 parent 68db66a
commit 07b6b14
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 6 deletions.
diff --git a/.buildkite/run-amd-test.sh b/.buildkite/run-amd-test.sh
@@ -57,9 +57,9 @@ remove_docker_container() {
 trap remove_docker_container EXIT
 
 echo "--- Running container"
-
+echo "Render devices: $BUILDKITE_AGENT_META_DATA_RENDER_DEVICES"
 docker run \
-        --device /dev/kfd --device /dev/dri \
+        --device /dev/kfd $BUILDKITE_AGENT_META_DATA_RENDER_DEVICES \
         --network host \
         --rm \
         -e HF_TOKEN \

diff --git a/.buildkite/test-template.j2 b/.buildkite/test-template.j2
@@ -1,5 +1,5 @@
 {% set docker_image = "public.ecr.aws/q9t5s3a7/vllm-ci-test-repo:$BUILDKITE_COMMIT" %}
-{% set docker_image_amd = "rocmshared/vllm-ci:$BUILDKITE_COMMIT" %}
+{% set docker_image_amd = "rocm/vllm-ci:$BUILDKITE_COMMIT" %}
 {% set default_working_dir = "vllm/tests" %}
 {% set hf_home = "/root/.cache/huggingface" %}
 
@@ -9,9 +9,6 @@ steps:
     commands:
       - "docker build --build-arg max_jobs=16 --tag {{ docker_image_amd }} -f Dockerfile.rocm --progress plain ."
       - "docker push {{ docker_image_amd }}"
-    plugins:
-      - docker-login#v3.0.0:
-          username: rocm
     key: "amd-build"
     env:
       DOCKER_BUILDKIT: "1"