mirror of
https://github.com/vale981/ray
synced 2025-03-09 12:56:46 -04:00
44 lines
901 B
Bash
44 lines
901 B
Bash
|
#!/usr/bin/env bash
|
||
|
|
|
||
|
|
# Cause the script to exit if a single command fails.
|
||
|
|
set -e
|
||
|
|
|
||
|
|
# Show explicitly which commands are currently running.
|
||
|
|
set -x
|
||
|
|
|
||
|
|
|
||
|
|
MEMORY_SIZE="8G"
|
||
|
|
SHM_SIZE="4G"
|
||
|
|
|
||
|
|
ROOT_DIR=$(cd "$(dirname "${BASH_SOURCE:-$0}")"; pwd)
|
||
|
|
|
||
|
|
SUPPRESS_OUTPUT=$ROOT_DIR/../suppress_output
|
||
|
|
|
||
|
|
function retry {
|
||
|
|
local n=1
|
||
|
|
local max=3
|
||
|
|
|
||
|
|
while true; do
|
||
|
|
"$@" && break || {
|
||
|
|
if [[ $n -lt $max ]]; then
|
||
|
|
((n++))
|
||
|
|
echo "Command failed. Attempt $n/$max:"
|
||
|
|
else
|
||
|
|
echo "The command has failed after $n attempts."
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
done
|
||
|
|
}
|
||
|
|
|
||
|
|
if [[ ! -z "$RUN_TUNE_TESTS" ]]; then
|
||
|
|
retry bash $ROOT_DIR/run_tune_tests.sh ${MEMORY_SIZE} ${SHM_SIZE}
|
||
|
|
fi
|
||
|
|
|
||
|
|
if [[ ! -z "$RUN_DOC_TESTS" ]]; then
|
||
|
|
retry bash $ROOT_DIR/run_doc_tests.sh ${MEMORY_SIZE} ${SHM_SIZE}
|
||
|
|
fi
|
||
|
|
|
||
|
|
if [[ ! -z "$RUN_SGD_TESTS" ]]; then
|
||
|
|
retry bash $ROOT_DIR/run_sgd_tests.sh ${MEMORY_SIZE} ${SHM_SIZE}
|
||
|
|
fi
|