Signed-off-by: Matthias Gehre <matthias.gehre@amd.com>
torch.cuda.synchronize()
torch.accelerator.synchronize