torch::npu::synchronize(device_index)
npu设备同步接口,能与torch::cuda::synchronize()相同。
device_index:用来同步设备的index,默认-1,即同步当前设备。
无