Skip to content
Snippets Groups Projects
Commit 7de460fa authored by Xuan Gu's avatar Xuan Gu
Browse files

Update 3 files

- /scripts/benchmark_single_node.sbatch
- /scripts/benchmark_multi_node.sbatch
- /scripts/run_benchmark.sh
parent f937abc3
No related branches found
No related tags found
No related merge requests found
......@@ -9,6 +9,7 @@ sbatch <<EOT
#SBATCH --gres=gpu:8
#SBATCH --ntasks-per-node=8
#SBATCH --time=0-00:10:00
#SBATCH -o "sbatch_out/benchmark_dim"${1}"_nodes"${2}"_gpus"${3}"_batchsize"${4}"_iteration"${5}".out"
#SBATCH --reservation=bt-xuan_2nodes
# For apptainer
......
......@@ -8,6 +8,7 @@ sbatch <<EOT
#SBATCH --nodes=1
#SBATCH --gres=gpu:${3}
#SBATCH --time=0-0:10:00
#SBATCH -o "sbatch_out/benchmark_dim"${1}"_nodes"${2}"_gpus"${3}"_batchsize"${4}"_iteration"${5}".out"
#SBATCH --reservation=devel
###################21.11.0
......
......@@ -11,14 +11,14 @@ for nodes in {1..1}; do
# For single node
if [ $nodes -eq 1 ]; then
bash -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}
bash scripts/benchmark_single_node.sbatch ${dim} ${nodes} ${gpus} ${batch_size} ${iteration}
sleep 1 # pause to be kind to the scheduler
else
# For multi node
gpus=8
bashsbatch_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}
bash scripts/benchmark_multi_node.sbatch ${dim} ${nodes} ${gpus} ${batch_size} ${iteration}
sleep 1 # pause to be kind to the scheduler
fi
......@@ -37,14 +37,14 @@ for nodes in {1..1}; do
# For single node
if [ $nodes -eq 1 ]; then
bashsbatch_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}
bash scripts/benchmark_single_node.sbatch ${dim} ${nodes} ${gpus} ${batch_size} ${iteration}
sleep 1 # pause to be kind to the scheduler
else
# For multi node
gpus=8
bashsbatch_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}
bash scripts/benchmark_multi_node.sbatch ${dim} ${nodes} ${gpus} ${batch_size} ${iteration}
sleep 1 # pause to be kind to the scheduler
fi
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment