Skip to content

Commit 1a62f7f

Browse files
kragnizBorda
andauthored
ci: adjust torch version requirements in IPU pipeline (#7383)
Co-authored-by: Jirka Borovec <[email protected]>
1 parent d52e0a8 commit 1a62f7f

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

.azure-pipelines/ipu-tests.yml

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,10 +51,14 @@ jobs:
5151
5252
- bash: |
5353
export GIT_TERMINAL_PROMPT=1
54-
pip install --requirement requirements.txt
5554
python -c "fname = 'requirements/extra.txt' ; lines = [line for line in open(fname).readlines() if 'fairscale' not in line] ; open(fname, 'w').writelines(lines)"
5655
python -c "fname = 'requirements/extra.txt' ; lines = [line for line in open(fname).readlines() if 'horovod' not in line] ; open(fname, 'w').writelines(lines)"
56+
57+
python ./requirements/adjust_versions.py requirements/extra.txt
58+
python ./requirements/adjust_versions.py requirements/examples.txt
59+
5760
pip install --requirement ./requirements/devel.txt --upgrade-strategy only-if-needed
61+
5862
pip list
5963
displayName: 'Install dependencies'
6064
@@ -82,4 +86,6 @@ jobs:
8286
source ${{ variables.poplar_sdk }}/popart-ubuntu*/enable.sh
8387
8488
python -m coverage run --source pytorch_lightning -m pytest pytorch_lightning tests -v --junitxml=$(Build.StagingDirectory)/test-results.xml --durations=50
89+
env:
90+
MKL_THREADING_LAYER: "GNU"
8591
displayName: 'Testing: standard'

0 commit comments

Comments
 (0)