DeepSpeed

Fix partition id in the fp32->fp16 param copying step for z2+cpu-offload

#2059

Merged