30 lines
605 B
Bash
30 lines
605 B
Bash
#!/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} |