DeepSpeed

05cb79db - _exec_forward_pass: place zeros(1) on the same device as the param (#5576)

_exec_forward_pass: place zeros(1) on the same device as the param (#5576) avoid mismatch between the param storage device and the .data device Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com>