DeepSpeed

b24dfa9d - Explictly set device when reusing dist env (#6696)

Explictly set device when reusing dist env (#6696) A rank of a process can change when reusing the environment. This PR explicitly sets the device when reusing the environment.