Move `torch.cuda` annotations inline (#40075)
Summary:
Also enable `torch.cuda` typechecking
Pull Request resolved: https://github.com/pytorch/pytorch/pull/40075
Differential Revision: D22121275
Pulled By: malfet
fbshipit-source-id: dbecef09911334e8f3d87f5ecab66349da9f2325
