File: run_tpu_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 (20 lines) | stat: -rw-r--r-- 503 bytes parent folder | download 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
 
#!/bin/bash
source "$(dirname "$0")/common_test_functionality.sh"
set -xeu
use_last_failed=${USE_LAST_FAILED:-0}

run_tests \
    --core_args "-vvv -m tpu tests/ignite" \
    --cache_dir ".tpu" \
    --use_coverage 1 \
    --use_last_failed ${use_last_failed}


if [ -z ${NUM_TPU_WORKERS+x} ]; then
    export NUM_TPU_WORKERS=1
    run_tests \
        --core_args "-vvv -m tpu tests/ignite" \
        --cache_dir ".tpu-multi" \
        --use_coverage 1 \
        --use_last_failed ${use_last_failed}
fi