CHPC SLURM convenience utilities

scluster

#!/bin/bash
set -e

if [[ "$HOSTNAME" == "notchpeak"* ]]; then
  CLUSTER="np"
elif [[ "$HOSTNAME" == "kingspeak"* ]]; then
  CLUSTER="kp"
else
  echo "Unable to determine cluster from hostname: $HOSTNAME" > /dev/stderr
  exit 1
fi

echo $CLUSTER

sq

#!/bin/bash
set -e

CLUSTER=$(scluster)

squeue --account=lin-${CLUSTER}

srunb

#!/bin/bash
set -e

CLUSTER=$(scluster)

srun --time=24:00:00 --nodes=1 --account=lin-${CLUSTER} --partition=lin-${CLUSTER} -o stdout.log -e stderr.log $@ &

sruni

#!/bin/bash
set -e

CLUSTER=$(scluster)

srun --time=24:00:00 --nodes=1 --account=lin-${CLUSTER} --partition=lin-${CLUSTER} --pty bash