Update srun infrastructure for LAM/MPI support:

1. New srun argument --mpi=<type>  (e.g. "--mpi=lam")
2. New environment variable set SLURM_TASKS_PER_NODE
3. Launch a single task per node if --mpi=lam