Sync from v0.13
This commit is contained in:
42
tests/standalone_tests/pytorch_nightly_dependency.sh
Normal file
42
tests/standalone_tests/pytorch_nightly_dependency.sh
Normal file
@@ -0,0 +1,42 @@
|
||||
#!/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/pre_commit/generate_nightly_torch_test.py"
|
||||
exit 1
|
||||
fi
|
||||
Reference in New Issue
Block a user