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)