Fix setting of SLURM_TASKS_PER_NODE

Fix the setting of SLURM_TASKS_PER_NODE environment variable
in job step as needed by some MPI implementations. This was
just broken in SLURM while adding new environment variables
for BlueGene/Q sub-block jobs.