pytorch

40be6b72 - [ez] Type function in distributed_c10d (#110435)

[ez] Type function in distributed_c10d (#110435) This function returns a `torch.device`. Pull Request resolved: https://github.com/pytorch/pytorch/pull/110435 Approved by: https://github.com/awgu