SLURM utils. Add to your .bashrc

slurm_utils.sh

# =================================================
# SLURM UTILS
# =================================================

# sbatlog: runs sbatch and automatically tail -f slurm-$number.out
function sbatlog
{
    result=$(command sbatch "$@")
    outputid=$(echo $result | awk -F' ' '{print $NF}')
    # fileout="slurm-$outputid.out"  # assumes default file output name
    fileout="$2"  # assuming that $2 is the argument for -o FIXME
    fileout=$(echo $fileout | sed "s:%j:$outputid:g")

    # wait file to be created
    echo "Watching file $fileout"
    while [ ! -f $fileout ]
    do
        sleep 1
    done
    printf "\033c"  # clears screen
    tail -f $fileout
}


# slastlog: opens last created slurm-<pid>.out log with less
function slastlog
{
    last_log="$(ls -lat ./slurm-*.out | head -1 | awk '{print $NF}')"
    echo "Opening $last_log"
    less "$last_log"
}

# slastlog: opens last created slurm-<pid>.out log with tail
function stail
{
    last_log="$(ls -lat ./slurm-*.out | head -1 | awk '{print $NF}')"
    echo "Opening $last_log"
    command tail -f "$last_log"
}

# sgrep: greps the last log file
function sgrep
{
    last_log="$(ls -lat ./slurm-*.out | head -1 | awk '{print $NF}')"
    echo "grepping $last_log"
    command grep "$@" "$last_log" | command less
}