ray.experimental.TensorTransportManager.get_communicator_metadata#

abstract TensorTransportManager.get_communicator_metadata(src_actor: ray.actor.ActorHandle, dst_actor: ray.actor.ActorHandle, backend: str | None = None) → CommunicatorMetadata[source]#

Get the communicator metadata (e.g. communicator name, src/dst rank) for the send/recv operation. This function is called on the owner process before it orchestrates the transfer.

Parameters:

  • src_actor – The actor that runs this function.

  • dst_actor – The actor that runs this function.

  • backend – The backend to use for the collective operation.

Returns:

The communicator metadata.

Return type:

CommunicatorMetadata