diff --git a/.buildkite/run-amd-test.sh b/.buildkite/run-amd-test.sh index 706737e70d0..6f023dc4727 100755 --- a/.buildkite/run-amd-test.sh +++ b/.buildkite/run-amd-test.sh @@ -152,7 +152,7 @@ else echo "Render devices: $BUILDKITE_AGENT_META_DATA_RENDER_DEVICES" docker run \ --device /dev/kfd $BUILDKITE_AGENT_META_DATA_RENDER_DEVICES \ - --network host \ + --network=host \ --shm-size=16gb \ --rm \ -e HIP_VISIBLE_DEVICES=0 \ diff --git a/.buildkite/test-template.j2 b/.buildkite/test-template.j2 index 3ecbd64a973..d7725fcb33d 100644 --- a/.buildkite/test-template.j2 +++ b/.buildkite/test-template.j2 @@ -34,7 +34,7 @@ steps: {% elif step.amd_gpus and step.amd_gpus==2%} queue: amd_gpu {% else%} - queue: amd_gpu + queue: amd_gpu_1_oci {% endif%} commands: - bash .buildkite/run-amd-test.sh "cd {{ (step.working_dir or default_working_dir) | safe }} ; {{ step.command or (step.commands | join(" && ")) | safe }}" diff --git a/Dockerfile.rocm b/Dockerfile.rocm index d2c2d3b1467..7188f7a7341 100644 --- a/Dockerfile.rocm +++ b/Dockerfile.rocm @@ -12,7 +12,7 @@ ENV PYTORCH_ROCM_ARCH=${ARG_PYTORCH_ROCM_ARCH:-${PYTORCH_ROCM_ARCH}} # Install some basic utilities RUN apt-get update -q -y && apt-get install -q -y \ - sqlite3 libsqlite3-dev libfmt-dev libmsgpack-dev libsuitesparse-dev + sqlite3 libsqlite3-dev libfmt-dev libmsgpack-dev libsuitesparse-dev apt-transport-https ca-certificates wget curl # Remove sccache RUN python3 -m pip install --upgrade pip && pip install setuptools_scm RUN apt-get purge -y sccache; python3 -m pip uninstall -y sccache; rm -f "$(which sccache)"