Rate this Page

torch.cuda.is_current_stream_capturing#

torch.cuda.is_current_stream_capturing()[source]#

Return True if CUDA graph capture is underway on the current CUDA stream, False otherwise.

If a CUDA context does not exist on the current device, returns False without initializing the context.