torch.distributed.checkpoint
若API“是否支持”为“是”，“限制与说明”为空，说明此API和原生API支持度保持一致。
API名称 |
是否支持 |
限制与说明 |
|---|---|---|
torch.distributed.checkpoint.load_state_dict |
是 |
|
torch.distributed.checkpoint.save_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.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 |
是 |