diff --git a/.buildkite/test-amd.yaml b/.buildkite/test-amd.yaml index ba6edb92f..ffdf4b83c 100644 --- a/.buildkite/test-amd.yaml +++ b/.buildkite/test-amd.yaml @@ -67,7 +67,7 @@ steps: timeout_in_minutes: 30 mirror_hardwares: [amdexperimental, amdproduction, amdtentative] agent_pool: mi325_1 - grade: Blocking + # grade: Blocking source_file_dependencies: - vllm/ - tests/test_inputs.py diff --git a/.buildkite/test_areas/distributed.yaml b/.buildkite/test_areas/distributed.yaml index 9ded5ffda..f15e5018b 100644 --- a/.buildkite/test_areas/distributed.yaml +++ b/.buildkite/test_areas/distributed.yaml @@ -209,7 +209,7 @@ steps: - uv pip install --system -r /vllm-workspace/requirements/kv_connectors.txt - CROSS_LAYERS_BLOCKS=True bash v1/kv_connector/nixl_integration/config_sweep_accuracy_test.sh -- label: Pipeline + Context Parallelism (4 GPUs)) +- label: Pipeline + Context Parallelism (4 GPUs) timeout_in_minutes: 60 working_dir: "/vllm-workspace/tests" num_devices: 4 diff --git a/requirements/rocm-test.txt b/requirements/rocm-test.txt index 1983392a1..5cfda430b 100644 --- a/requirements/rocm-test.txt +++ b/requirements/rocm-test.txt @@ -104,3 +104,5 @@ segmentation-models-pytorch==0.5.0 imagehash==4.3.2 # Required for bitsandbytes quantization test bitsandbytes==0.49.2 +# Examples (tensorizer) tests +tensorizer==2.10.1