#!/bin/bash
#SBATCH --job-name="verify"
#SBATCH --partition=gpuA40x4
#SBATCH --mem=12G
#SBATCH --nodes=1
#SBATCH --ntasks-per-node=1
#SBATCH --cpus-per-task=1
#SBATCH --constraint="projects"
#SBATCH --gpus-per-node=1
#SBATCH --gpu-bind=closest
#SBATCH --account=bche-delta-gpu
#SBATCH -t 00:25:00

module reset

SUFFIX=${1:-op9}

OUTDIR=output/simple
mkdir -p $OUTDIR

OUTFILE=$OUTDIR/verify_${SUFFIX}.out
ERRFILE=$OUTDIR/verify_${SUFFIX}.err

echo "Logging to $OUTFILE and $ERRFILE"

exec > $OUTFILE 2> $ERRFILE

echo -e "job $SLURM_JOBID is starting on $(hostname)\n\n"

srun ./build/test_gpt2_${SUFFIX}