mirror of
https://github.com/vale981/ray
synced 2025-03-06 18:41:40 -05:00
131 lines
4.4 KiB
Bash
Executable file
131 lines
4.4 KiB
Bash
Executable file
#!/usr/bin/env bash
|
|
# This script runs all of the application tests.
|
|
# Currently includes an IMPALA stress test and a SGD stress test.
|
|
# on both Python 2.7 and 3.6.
|
|
# All tests use a separate cluster, and each cluster
|
|
# will be destroyed upon test completion (or failure).
|
|
|
|
# Note that if the environment variable DEBUG_MODE is detected,
|
|
# the clusters will not be automatically shut down after the test runs.
|
|
|
|
# This script will exit with code 1 if the test did not run successfully.
|
|
|
|
|
|
ROOT_DIR=$(cd "$(dirname "${BASH_SOURCE:-$0}")"; pwd)
|
|
RAY_VERSION=$(git describe --tags --abbrev=0)
|
|
RESULT_FILE=$ROOT_DIR/"results-$(date '+%Y-%m-%d_%H-%M-%S').log"
|
|
|
|
echo "Testing on latest version of Ray: $RAY_VERSION"
|
|
echo "Logging to" $RESULT_FILE
|
|
touch $RESULT_FILE
|
|
|
|
# This function identifies the right string for the Ray wheel.
|
|
_find_wheel_str(){
|
|
local python_version=$1
|
|
# echo "PYTHON_VERSION", $python_version
|
|
local wheel_str=""
|
|
if [ $python_version == "p27" ]; then
|
|
wheel_str="cp27-cp27mu"
|
|
else
|
|
wheel_str="cp36-cp36m"
|
|
fi
|
|
echo $wheel_str
|
|
}
|
|
|
|
# Total time is roughly 25 minutes.
|
|
# Actual test runtime is roughly 10 minutes.
|
|
test_impala(){
|
|
local PYTHON_VERSION=$1
|
|
local WHEEL_STR=$(_find_wheel_str $PYTHON_VERSION)
|
|
|
|
pushd "$ROOT_DIR"
|
|
local TEST_NAME="rllib_impala_$PYTHON_VERSION"
|
|
local CLUSTER="$TEST_NAME.yaml"
|
|
echo "Creating IMPALA cluster YAML from template."
|
|
|
|
cat application_cluster_template.yaml |
|
|
sed -e "
|
|
s/<<<CLUSTER_NAME>>>/$TEST_NAME/;
|
|
s/<<<RAY_VERSION>>>/$RAY_VERSION/;
|
|
s/<<<HEAD_TYPE>>>/g3.16xlarge/;
|
|
s/<<<WORKER_TYPE>>>/m5.24xlarge/;
|
|
s/<<<MIN_WORKERS>>>/5/;
|
|
s/<<<MAX_WORKERS>>>/5/;
|
|
s/<<<PYTHON_VERSION>>>/$PYTHON_VERSION/;
|
|
s/<<<WHEEL_STR>>>/$WHEEL_STR/;" > $CLUSTER
|
|
|
|
echo "Try running IMPALA stress test."
|
|
{
|
|
RLLIB_DIR=../../python/ray/rllib/
|
|
ray up -y $CLUSTER &&
|
|
ray rsync_up $CLUSTER $RLLIB_DIR/tuned_examples/ tuned_examples/ &&
|
|
sleep 1 &&
|
|
ray exec $CLUSTER "
|
|
rllib train -f tuned_examples/atari-impala-large.yaml --redis-address='localhost:6379' --queue-trials" &&
|
|
echo "PASS: IMPALA Test for" $PYTHON_VERSION >> $RESULT_FILE
|
|
} || echo "FAIL: IMPALA Test for" $PYTHON_VERSION >> $RESULT_FILE
|
|
|
|
# Tear down cluster.
|
|
if [ "$DEBUG_MODE" = "" ]; then
|
|
ray down -y $CLUSTER
|
|
rm $CLUSTER
|
|
else
|
|
echo "Not tearing down cluster" $CLUSTER
|
|
fi
|
|
popd
|
|
}
|
|
|
|
# Total runtime is about 20 minutes (if the AWS spot instance order is fulfilled).
|
|
# Actual test runtime is roughly 10 minutes.
|
|
test_sgd(){
|
|
local PYTHON_VERSION=$1
|
|
local WHEEL_STR=$(_find_wheel_str $PYTHON_VERSION)
|
|
|
|
pushd "$ROOT_DIR"
|
|
local TEST_NAME="sgd_$PYTHON_VERSION"
|
|
local CLUSTER="$TEST_NAME.yaml"
|
|
echo "Creating SGD cluster YAML from template."
|
|
|
|
cat application_cluster_template.yaml |
|
|
sed -e "
|
|
s/<<<CLUSTER_NAME>>>/$TEST_NAME/;
|
|
s/<<<RAY_VERSION>>>/$RAY_VERSION/;
|
|
s/<<<HEAD_TYPE>>>/g3.16xlarge/;
|
|
s/<<<WORKER_TYPE>>>/g3.16xlarge/;
|
|
s/<<<MIN_WORKERS>>>/3/;
|
|
s/<<<MAX_WORKERS>>>/3/;
|
|
s/<<<PYTHON_VERSION>>>/$PYTHON_VERSION/;
|
|
s/<<<WHEEL_STR>>>/$WHEEL_STR/;" > $CLUSTER
|
|
|
|
echo "Try running SGD stress test."
|
|
{
|
|
SGD_DIR=$ROOT_DIR/../../python/ray/experimental/sgd/
|
|
ray up -y $CLUSTER &&
|
|
# TODO: fix submit so that args work
|
|
ray rsync_up $CLUSTER $SGD_DIR/mnist_example.py mnist_example.py &&
|
|
sleep 1 &&
|
|
ray exec $CLUSTER "
|
|
python mnist_example.py --redis-address=localhost:6379 --num-iters=2000 --num-workers=8 --devices-per-worker=2 --gpu" &&
|
|
echo "PASS: SGD Test for" $PYTHON_VERSION >> $RESULT_FILE
|
|
} || echo "FAIL: SGD Test for" $PYTHON_VERSION >> $RESULT_FILE
|
|
|
|
# Tear down cluster.
|
|
if [ "$DEBUG_MODE" = "" ]; then
|
|
ray down -y $CLUSTER
|
|
rm $CLUSTER
|
|
else
|
|
echo "Not tearing down cluster" $CLUSTER
|
|
fi
|
|
popd
|
|
}
|
|
|
|
# RUN TESTS
|
|
for PYTHON_VERSION in "p27" "p36"
|
|
do
|
|
test_impala $PYTHON_VERSION
|
|
test_sgd $PYTHON_VERSION
|
|
done
|
|
|
|
cat $RESULT_FILE
|
|
cat $RESULT_FILE | grep FAIL > test.log
|
|
[ ! -s test.log ] || exit 1
|