Skip to content

Commit

Permalink
Added windows gpu test to circle ci (#1440)
Browse files Browse the repository at this point in the history
* [WIP] Added windows gpu test to circle ci

* Updated windows config

* Updated conda installation

* Updated miniconda install command

* Removed conda installation

* Updated configuration
  • Loading branch information
vfdev-5 authored Nov 6, 2020
1 parent 5a83aea commit 1296c74
Show file tree
Hide file tree
Showing 2 changed files with 41 additions and 1 deletion.
37 changes: 36 additions & 1 deletion .circleci/config.yml
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,14 @@ one_gpu: &one_gpu
# https://circleci.com/product/features/resource-classes/#linux-vm
resource_class: gpu.small


one_gpu_windows: &one_gpu_windows
machine:
resource_class: windows.gpu.nvidia.medium
image: windows-server-2019-nvidia:stable
shell: bash.exe


two_gpus: &two_gpus
machine:
# https://circleci.com/docs/2.0/configuration-reference/#available-linux-gpu-images
Expand Down Expand Up @@ -100,7 +108,7 @@ jobs:
- <<: *run_pytorch_container
- <<: *install_dependencies
- run:
name: Run GPU Unit Tests
name: Run GPU Unit Tests and Examples
command: |
# pytest on cuda
Expand Down Expand Up @@ -146,6 +154,32 @@ jobs:
bash <(curl -s https://codecov.io/bash) -Z -F gpu
one_gpu_windows_tests:
<<: *one_gpu_windows

working_directory: << pipeline.parameters.workingdir >>

steps:
- checkout
- run:
name: Trigger job if modified
command: |
bash .circleci/trigger_if_modified.sh "^(ignite|tests|examples|\.circleci).*"
- run:
name: Install dependencies
command: |
conda --version
conda install -y pytorch torchvision cudatoolkit=10.1 -c pytorch
pip install -r requirements-dev.txt
pip install .
- run:
name: Run GPU Unit Tests
command: |
# pytest on cuda
SKIP_DISTRIB_TESTS=1 bash tests/run_gpu_tests.sh
two_gpus_tests:
<<: *two_gpus

Expand Down Expand Up @@ -302,6 +336,7 @@ workflows:

jobs:
- one_gpu_tests
- one_gpu_windows_tests
- two_gpus_tests
- two_gpus_check_dist_cifar10_example
- two_gpus_hvd_tests
5 changes: 5 additions & 0 deletions tests/run_gpu_tests.sh
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,11 @@ set -xeu

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

# https://pubs.opengroup.org/onlinepubs/009695399/utilities/xcu_chap02.html#tag_02_06_02
if [ "${SKIP_DISTRIB_TESTS:-0}" -eq "1" ]; then
exit 0
fi

py.test --cov ignite --cov-append --cov-report term-missing --cov-report xml -vvv tests/ -m distributed


Expand Down

0 comments on commit 1296c74

Please sign in to comment.