DeepSpeed

a691ec60 - Add tensor parallel inference unit tests (#2232)

Add tensor parallel inference unit tests (#2232) Co-authored-by: Reza Yazdani <44502768+RezaYazdaniAminabadi@users.noreply.github.com> Co-authored-by: Jeff Rasley <jerasley@microsoft.com> Co-authored-by: Sam Ade Jacobs <samjacobs@microsoft.com>