torch.cpu.synchronize

torch.cpu.synchronize(device=None)

等待CPU设备上所有流中的所有内核完成。

Parameters

设备 (torch.device 或 int, 可选) – 忽略，只有一个CPU设备。

注意：此函数仅用于促进与设备无关的代码。