若API未标明支持情况则代表该API暂无验证结果,待验证后更新表格。
API名称
是否支持
限制与说明
torch.utils.checkpoint.checkpoint
是
torch.utils.checkpoint.checkpoint_sequential
支持fp32