pytorch

d03d43df - Adds torch.cuda.is_current_stream_capturing (#77673)

Adds torch.cuda.is_current_stream_capturing (#77673) Exposes a way to query if CUDA graph capture is underway on the current stream. Pull Request resolved: https://github.com/pytorch/pytorch/pull/77673 Approved by: https://github.com/ezyang