mirror of
https://github.com/vale981/ray
synced 2025-03-09 12:56:46 -04:00
36 lines
1,010 B
Bash
Executable file
36 lines
1,010 B
Bash
Executable file
#!/usr/bin/env bash
|
|
|
|
# set -x
|
|
|
|
ROOT_DIR=$(cd "$(dirname "${BASH_SOURCE:-$0}")"; pwd)
|
|
|
|
if [ "$1" == "--load" ]; then
|
|
check_load=true
|
|
elif [ "$1" == "--logs" ]; then
|
|
check_load=false
|
|
else
|
|
echo "Usage: $0 [--load|--logs]"
|
|
exit 1
|
|
fi
|
|
|
|
cd "$ROOT_DIR"
|
|
|
|
for workload_file in "$ROOT_DIR"/workloads/*; do
|
|
file_name=$(basename -- $workload_file)
|
|
workload_name="${file_name%.*}"
|
|
if $check_load; then
|
|
echo -n "$workload_name: "
|
|
ray --logging-level=WARNING exec config.yaml --cluster-name="$workload_name" uptime 2>/dev/null || echo "<offline>"
|
|
else
|
|
echo "======================================================================"
|
|
echo "WORKLOAD: $workload_name"
|
|
echo "======================================================================"
|
|
|
|
ray exec config.yaml --cluster-name="$workload_name" "tmux capture-pane -p"
|
|
echo ""
|
|
echo "ssh to this machine with:"
|
|
echo " ray attach $ROOT_DIR/config.yaml --cluster-name=$workload_name"
|
|
echo ""
|
|
echo ""
|
|
fi
|
|
done
|