Better code for handling memory required by a task on a heterogeneous

system.

What this patch does is looks at the partition the job is running in to get
the cpus per node off the first node in the partition.  Before it would
just look at the first node on the system which isn't that great on a
heterogeneous system.

Bug 3486