torch.xpu.synchronize

torch.xpu.synchronize(device=None) [исходный код]

Дождаться завершения всех ядер во всех потоках на устройстве XPU.

Параметры:

device (torch.device или int, необязательно) – устройство, для которого выполняется синхронизация. Используется текущее устройство, задаваемое функцией current_device(), если device равно None (по умолчанию).