def test_register_rpc_backend_and_init_rpc_backend( self, mock_init_rref_context, mock_dist_autograd_init): backend_name = "stub_backend" rpc.register_backend(backend_name, stub_init_rpc_backend_handler) rpc.init_model_parallel(self_name="worker1", backend=backend_name, self_rank=1)
def test_register_rpc_backend_and_start_rpc_backend( self, mock_rpc_agent, mock_dist_autograd_init): backend_name = "stub_backend" rpc.register_backend(backend_name, stub_start_rpc_backend_handler) rpc.init_model_parallel( self_name="worker1", backend=backend_name, init_method=self.init_method, self_rank=self.rank, worker_name_to_id=self.worker_name_to_id, )