Add current stream source (#165211)
Summary:
X-link: https://github.com/pytorch/pytorch/pull/165211
Approved by: https://github.com/anijain2305
ghstack dependencies: #164819
Reviewed By: donigian
Differential Revision: D86056504
fbshipit-source-id: 84112edf57013357bb7f5fbec39f99b54678b278