Signed-off-by: tc-mb <caitianchi@modelbest.cn> Co-authored-by: hezhihui <hezhihui@modelbest.cn>
torch.cuda.empty_cache
torch.accelerator.empty_cache