mirror of
https://github.com/vale981/ray
synced 2025-03-06 10:31:39 -05:00

Stress testing for Jenkins. <!-- Thank you for your contribution! Please review https://github.com/ray-project/ray/blob/master/CONTRIBUTING.rst before opening a pull request. --> <!-- Please give a short brief about these changes. --> TODO: - [x] Enable a common keypair for autoscaling - [x] Add automatic timeouts? - [x] Switch out key pair one last time before merge
28 lines
939 B
Bash
Executable file
28 lines
939 B
Bash
Executable file
#!/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="20G"
|
|
SHM_SIZE="20G"
|
|
|
|
docker build -q --no-cache -t ray-project/base-deps docker/base-deps
|
|
|
|
# Add Ray source
|
|
git rev-parse HEAD > ./docker/stress_test/git-rev
|
|
git archive -o ./docker/stress_test/ray.tar $(git rev-parse HEAD)
|
|
DOCKER_SHA=$(docker build --no-cache -q -t ray-project/stress_test docker/stress_test)
|
|
|
|
echo "Using Docker image" $DOCKER_SHA
|
|
docker run --rm --shm-size=${SHM_SIZE} --memory=${MEMORY_SIZE} \
|
|
-e AWS_ACCESS_KEY_ID -e AWS_SECRET_ACCESS_KEY -e RAY_AWS_SSH_KEY \
|
|
$DOCKER_SHA \
|
|
bash /ray/ci/stress_tests/run_stress_tests.sh
|
|
|
|
# docker run --rm --shm-size=${SHM_SIZE} --memory=${MEMORY_SIZE} \
|
|
# -e AWS_ACCESS_KEY_ID -e AWS_SECRET_ACCESS_KEY -e RAY_AWS_SSH_KEY \
|
|
# $DOCKER_SHA \
|
|
# bash /ray/ci/stress_tests/run_application_stress_tests.sh
|