| 1 | #!/bin/sh |
| 2 | # This script tests if the nightly torch packages are not overridden by the dependencies |
| 3 | |
| 4 | set -e |
| 5 | set -x |
| 6 | |
| 7 | if command -v rocminfo >/dev/null 2>&1; then |
| 8 | echo "Skipping test for ROCm platform" |
| 9 | exit 0 |
| 10 | fi |
| 11 | |
| 12 | cd /vllm-workspace/ |
| 13 | |
| 14 | rm -rf .venv |
| 15 | |
| 16 | uv venv .venv |
| 17 | |
| 18 | source .venv/bin/activate |
| 19 | |
| 20 | # check the environment |
| 21 | uv pip freeze |
| 22 | |
| 23 | echo ">>> Installing nightly torch packages" |
| 24 | uv pip install --quiet torch torchvision torchaudio --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu128 |
| 25 | |
| 26 | echo ">>> Capturing torch-related versions before requirements install" |
| 27 | uv pip freeze | grep -E '^torch|^torchvision|^torchaudio' | sort > before.txt |
| 28 | echo "Before:" |
| 29 | cat before.txt |
| 30 | |
| 31 | echo ">>> Installing requirements/test/nightly-torch.txt" |
| 32 | uv pip install --quiet -r requirements/test/nightly-torch.txt |
| 33 | |
| 34 | echo ">>> Capturing torch-related versions after requirements install" |
| 35 | uv pip freeze | grep -E '^torch|^torchvision|^torchaudio' | sort > after.txt |
| 36 | echo "After:" |
| 37 | cat after.txt |
| 38 | |
| 39 | echo ">>> Comparing versions" |
| 40 | if diff before.txt after.txt; then |
| 41 | echo "torch version not overridden." |
| 42 | else |
| 43 | echo "torch version overridden by test/nightly-torch.txt, \ |
| 44 | if the dependency is not triggered by the pytorch nightly test,\ |
| 45 | please add the dependency to the list 'white_list' in tools/pre_commit/generate_nightly_torch_test.py" |
| 46 | exit 1 |
| 47 | fi |