Skip to content

Commit 1296c74

Browse files
authored
Added windows gpu test to circle ci (#1440)
* [WIP] Added windows gpu test to circle ci * Updated windows config * Updated conda installation * Updated miniconda install command * Removed conda installation * Updated configuration
1 parent 5a83aea commit 1296c74

File tree

2 files changed

+41
-1
lines changed

2 files changed

+41
-1
lines changed

.circleci/config.yml

Lines changed: 36 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,14 @@ one_gpu: &one_gpu
2525
# https://circleci.com/product/features/resource-classes/#linux-vm
2626
resource_class: gpu.small
2727

28+
29+
one_gpu_windows: &one_gpu_windows
30+
machine:
31+
resource_class: windows.gpu.nvidia.medium
32+
image: windows-server-2019-nvidia:stable
33+
shell: bash.exe
34+
35+
2836
two_gpus: &two_gpus
2937
machine:
3038
# https://circleci.com/docs/2.0/configuration-reference/#available-linux-gpu-images
@@ -100,7 +108,7 @@ jobs:
100108
- <<: *run_pytorch_container
101109
- <<: *install_dependencies
102110
- run:
103-
name: Run GPU Unit Tests
111+
name: Run GPU Unit Tests and Examples
104112
command: |
105113
106114
# pytest on cuda
@@ -146,6 +154,32 @@ jobs:
146154
bash <(curl -s https://codecov.io/bash) -Z -F gpu
147155
148156
157+
one_gpu_windows_tests:
158+
<<: *one_gpu_windows
159+
160+
working_directory: << pipeline.parameters.workingdir >>
161+
162+
steps:
163+
- checkout
164+
- run:
165+
name: Trigger job if modified
166+
command: |
167+
bash .circleci/trigger_if_modified.sh "^(ignite|tests|examples|\.circleci).*"
168+
169+
- run:
170+
name: Install dependencies
171+
command: |
172+
conda --version
173+
conda install -y pytorch torchvision cudatoolkit=10.1 -c pytorch
174+
pip install -r requirements-dev.txt
175+
pip install .
176+
177+
- run:
178+
name: Run GPU Unit Tests
179+
command: |
180+
# pytest on cuda
181+
SKIP_DISTRIB_TESTS=1 bash tests/run_gpu_tests.sh
182+
149183
two_gpus_tests:
150184
<<: *two_gpus
151185

@@ -302,6 +336,7 @@ workflows:
302336

303337
jobs:
304338
- one_gpu_tests
339+
- one_gpu_windows_tests
305340
- two_gpus_tests
306341
- two_gpus_check_dist_cifar10_example
307342
- two_gpus_hvd_tests

tests/run_gpu_tests.sh

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,11 @@ set -xeu
1010

1111
py.test --cov ignite --cov-report term-missing --cov-report xml -vvv tests/ -k 'on_cuda'
1212

13+
# https://pubs.opengroup.org/onlinepubs/009695399/utilities/xcu_chap02.html#tag_02_06_02
14+
if [ "${SKIP_DISTRIB_TESTS:-0}" -eq "1" ]; then
15+
exit 0
16+
fi
17+
1318
py.test --cov ignite --cov-append --cov-report term-missing --cov-report xml -vvv tests/ -m distributed
1419

1520

0 commit comments

Comments
 (0)