diff --git a/.buildkite/run-amd-test.sh b/.buildkite/run-amd-test.sh index ccc2f090565e4..5548071390aff 100644 --- a/.buildkite/run-amd-test.sh +++ b/.buildkite/run-amd-test.sh @@ -75,6 +75,7 @@ docker run \ --network host \ --shm-size=16gb \ --rm \ + -e HIP_VISIBLE_DEVICES=0 \ -e HF_TOKEN \ -v ${HF_CACHE}:${HF_MOUNT} \ -e HF_HOME=${HF_MOUNT} \ diff --git a/.buildkite/test-pipeline.yaml b/.buildkite/test-pipeline.yaml index e406938647479..9f449ff650b90 100644 --- a/.buildkite/test-pipeline.yaml +++ b/.buildkite/test-pipeline.yaml @@ -233,12 +233,13 @@ steps: parallelism: 4 - label: Tensorizer Test # 11min + mirror_hardwares: [amd] soft_fail: true source_file_dependencies: - vllm/model_executor/model_loader - tests/tensorizer_loader commands: - - apt-get install -y curl libsodium23 + - apt-get update && apt-get install -y curl libsodium23 - export VLLM_WORKER_MULTIPROC_METHOD=spawn - pytest -v -s tensorizer_loader diff --git a/requirements-rocm.txt b/requirements-rocm.txt index cc955e279a845..121123611d2da 100644 --- a/requirements-rocm.txt +++ b/requirements-rocm.txt @@ -8,3 +8,4 @@ botocore ray >= 2.10.0 peft pytest-asyncio +tensorizer>=2.9.0 \ No newline at end of file