torch.cuda.synchronize()
torch.accelerator.synchronize
Signed-off-by: Kunshang Ji <kunshang.ji@intel.com>
Signed-off-by: Woosuk Kwon <woosuk@inferact.ai>