Signed-off-by: Hashem Hashemi <hashem.hashemi@amd.com>
torch.cuda.synchronize()
torch.accelerator.synchronize
mypy