skypilot

1
num_nodes: 2
2

3
setup: |
4
  echo "here"
5
  echo "export TEST_VAR=test" >> ~/.bashrc
6
  [[ -v CUDA_VISIBLE_DEVICES ]] && exit 1 || exit 0
7
  export CUDA_VISIBLE_DEVICES=1
8
  [[ -v CUDA_VISIBLE_DEVICES ]] && exit 0 || exit 1
9

10
run: |
11
  if [[ -z "${TEST_VAR}" ]]; then
12
    echo "TEST_VAR is not set"
13
    exit 1
14
  else
15
    echo "TEST_VAR is set to ${TEST_VAR}"
16
  fi
17

18
  if [[ -z "$SKYPILOT_NODE_RANK" ]]; then
19
    echo "SKYPILOT_NODE_RANK is not set"
20
    exit 1
21
  else
22
    echo "SKYPILOT_NODE_RANK is set to ${SKYPILOT_NODE_RANK}"
23
  fi
24

25
  if [[ -z "$SKYPILOT_NODE_IPS" ]]; then
26
    echo "SKYPILOT_NODE_IPS is not set"
27
    exit 1
28
  else
29
    echo "SKYPILOT_NODE_IPS is set to ${SKYPILOT_NODE_IPS}"
30
    echo "${SKYPILOT_NODE_IPS}"
31
    echo "${SKYPILOT_NODE_IPS}" | wc -l | grep 2 || exit 1
32
  fi
33
  worker_addr=`echo "$SKYPILOT_NODE_IPS" | sed -n 2p`
34
  echo Worker IP: $worker_addr
35

36
  if [[ -z "$SKYPILOT_TASK_ID" ]]; then
37
    echo "SKYPILOT_TASK_ID is not set"
38
    exit 1
39
  else
40
    echo "SKYPILOT_TASK_ID is set to ${SKYPILOT_TASK_ID}"
41
  fi
42