Signed-off-by: phaedonsun <phaedonsun@tencent.com> Co-authored-by: phaedonsun <phaedonsun@tencent.com>
torch.cuda.empty_cache
torch.accelerator.empty_cache
--format
torch.cuda.synchronize()
torch.accelerator.synchronize