2017-03-18 23:44:54 -07:00
|
|
|
#!/usr/bin/env bash
|
|
|
|
|
|
|
|
# Cause the script to exit if a single command fails.
|
|
|
|
set -e
|
|
|
|
|
2017-05-31 17:30:46 -07:00
|
|
|
# Show explicitly which commands are currently running.
|
2017-05-18 17:04:56 -07:00
|
|
|
set -x
|
|
|
|
|
2018-12-05 14:59:07 -08:00
|
|
|
MEMORY_SIZE="20G"
|
|
|
|
SHM_SIZE="20G"
|
|
|
|
|
2017-03-18 23:44:54 -07:00
|
|
|
ROOT_DIR=$(cd "$(dirname "${BASH_SOURCE:-$0}")"; pwd)
|
|
|
|
|
2017-07-16 11:51:33 -07:00
|
|
|
DOCKER_SHA=$($ROOT_DIR/../../build-docker.sh --output-sha --no-cache)
|
2019-03-07 12:09:03 -08:00
|
|
|
SUPPRESS_OUTPUT=$ROOT_DIR/../suppress_output
|
2017-03-18 23:44:54 -07:00
|
|
|
echo "Using Docker image" $DOCKER_SHA
|
|
|
|
|
2019-02-28 16:32:22 -08:00
|
|
|
######################## RLLIB TESTS #################################
|
2019-02-21 14:35:18 +08:00
|
|
|
|
2019-02-28 16:32:22 -08:00
|
|
|
source $ROOT_DIR/run_rllib_tests.sh
|
2018-11-29 13:33:39 -08:00
|
|
|
|
2019-03-02 19:21:08 -08:00
|
|
|
######################## TUNE TESTS #################################
|
|
|
|
|
|
|
|
bash $ROOT_DIR/run_tune_tests.sh ${MEMORY_SIZE} ${SHM_SIZE} $DOCKER_SHA
|
2018-11-29 13:33:39 -08:00
|
|
|
|
2019-04-19 11:43:22 -07:00
|
|
|
######################## RAY BACKEND TESTS #################################
|
|
|
|
|
|
|
|
$SUPPRESS_OUTPUT docker run --rm --shm-size=60G --memory=60G $DOCKER_SHA \
|
|
|
|
python /ray/ci/jenkins_tests/miscellaneous/large_memory_test.py
|
|
|
|
|
2019-02-28 16:32:22 -08:00
|
|
|
######################## SGD TESTS #################################
|
2018-12-18 10:40:01 -08:00
|
|
|
|
2019-03-07 12:09:03 -08:00
|
|
|
$SUPPRESS_OUTPUT docker run --rm --shm-size=${SHM_SIZE} --memory=${MEMORY_SIZE} $DOCKER_SHA \
|
2019-06-01 21:39:22 -07:00
|
|
|
python -m pytest /ray/python/ray/experimental/sgd/tests
|