#!/bin/bash
#SBATCH -N 2 -n 6
#SBATCH --partition=checkpt
#SBATCH --ntasks-per-node=4

env | grep SLURM > slurm-env.txt
# SLURM_TASKS_PER_NODE has format NUM1(xNUM2)
# SLURM_NTASKS is the product of NUM1 and NUM2
# SLURM_NNODES is NUM2
# SLURM_JOB_NUM_NODES is NUM2
TASKS_PER_NODE=$(echo ${SLURM_TASKS_PER_NODE} | cut -f1 -d\()
echo "SLURM_TASKS_PER_NODE=$SLURM_TASKS_PER_NODE"
echo "TASKS_PER_NODE=$TASKS_PER_NODE"
echo "SLURM_NTASKS=$SLURM_NTASKS"
echo "SLURM_JOB_NUM_NODES=$SLURM_JOB_NUM_NODES"

mpirun hostname