From a4db6e5228ea67e4744e2f2f1e3e8ec221e73303 Mon Sep 17 00:00:00 2001
From: Xuan Gu <xuan.gu@liu.se>
Date: Tue, 30 May 2023 12:49:12 +0000
Subject: [PATCH] Update file run_benchmark.sh
---
scripts/run_benchmark.sh | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/scripts/run_benchmark.sh b/scripts/run_benchmark.sh
index 7b8589c..6c0ef4b 100644
--- a/scripts/run_benchmark.sh
+++ b/scripts/run_benchmark.sh
@@ -10,7 +10,7 @@ for nodes in {1..1}; do
echo dim ${dim}, nodes ${nodes}, gpus ${gpus}, batch_size ${batch_size}, iteration ${iteration}
# For single node
- if [ $NUM_NODES -eq 1 ]; then
+ if [ $nodes -eq 1 ]; then
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}
sleep 1 # pause to be kind to the scheduler
@@ -36,7 +36,7 @@ for nodes in {1..1}; do
echo dim ${dim}, nodes ${nodes}, gpus ${gpus}, batch_size ${batch_size}, iteration ${iteration}
# For single node
- if [ $NUM_NODES -eq 1 ]; then
+ if [ $nodes -eq 1 ]; then
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}
sleep 1 # pause to be kind to the scheduler
--
GitLab