Skip to content

Commit 4838d53

Browse files
committed
Some more tweaks
Signed-off-by: Huy Do <huydhn@gmail.com>
1 parent 456985c commit 4838d53

File tree

2 files changed

+14
-1
lines changed

2 files changed

+14
-1
lines changed

docs/contributing/ci/update_pytorch_version.md

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,17 @@ via `UV_INDEX_STRATEGY` env variable or via `--index-strategy unsafe-best-match`
3939
If failures are found in the pull request, raise them as issues on vLLM and
4040
cc the PyTorch release team to initiate discussion on how to address them.
4141

42+
### Update some tests to use PyTorch RC
43+
44+
#### Python-only installation test
45+
46+
Update tests/standalone_tests/python_only_compile.sh to
47+
48+
```
49+
VLLM_TEST_USE_PRECOMPILED_NIGHTLY_WHEEL=1 VLLM_USE_PRECOMPILED=1 pip3 install -vvv -e . \
50+
--extra-index-url https://download.pytorch.org/whl/test/cu128
51+
```
52+
4253
## Update CUDA version
4354

4455
The PyTorch release matrix includes both stable and experimental [CUDA versions](https://github.com/pytorch/pytorch/blob/main/RELEASE.md#release-compatibility-matrix). Due to limitations, only the latest stable CUDA version (for example,

tests/standalone_tests/python_only_compile.sh

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,9 @@ apt autoremove -y
1818

1919
echo 'import os; os.system("touch /tmp/changed.file")' >> vllm/__init__.py
2020

21-
VLLM_TEST_USE_PRECOMPILED_NIGHTLY_WHEEL=1 VLLM_USE_PRECOMPILED=1 pip3 install -vvv -e .
21+
# TESTING, TO BE REMOVED
22+
VLLM_TEST_USE_PRECOMPILED_NIGHTLY_WHEEL=1 VLLM_USE_PRECOMPILED=1 pip3 install -vvv -e . \
23+
--extra-index-url https://download.pytorch.org/whl/test/cu128
2224

2325
# Run the script
2426
python3 -c 'import vllm'

0 commit comments

Comments
 (0)