Просмотр исходного кода

Script and docker wrapper for a short "kick-the-tires" test

Ian Goldberg 11 месяцев назад
Родитель
Сommit
ad7b01c95c
2 измененных файлов с 45 добавлено и 0 удалено
  1. 25 0
      docker/short-test
  2. 20 0
      short_exp.py

+ 25 - 0
docker/short-test

@@ -0,0 +1,25 @@
+#!/bin/bash
+
+# cd into the directory containing this script (from the bash faq 028)
+if [[ $BASH_SOURCE = */* ]]; then
+  cd -- "${BASH_SOURCE%/*}/" || exit
+fi
+
+DOCKER_ENV="env"
+if [ "$OVERLOAD_CORES" != "" ]; then
+    DOCKER_ENV="$DOCKER_ENV OVERLOAD_CORES=$OVERLOAD_CORES"
+fi
+if [ "$SHRINK_TO_MEM" != "" ]; then
+    DOCKER_ENV="$DOCKER_ENV SHRINK_TO_MEM=$SHRINK_TO_MEM"
+fi
+
+./start-docker
+docker exec -it ${TEEMS_DOCKER_PREFIX}teems $DOCKER_ENV ./short_exp.py
+docker exec -it ${TEEMS_DOCKER_PREFIX}teems ./logs_to_csv.py Short_test/Public Short_test/public.csv
+echo
+echo "=== Short test output ==="
+echo
+docker exec -it ${TEEMS_DOCKER_PREFIX}teems cat Short_test/public.csv
+echo
+
+./stop-docker >/dev/null

+ 20 - 0
short_exp.py

@@ -0,0 +1,20 @@
+#!/usr/bin/python3
+
+from run_experiments import run_exp
+
+LOG_FOLDER = "Short_test/"
+
+NUM_EPOCHS = 3
+PRIV_OUT = 1
+PRIV_IN = 1
+PUB_OUT = 1
+PUB_IN = 1
+# B = message size (bytes)
+B = 256
+
+PRIVATE_ROUTE = False
+N = [1<<7]
+M = [4]
+T = [1]
+
+run_exp(LOG_FOLDER, PRIVATE_ROUTE, NUM_EPOCHS, N, M, T, B, PRIV_OUT, PRIV_IN, PUB_OUT, PUB_IN)