remote/bin/slurm-shell @ d1b9a524e5fe default tip

#!/usr/bin/env bash

set -euo pipefail

read -p "Account [apboyle0]? "
acct="${REPLY:-apboyle0}"

read -p "Hours [4]? "
hours="${REPLY:-4}"

read -p "CPUs [4]? "
cpus="${REPLY:-4}"

read -p "Memory (gb) [16]? "
mem_gb="${REPLY:-16}"
mem_kb=$(( mem_gb * 1024 * 1024 / cpus ))

read -p "X11? "
case $REPLY in
    y | yes | Y | YES)
    x11="--x11"
    ;;

    *)
    x11=""
    ;;
esac

if test -z "$x11"; then
    exec salloc --account="$acct" --nodes=1 --ntasks-per-node=1 --mem-per-cpu="$mem_kb"K --cpus-per-task="$cpus" --time="$hours":00:00
else
    exec salloc --account="$acct" --nodes=1 --ntasks-per-node=1 --mem-per-cpu="$mem_kb"K --cpus-per-task="$cpus" --time="$hours":00:00 --x11
fi