def current_stream(): _lazy_init() return torch.cuda.Stream(_cdata=torch._C._cuda_getCurrentStream())