若API“是否支持”为“是”，“限制与说明”为“-”，说明此API和原生API支持度保持一致。
API名称 |
是否支持 |
限制与说明 |
|---|---|---|
torch.distributed.rpc.init_rpc |
是 |
NPU设备启用rpc时需要在init_rpc中进行特定的设置： backend绑定rpc.backend_registry.BackendType.NPU_TENSORPIPE； options绑定NPUTensorPipeRpcBackendOptions， 需要from torch_npu.distributed.rpc.options import NPUTensorPipeRpcBackendOptions并设置option选项，参数格式和原版TensorPipeRpcBackendOptions相同。 |
torch.distributed.rpc.rpc_sync |
是 |
- |
torch.distributed.rpc.rpc_async |
是 |
- |
torch.distributed.rpc.remote |
是 |
- |
torch.distributed.rpc.get_worker_info |
是 |
- |
torch.distributed.rpc.shutdown |
是 |
- |
torch.distributed.rpc.WorkerInfo |
是 |
- |
torch.distributed.rpc.functions.async_execution |
是 |
- |
torch.distributed.rpc.BackendType |
是 |
- |
torch.distributed.rpc.RpcBackendOptions |
是 |
- |
torch.distributed.rpc.TensorPipeRpcBackendOptions |
是 |
建议使用已适配的torch.distributed.rpc.NPUTensorPipeRpcBackendOptions |
torch.distributed.rpc.TensorPipeRpcBackendOptions.set_device_map |
是 |
建议使用已适配的torch.distributed.rpc.NPUTensorPipeRpcBackendOptions.set_device_map |
torch.distributed.rpc.TensorPipeRpcBackendOptions.set_devices |
是 |
建议使用已适配的torch.distributed.rpc.NPUTensorPipeRpcBackendOptions.set_devices |
torch.distributed.rpc.RRef |
是 |
- |
torch.distributed.rpc.PyRRef |
是 |
- |
torch.distributed.nn.api.remote_module.RemoteModule |
是 |
- |
torch.distributed.nn.api.remote_module.RemoteModule.get_module_rref |
是 |
- |
torch.distributed.nn.api.remote_module.RemoteModule.remote_parameters |
是 |
- |
torch.distributed.autograd.backward |
是 |
- |
torch.distributed.autograd.context |
是 |
- |
torch.distributed.autograd.get_gradients |
是 |
- |