若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.WorkerInfo.id |
||
torch.distributed.rpc.WorkerInfo.name |
||
torch.distributed.rpc.functions.async_execution |
是 |
|
torch.distributed.rpc.BackendType |
是 |
|
torch.distributed.rpc.RpcBackendOptions |
是 |
|
torch.distributed.rpc.backend_registry.BackendType.TENSORPIPE |
||
torch.distributed.rpc.backend_registry.BackendType.NPU_TENSORPIPE |
||
torch.distributed.rpc.TensorPipeRpcBackendOptions |
是 |
建议使用已适配的torch.distributed.rpc.NPUTensorPipeRpcBackendOptions |
torch.distributed.rpc.TensorPipeRpcBackendOptions.init_method |
||
torch.distributed.rpc.TensorPipeRpcBackendOptions.rpc_timeout |
||
torch.distributed.rpc.NPUTensorPipeRpcBackendOptions.device_maps |
||
torch.distributed.rpc.NPUTensorPipeRpcBackendOptions.devices |
||
torch.distributed.rpc.NPUTensorPipeRpcBackendOptions.init_method |
||
torch.distributed.rpc.NPUTensorPipeRpcBackendOptions.num_worker_threads |
||
torch.distributed.rpc.NPUTensorPipeRpcBackendOptions.rpc_timeout |
||
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 |
是 |