Distributed RPC Framework

若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.backend_registry.BackendType.TENSORPIPE

否

建议使用已适配的torch.distributed.rpc.backend_registry.BackendType.NPU_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.NPUTensorPipeRpcBackendOptions

torch.distributed.rpc.TensorPipeRpcBackendOptions.rpc_timeout

否

建议使用已适配的torch.distributed.rpc.NPUTensorPipeRpcBackendOptions

torch.distributed.rpc.NPUTensorPipeRpcBackendOptions

是

   

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.NPUTensorPipeRpcBackendOptions.set_device_map

是

   

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

      
父主题： PyTorch原生API