Handle returning the current stream with/without device index (#165356)
Summary:
X-link: https://github.com/pytorch/pytorch/pull/165356
Approved by: https://github.com/anijain2305
ghstack dependencies: #164819, #165211, #165212
Reviewed By: donigian
Differential Revision: D86056516
fbshipit-source-id: 8509369a395b4489ae1c11919c38edd85073ab36