Add SLURM_JOB_PARTITION to prolog env vars

Currently only set for batch jobs and prolog run prior to any
step allocations (for Cray CCM).
Set SLURM_JOB_PARTITION env var for Prolog

Partial response to bug 521