若API未标明支持情况则代表该API暂无验证结果，待验证后更新表格。
API名称 |
是否支持 |
限制与说明 |
|---|---|---|
torch.distributed.optim.DistributedOptimizer |
||
torch.distributed.optim.DistributedOptimizer.step |
||
torch.distributed.optim.PostLocalSGDOptimizer |
是 |
|
torch.distributed.optim.PostLocalSGDOptimizer.load_state_dict |
是 |
|
torch.distributed.optim.PostLocalSGDOptimizer.state_dict |
是 |
|
torch.distributed.optim.PostLocalSGDOptimizer.step |
是 |
|
torch.distributed.optim.ZeroRedundancyOptimizer |
是 |
支持的输入类型为torch.nn.Optimizer对象 不支持NPU融合优化器对象 |
torch.distributed.optim.ZeroRedundancyOptimizer.add_param_group |
是 |
|
torch.distributed.optim.ZeroRedundancyOptimizer.consolidate_state_dict |
否 |
|
torch.distributed.optim.ZeroRedundancyOptimizer.join_hook |
否 |
|
torch.distributed.optim.ZeroRedundancyOptimizer.load_state_dict |
否 |
|
torch.distributed.optim.ZeroRedundancyOptimizer.state_dict |
是 |
|
torch.distributed.optim.ZeroRedundancyOptimizer.step |
否 |