torch.accelerator.synchronize

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

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

Параметры:

device (torch.device, str, int, optional) – устройство, для которого нужно выполнить синхронизацию. Его тип должен соответствовать типу текущего устройства ускорителя. Если не указано, по умолчанию используется torch.accelerator.current_device_index().

Примечание

Эта функция ничего не делает, если текущий ускоритель не инициализирован.

Пример:

>>> assert torch.accelerator.is_available() "No available accelerators detected."
>>> start_event = torch.Event(enable_timing=True)
>>> end_event = torch.Event(enable_timing=True)
>>> start_event.record()
>>> tensor = torch.randn(100, device=torch.accelerator.current_accelerator())
>>> sum = torch.sum(tensor)
>>> end_event.record()
>>> torch.accelerator.synchronize()
>>> elapsed_time_ms = start_event.elapsed_time(end_event)