若API未标明支持情况，则代表该API的支持情况待验证。
API名称 |
是否支持 |
限制与说明 |
|---|---|---|
torch.distributed.rpc.init_rpc |
否 |
|
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.TensorPipeRpcBackendOptions.set_device_map |
否 |
|
torch.distributed.rpc.TensorPipeRpcBackendOptions.set_devices |
否 |
|
torch.distributed.rpc.RRef |
否 |
|
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 |
否 |