torch.distributed.checkpoint
若API“是否支持”为“是”，“限制与说明”为空，说明此API和原生API支持度保持一致。
API名称 |
是否支持 |
限制与说明 |
|---|---|---|
torch.distributed.checkpoint.state_dict_saver.save_state_dict |
是 |
|
torch.distributed.checkpoint.state_dict_loader.load_state_dict |
是 |
|
torch.distributed.checkpoint.stateful.Stateful |
是 |
|
torch.distributed.checkpoint.stateful.Stateful.load_state_dict |
是 |
|
torch.distributed.checkpoint.stateful.Stateful.state_dict |
是 |
|
torch.distributed.checkpoint.StorageReader |
否 |
|
torch.distributed.checkpoint.StorageReader.prepare_global_plan |
否 |
|
torch.distributed.checkpoint.StorageReader.prepare_local_plan |
否 |
|
torch.distributed.checkpoint.StorageReader.read_data |
否 |
|
torch.distributed.checkpoint.StorageReader.read_metadata |
否 |
|
torch.distributed.checkpoint.StorageReader.set_up_storage_reader |
否 |
|
torch.distributed.checkpoint.StorageWriter |
否 |
|
torch.distributed.checkpoint.StorageWriter.finish |
否 |
|
torch.distributed.checkpoint.StorageWriter.prepare_global_plan |
否 |
|
torch.distributed.checkpoint.StorageWriter.prepare_local_plan |
否 |
|
torch.distributed.checkpoint.StorageWriter.set_up_storage_writer |
否 |
|
torch.distributed.checkpoint.StorageWriter.write_data |
否 |
|
torch.distributed.checkpoint.LoadPlanner |
否 |
|
torch.distributed.checkpoint.LoadPlanner.commit_tensor |
否 |
|
torch.distributed.checkpoint.LoadPlanner.create_global_plan |
否 |
|
torch.distributed.checkpoint.LoadPlanner.create_local_plan |
否 |
|
torch.distributed.checkpoint.LoadPlanner.finish_plan |
否 |
|
torch.distributed.checkpoint.LoadPlanner.load_bytes |
否 |
|
torch.distributed.checkpoint.LoadPlanner.resolve_tensor |
否 |
|
torch.distributed.checkpoint.LoadPlanner.set_up_planner |
否 |
|
torch.distributed.checkpoint.LoadPlan |
是 |
|
torch.distributed.checkpoint.ReadItem |
是 |
|
torch.distributed.checkpoint.SavePlanner |
否 |
|
torch.distributed.checkpoint.SavePlanner.create_global_plan |
否 |
|
torch.distributed.checkpoint.SavePlanner.create_local_plan |
否 |
|
torch.distributed.checkpoint.SavePlanner.finish_plan |
否 |
|
torch.distributed.checkpoint.SavePlanner.resolve_data |
否 |
|
torch.distributed.checkpoint.SavePlanner.set_up_planner |
否 |
|
torch.distributed.checkpoint.SavePlan |
是 |
|
torch.distributed.checkpoint.planner.WriteItem |
是 |
|
torch.distributed.checkpoint.FileSystemReader |
否 |
|
torch.distributed.checkpoint.FileSystemWriter |
否 |
|
torch.distributed.checkpoint.DefaultSavePlanner |
否 |
|
torch.distributed.checkpoint.DefaultSavePlanner.lookup_object |
否 |
|
torch.distributed.checkpoint.DefaultSavePlanner.transform_object |
否 |
|
torch.distributed.checkpoint.DefaultLoadPlanner |
否 |
|
torch.distributed.checkpoint.DefaultLoadPlanner.lookup_tensor |
否 |
|
torch.distributed.checkpoint.DefaultLoadPlanner.transform_tensor |
否 |
|
torch.distributed.checkpoint.state_dict.get_state_dict |
是 |
|
torch.distributed.checkpoint.state_dict.get_model_state_dict |
是 |
|
torch.distributed.checkpoint.state_dict.get_optimizer_state_dict |
是 |
|
torch.distributed.checkpoint.state_dict.set_state_dict |
是 |
|
torch.distributed.checkpoint.state_dict.set_model_state_dict |
是 |
|
torch.distributed.checkpoint.state_dict.set_optimizer_state_dict |
是 |
|
torch.distributed.checkpoint.state_dict.StateDictOptions |
是 |