torch.cuda.is_current_stream_capturing

torch.cuda.is_current_stream_capturing()

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.

Return type

bool