DeepSpeed

195563a2 - fix a small type error on bf16+pp (#3441)

fix a small type error on bf16+pp (#3441) Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com> Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com>