torch.xpu.is_current_stream_capturing

torch.xpu.is_current_stream_capturing() [source]

Возвращает True, если в данный момент выполняется захват графа XPU в текущем потоке XPU, и False в противном случае.

Если контекст XPU на текущем устройстве отсутствует, возвращает False, не инициализируя контекст.