Skip to content

Commit

Permalink
Update torch version for nv-torch-latest-cpu (deepspeedai#5086)
Browse files Browse the repository at this point in the history
Given the name of this test, we should be running a newer version of
torch than we were, this updates to a newer version.

Total test coverage remains the same:
```
torch 1.12
========== 151 passed, 891 skipped, 20 warnings in 126.58s (0:02:06) ===========
========= 4 passed, 61 skipped, 5342 deselected, 2 warnings in 48.66s ==========

torch 2.2
========== 151 passed, 880 skipped, 20 warnings in 157.64s (0:02:37) ===========
========= 4 passed, 62 skipped, 5330 deselected, 2 warnings in 55.73s ==========
```
  • Loading branch information
loadams authored and amaurya committed Feb 17, 2024
1 parent e53ff93 commit 2628424
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions .github/workflows/nv-torch-latest-cpu.yml
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@ jobs:

- name: Install pytorch
run: |
pip install torch==1.12.0+cpu torchvision==0.13.0+cpu torchaudio==0.12.0 --extra-index-url https://download.pytorch.org/whl/cpu
pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/cpu
python -c "import torch; print('torch:', torch.__version__, torch)"
python -c "import torch; print('CUDA available:', torch.cuda.is_available())"
Expand All @@ -46,5 +46,5 @@ jobs:
run: |
unset TORCH_CUDA_ARCH_LIST # only jit compile for current arch
cd tests
TRANSFORMERS_CACHE=/tmp/transformers_cache/ pytest $PYTEST_OPTS -n 4 unit/ --torch_ver="1.12"
TRANSFORMERS_CACHE=/tmp/transformers_cache/ pytest $PYTEST_OPTS -m 'sequential' unit/ --torch_ver="1.12"
TRANSFORMERS_CACHE=/tmp/transformers_cache/ pytest $PYTEST_OPTS -n 4 unit/ --torch_ver="2.2"
TRANSFORMERS_CACHE=/tmp/transformers_cache/ pytest $PYTEST_OPTS -m 'sequential' unit/ --torch_ver="2.2"

0 comments on commit 2628424

Please sign in to comment.