Skip to content

Commit 645eabe

Browse files
carmoccalexierule
authored andcommitted
Speed up IPU CI
1 parent 6aec313 commit 645eabe

File tree

1 file changed

+1
-16
lines changed

1 file changed

+1
-16
lines changed

.azure-pipelines/ipu-tests.yml

Lines changed: 1 addition & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -72,26 +72,11 @@ jobs:
7272
python -c "import poptorch; print(poptorch.__version__)"
7373
displayName: "Check poptorch installation"
7474
75-
- bash: |
76-
wget https://pl-public-data.s3.amazonaws.com/legacy/checkpoints.zip -P legacy/
77-
unzip -o legacy/checkpoints.zip -d legacy/
78-
ls -l legacy/checkpoints/
79-
displayName: 'Get legacy checkpoints'
80-
8175
- bash: |
8276
source ${{ variables.poplar_sdk }}/poplar-ubuntu*/enable.sh
8377
source ${{ variables.poplar_sdk }}/popart-ubuntu*/enable.sh
8478
export POPTORCH_WAIT_FOR_IPU=1
85-
python -m coverage run --source pytorch_lightning -m pytest pytorch_lightning tests -v --junitxml=$(Build.StagingDirectory)/test-results.xml --durations=50
79+
python -m coverage run --source pytorch_lightning -m pytest tests/accelerators/test_ipu.py -v --junitxml=$(Build.StagingDirectory)/test-results.xml --durations=50
8680
env:
8781
MKL_THREADING_LAYER: "GNU"
8882
displayName: 'Testing: standard'
89-
90-
- bash: |
91-
source ${{ variables.poplar_sdk }}/poplar-ubuntu*/enable.sh
92-
source ${{ variables.poplar_sdk }}/popart-ubuntu*/enable.sh
93-
export POPTORCH_WAIT_FOR_IPU=1
94-
bash tests/special_tests.sh
95-
env:
96-
MKL_THREADING_LAYER: "GNU"
97-
displayName: 'Testing: special'

0 commit comments

Comments
 (0)