Add type informations to torch.cuda (#47134)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/47133
Pull Request resolved: https://github.com/pytorch/pytorch/pull/47134
Reviewed By: smessmer
Differential Revision: D24955031
Pulled By: ezyang
fbshipit-source-id: 87f4623643715baa6ac0627383f009956f80cd46
