File: run_gpu_tests.sh

package info (click to toggle)
pytorch-ignite 0.5.1-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 11,712 kB
  • sloc: python: 46,874; sh: 376; makefile: 27
file content (47 lines) | stat: -rw-r--r-- 1,327 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
#!/bin/bash
source "$(dirname "$0")/common_test_functionality.sh"
set -xeu

skip_distrib_tests=${SKIP_DISTRIB_TESTS:-1}
use_last_failed=${USE_LAST_FAILED:-0}
ngpus=${1:-1}

match_tests_expression=${2:-""}
if [ -z "$match_tests_expression" ]; then
    cuda_pattern="cuda"
else
    cuda_pattern="cuda and $match_tests_expression"
fi

run_tests \
    --core_args "-vvv tests/ignite" \
    --cache_dir ".gpu-cuda" \
    --skip_distrib_tests "${skip_distrib_tests}" \
    --use_coverage 1 \
    --match_tests_expression "${cuda_pattern}" \
    --use_last_failed ${use_last_failed}

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

run_tests \
    --core_args "-vvv -m distributed tests/ignite" \
    --cache_dir ".gpu-distrib" \
    --skip_distrib_tests 0 \
    --use_coverage 1 \
    --match_tests_expression "${match_tests_expression}" \
    --use_last_failed ${use_last_failed}


if [ ${ngpus} -gt 1 ]; then
    run_tests \
        --core_args "-vvv -m distributed tests/ignite" \
        --world_size "${ngpus}" \
        --cache_dir ".gpu-distrib-multi" \
        --skip_distrib_tests 0 \
        --use_coverage 1 \
        --match_tests_expression "${match_tests_expression}" \
        --use_last_failed ${use_last_failed}
fi