Modify logic to compute the count of CPUs allocated to a job step as

used to set SLURM_CPUS_ON_NODE env var and compute memory limit.
We compare the allocated CPUs information passed in the job step 
credential (typically cores) and compare with the CPUs configured
on the node and scale the value as needed.