GPA-Benchmark/drive-gpa-drgpu.sbatch at develop · parallelcodefoundry/GPA-Benchmark · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
#!/bin/bash
#SBATCH -n 1
#SBATCH -t 08:00:00
#SBATCH -c 24
#SBATCH --mem-per-cpu=5g
#SBATCH -p gpu
#SBATCH --gpus-per-task=h100:1
#SBATCH -J drive-gpa-drgpu
#SBATCH -o drive-gpa-drgpu-%A_%a.out
#SBATCH --array=0-16

module load cuda/gcc/11.3.0/zen2/12.3.0
. ~/scratch/spack/share/spack/setup-env.sh
spack load nvhpc
cd ~/scratch/llms4hpc/GPA-Benchmark
. .venv/bin/activate

driver_srun() {
    srun -n 1 -c 24 --mem-per-cpu=5g --gpus-per-task 1 --exact python -m gpa_bench_driver \
    --app "$1" --nsys --swaps ~/scratch/llms4hpc/gpa-bench-drgpu --output-file drgpu-"$1".json \
    --no-progress
}

apps=(
    backprop
    bfs
    b+tree
    gaussian
    heartwall
    hotspot
    huffman
    lavaMD
    lud
    nw
    particlefilter
    pathfinder
    srad
    streamcluster
    exatensor
    lulesh
    xsbench
)

if [ -z "${SLURM_ARRAY_TASK_ID+x}" ]; then
    echo "SLURM_ARRAY_TASK_ID is not set."
    exit 1
fi

APP_NAME=${apps[$SLURM_ARRAY_TASK_ID]}

if [ -z "$APP_NAME" ]; then
    echo "Invalid SLURM_ARRAY_TASK_ID: ${SLURM_ARRAY_TASK_ID}"
    exit 1
fi

driver_srun "$APP_NAME"

wait

exit