torch.accelerator.synchronize

torch.accelerator.synchronize(device=None, /)

等待給定裝置上的所有流中的所有核心完成。

引數

device (torch.device, str, int, optional) – 要同步的裝置。它必須與當前的 accelerator 裝置型別匹配。如果未給出，則預設使用 torch.accelerator.current_device_index()。

注意

如果當前的 accelerator 未初始化，則此函式不執行任何操作。

示例

>>> 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)