diff --git a/.github/workflows/phoenix/bench.sh b/.github/workflows/phoenix/bench.sh index 78df3dd13..23e73ebe9 100644 --- a/.github/workflows/phoenix/bench.sh +++ b/.github/workflows/phoenix/bench.sh @@ -1,6 +1,6 @@ #!/bin/bash -n_ranks=4 +n_ranks=2 if [ "$job_device" == "gpu" ]; then n_ranks=$(nvidia-smi -L | wc -l) # number of GPUs on node diff --git a/.github/workflows/phoenix/submit.sh b/.github/workflows/phoenix/submit.sh index d82d6aa5a..ba371c41a 100644 --- a/.github/workflows/phoenix/submit.sh +++ b/.github/workflows/phoenix/submit.sh @@ -15,11 +15,13 @@ fi sbatch_cpu_opts="\ #SBATCH -p cpu-small # partition +#SBATCH -N1 # Number of nodes required #SBATCH --ntasks-per-node=24 # Number of cores per node required #SBATCH --mem-per-cpu=2G # Memory per core\ " sbatch_gpu_opts="\ +#SBATCH -n4 # Number of ranks required #SBATCH -CV100-16GB #SBATCH -G2\ " @@ -37,9 +39,8 @@ job_slug="`basename "$1" | sed 's/\.sh$//' | sed 's/[^a-zA-Z0-9]/-/g'`-$2" sbatch <