Skip to content

Commit 921b878

Browse files
authored
Update submit-bench.sh
1 parent 43812d8 commit 921b878

File tree

1 file changed

+18
-2
lines changed

1 file changed

+18
-2
lines changed

.github/workflows/phoenix/submit-bench.sh

Lines changed: 18 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ sbatch_cpu_opts="\
2020
"
2121

2222
sbatch_gpu_opts="\
23-
#SBATCH -CV100
23+
#SBATCH -p gpu-v100,gpu-a100,gpu-h100,gpu-h200,gpu-l40s
2424
#SBATCH --ntasks-per-node=4 # Number of cores per node required
2525
#SBATCH -G2\
2626
"
@@ -42,7 +42,7 @@ sbatch <<EOT
4242
#SBATCH --account=gts-sbryngelson3 # charge account
4343
#SBATCH -N1 # Number of nodes required
4444
$sbatch_device_opts
45-
#SBATCH -t 02:00:00 # Duration of the job (Ex: 15 mins)
45+
#SBATCH -t 04:00:00 # Duration of the job (Ex: 15 mins)
4646
#SBATCH -q embers # QOS Name
4747
#SBATCH -o$job_slug.out # Combined output and error messages file
4848
#SBATCH -W # Do not exit until the submitted job terminates.
@@ -51,6 +51,22 @@ set -e
5151
set -x
5252
5353
cd "\$SLURM_SUBMIT_DIR"
54+
echo "Running PR"
55+
cd pr
56+
57+
echo "Running in $(pwd):"
58+
59+
job_slug="$job_slug"
60+
job_device="$2"
61+
62+
. ./mfc.sh load -c p -m $2
63+
64+
$sbatch_script_contents
65+
66+
cd "\$SLURM_SUBMIT_DIR"
67+
echo "Running Master"
68+
cd master
69+
5470
echo "Running in $(pwd):"
5571
5672
job_slug="$job_slug"

0 commit comments

Comments
 (0)