Signed-off-by: jiang1.li <jiang1.li@intel.com>
torch.cuda.synchronize()
torch.accelerator.synchronize