torch.cuda.synchronize()
torch.accelerator.synchronize
Signed-off-by: Kunshang Ji <kunshang.ji@intel.com>
Signed-off-by: Varun Sundar Rabindranath <vsundarr@redhat.com> Co-authored-by: Varun Sundar Rabindranath <vsundarr@redhat.com> Co-authored-by: Tyler Michael Smith <tyler@neuralmagic.com>