torch.cuda.synchronize()
torch.accelerator.synchronize
Signed-off-by: Kunshang Ji <kunshang.ji@intel.com>
Signed-off-by: mgoin <mgoin64@gmail.com> Co-authored-by: Robert Shaw <114415538+robertgshaw2-redhat@users.noreply.github.com>