Added support for srun, in addition to aprun and mpirun.