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

Update file run_benchmark.sh

parent d1310d7c
No related branches found
No related tags found
No related merge requests found
......@@ -11,14 +11,14 @@ for nodes in {1..1}; do
# For single node
if [ $nodes -eq 1 ]; then
bash scripts/benchmark_single_node.bash ${dim} ${nodes} ${gpus} ${batch_size} ${iteration}
bash scripts/benchmark_single_node.sh ${dim} ${nodes} ${gpus} ${batch_size} ${iteration}
sleep 1 # pause to be kind to the scheduler
else
# For multi node
gpus=8
bash scripts/benchmark_multi_node.bash ${dim} ${nodes} ${gpus} ${batch_size} ${iteration}
bash scripts/benchmark_multi_node.sh ${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
bash scripts/benchmark_single_node.bash ${dim} ${nodes} ${gpus} ${batch_size} ${iteration}
bash scripts/benchmark_single_node.sh ${dim} ${nodes} ${gpus} ${batch_size} ${iteration}
sleep 1 # pause to be kind to the scheduler
else
# For multi node
gpus=8
bash scripts/benchmark_multi_node.bash ${dim} ${nodes} ${gpus} ${batch_size} ${iteration}
bash scripts/benchmark_multi_node.sh ${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