若API未标明支持情况，则代表该API的支持情况待验证。
API名称 |
是否支持 |
限制与说明 |
|---|---|---|
torch.distributed.optim.DistributedOptimizer |
||
torch.distributed.optim.DistributedOptimizer.step |
||
torch.distributed.optim.PostLocalSGDOptimizer |
||
torch.distributed.optim.PostLocalSGDOptimizer.step |
||
torch.distributed.optim.ZeroRedundancyOptimizer |
||
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 |