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
|