mirror of https://github.com/vllm-project/vllm.git
43 lines
1.2 KiB
Bash
43 lines
1.2 KiB
Bash
#!/bin/sh
|
|
# This script tests if the nightly torch packages are not overridden by the dependencies
|
|
|
|
set -e
|
|
set -x
|
|
|
|
cd /vllm-workspace/
|
|
|
|
rm -rf .venv
|
|
|
|
uv venv .venv
|
|
|
|
source .venv/bin/activate
|
|
|
|
# check the environment
|
|
uv pip freeze
|
|
|
|
echo ">>> Installing nightly torch packages"
|
|
uv pip install --quiet torch torchvision torchaudio --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu128
|
|
|
|
echo ">>> Capturing torch-related versions before requirements install"
|
|
uv pip freeze | grep -E '^torch|^torchvision|^torchaudio' | sort > before.txt
|
|
echo "Before:"
|
|
cat before.txt
|
|
|
|
echo ">>> Installing requirements/nightly_torch_test.txt"
|
|
uv pip install --quiet -r requirements/nightly_torch_test.txt
|
|
|
|
echo ">>> Capturing torch-related versions after requirements install"
|
|
uv pip freeze | grep -E '^torch|^torchvision|^torchaudio' | sort > after.txt
|
|
echo "After:"
|
|
cat after.txt
|
|
|
|
echo ">>> Comparing versions"
|
|
if diff before.txt after.txt; then
|
|
echo "torch version not overridden."
|
|
else
|
|
echo "torch version overridden by nightly_torch_test.txt, \
|
|
if the dependency is not triggered by the pytroch nightly test,\
|
|
please add the dependency to the list 'white_list' in tools/generate_nightly_torch_test.py"
|
|
exit 1
|
|
fi
|