Function send_request_to_machine

python/dgl/distributed/rpc.py:680–711

Send one request to the target machine, which will randomly select a server node to process this request. The operation is non-blocking -- it does not guarantee the payloads have reached the target or even have left the sender process. However, all the payloads (i.e., data and array

(target, request)

def send_request_to_machine(target, request):
    """Send one request to the target machine, which will randomly
    select a server node to process this request.

    The operation is non-blocking -- it does not guarantee the payloads have
    reached the target or even have left the sender process. However,
    all the payloads (i.e., data and arrays) can be safely freed after this
    function returns.

    Parameters
    ----------
    target : int
        ID of target machine.
    request : Request
        The request to send.

    Raises
    ------
    ConnectionError if there is any problem with the connection.
    """
    service_id = request.service_id
    msg_seq = incr_msg_seq()
    client_id = get_rank()
    server_id = random.randint(
        target * get_num_server_per_machine(),
        (target + 1) * get_num_server_per_machine() - 1,
    )
    data, tensors = serialize_to_payload(request)
    msg = RPCMessage(
        service_id, msg_seq, client_id, server_id, data, tensors, get_group_id()
    )
    send_rpc_message(msg, server_id)


def send_response(target, response, group_id):