Signed-off-by: tianrengao <terrygao87@gmail.com>
torch.cuda.synchronize()
torch.accelerator.synchronize