From 1dc6916654099453f5786c0ee9a0f966685494c9 Mon Sep 17 00:00:00 2001
From: atalman <atalman@fb.com>
Date: Wed, 13 Dec 2023 13:49:00 -0800
Subject: [PATCH] userbenchmark/release-test/run_release_test.sh

---
 .../release-test/run_release_test.sh          | 61 +++++++++----------
 1 file changed, 30 insertions(+), 31 deletions(-)

diff --git a/userbenchmark/release-test/run_release_test.sh b/userbenchmark/release-test/run_release_test.sh
index e9992960d9..f2fecf40ba 100644
--- a/userbenchmark/release-test/run_release_test.sh
+++ b/userbenchmark/release-test/run_release_test.sh
@@ -20,40 +20,39 @@ fi
 nvcc --version
 sudo apt-get install bc
 # run mnist
-mkdir -p "${RESULT_DIR}/mnist"
-pushd "${EXAMPLES_DIR}/mnist"
-export LOG_FILE=${RESULT_DIR}/mnist/result.log
-export MEM_FILE=${RESULT_DIR}/mnist/result_mem.log
-${PREFIX} bash "${CURRENT_DIR}/monitor_proc.sh" python main.py --epochs 10
-
+#mkdir -p "${RESULT_DIR}/mnist"
+#pushd "${EXAMPLES_DIR}/mnist"
+#export LOG_FILE=${RESULT_DIR}/mnist/result.log
+#export MEM_FILE=${RESULT_DIR}/mnist/result_mem.log
+#${PREFIX} bash "${CURRENT_DIR}/monitor_proc.sh" python main.py --epochs 10
 
 # run mnist-hogwild
-#mkdir -p ${RESULT_DIR}/mnist_hogwild
-#pushd "${EXAMPLES_DIR}/mnist_hogwild"
-#export LOG_FILE=${RESULT_DIR}/mnist_hogwild/result.log
-#export MEM_FILE=${RESULT_DIR}/mnist_hogwild/result_mem.log
-#${PREFIX} bash "${CURRENT_DIR}/monitor_proc.sh" python main.py --epochs 10
+mkdir -p ${RESULT_DIR}/mnist_hogwild
+pushd "${EXAMPLES_DIR}/mnist_hogwild"
+export LOG_FILE=${RESULT_DIR}/mnist_hogwild/result.log
+export MEM_FILE=${RESULT_DIR}/mnist_hogwild/result_mem.log
+${PREFIX} bash "${CURRENT_DIR}/monitor_proc.sh" python main.py --epochs 10
 # run CPU WLM LSTM
-#mkdir -p ${RESULT_DIR}/wlm_cpu_lstm
-#pushd "${EXAMPLES_DIR}/word_language_model"
-#export LOG_FILE=${RESULT_DIR}/wlm_cpu_lstm/result.log
-#export MEM_FILE=${RESULT_DIR}/wlm_cpu_lstm/result_mem.log
-#${PREFIX} bash "${CURRENT_DIR}/monitor_proc.sh" python main.py --epochs 10 --model LSTM
+mkdir -p ${RESULT_DIR}/wlm_cpu_lstm
+pushd "${EXAMPLES_DIR}/word_language_model"
+export LOG_FILE=${RESULT_DIR}/wlm_cpu_lstm/result.log
+export MEM_FILE=${RESULT_DIR}/wlm_cpu_lstm/result_mem.log
+${PREFIX} bash "${CURRENT_DIR}/monitor_proc.sh" python main.py --epochs 10 --model LSTM
 # run GPU WLM LSTM
-#mkdir -p ${RESULT_DIR}/wlm_gpu_lstm
-#pushd "${EXAMPLES_DIR}/word_language_model"
-#export LOG_FILE=${RESULT_DIR}/wlm_gpu_lstm/result.log
-#export MEM_FILE=${RESULT_DIR}/wlm_gpu_lstm/result_mem.log
-#${PREFIX} bash "${CURRENT_DIR}/monitor_proc.sh" python main.py --epochs 10 --model LSTM --cuda
+mkdir -p ${RESULT_DIR}/wlm_gpu_lstm
+pushd "${EXAMPLES_DIR}/word_language_model"
+export LOG_FILE=${RESULT_DIR}/wlm_gpu_lstm/result.log
+export MEM_FILE=${RESULT_DIR}/wlm_gpu_lstm/result_mem.log
+${PREFIX} bash "${CURRENT_DIR}/monitor_proc.sh" python main.py --epochs 10 --model LSTM --cuda
 # run CPU WLM Transformer
-#mkdir -p ${RESULT_DIR}/wlm_cpu_trans
-#pushd "${EXAMPLES_DIR}/word_language_model"
-#export LOG_FILE=${RESULT_DIR}/wlm_cpu_trans/result.log
-#export MEM_FILE=${RESULT_DIR}/wlm_cpu_trans/result_mem.log
-#${PREFIX} bash "${CURRENT_DIR}/monitor_proc.sh" python main.py --epochs 10 --model Transformer
+mkdir -p ${RESULT_DIR}/wlm_cpu_trans
+pushd "${EXAMPLES_DIR}/word_language_model"
+export LOG_FILE=${RESULT_DIR}/wlm_cpu_trans/result.log
+export MEM_FILE=${RESULT_DIR}/wlm_cpu_trans/result_mem.log
+${PREFIX} bash "${CURRENT_DIR}/monitor_proc.sh" python main.py --epochs 10 --model Transformer
 # run GPU WLM Transformer
-#mkdir -p ${RESULT_DIR}/wlm_gpu_trans
-#pushd "${EXAMPLES_DIR}/word_language_model"
-#export LOG_FILE=${RESULT_DIR}/wlm_gpu_trans/result.log
-#export MEM_FILE=${RESULT_DIR}/wlm_gpu_trans/result_mem.log
-#${PREFIX} bash "${CURRENT_DIR}/monitor_proc.sh" python main.py --epochs 10 --model Transformer --cuda
+mkdir -p ${RESULT_DIR}/wlm_gpu_trans
+pushd "${EXAMPLES_DIR}/word_language_model"
+export LOG_FILE=${RESULT_DIR}/wlm_gpu_trans/result.log
+export MEM_FILE=${RESULT_DIR}/wlm_gpu_trans/result_mem.log
+${PREFIX} bash "${CURRENT_DIR}/monitor_proc.sh" python main.py --epochs 10 --model Transformer --cuda