userbenchmark/release-test/run_release_test.sh

pytorch · Dec 13, 2023 · 1dc6916 · 1dc6916
1 parent 041a093
commit 1dc6916
Showing 1 changed file with 30 additions and 31 deletions.
diff --git 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