Change dist to torch.distributed to fix a typo/bug in assert statements#638

jeffra merged 2 commits intomasterdeepspeedai/DeepSpeed:masterfrom

amawa/dist-fixdeepspeedai/DeepSpeed:amawa/dist-fixCopy head branch name to clipboard