From 37b604f5ed07640cb9c3d2bc87766a752d57a92a Mon Sep 17 00:00:00 2001
From: Xuan Gu <xuan.gu@liu.se>
Date: Tue, 30 May 2023 12:33:17 +0000
Subject: [PATCH] Update file benchmark_sbatch_submit.sh
---
scripts/benchmark_sbatch_submit.sh | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/scripts/benchmark_sbatch_submit.sh b/scripts/benchmark_sbatch_submit.sh
index f345921..a756470 100644
--- a/scripts/benchmark_sbatch_submit.sh
+++ b/scripts/benchmark_sbatch_submit.sh
@@ -8,6 +8,7 @@ if [ $NUM_NODES -eq 1 ]; then
for gpus in {1,8}; do
for batch_size in {64,128}; do
for iteration in {1..1}; do
+
echo dim ${dim}, nodes ${nodes}, gpus ${gpus}, batch_size ${batch_size}, iteration ${iteration}
# For single node
sbatch -o sbatch_out/benchmark_dim${dim}_nodes${nodes}_gpus${gpus}_batchsize${batch_size}_iteration${iteration}.out scripts/benchmark_single_node.sbatch ${dim} ${nodes} ${gpus} ${batch_size} ${iteration}
@@ -23,11 +24,12 @@ else
for gpus in {8,8}; do
for batch_size in {64,128}; do
for iteration in {1..1}; do
- batch_size=$BATCH_SIZE
+
# For multi node
sbatch -o sbatch_out/benchmark_dim${dim}_nodes${nodes}_gpus${gpus}_batchsize${batch_size}_iteration${iteration}.out scripts/benchmark_multi_node.sbatch ${dim} ${nodes} ${gpus} ${batch_size} ${iteration}
sleep 1 # pause to be kind to the scheduler
+
done
done
done
--
GitLab