* Use srun to replace mpirun for now