Signed-off-by: Divakar Verma <divakar.verma@amd.com>
torch.cuda.empty_cache
torch.accelerator.empty_cache