mlx.core.distributed.send

send(x: array, dst: int, *, group: Group | None = None, stream: None | Stream | Device = None) → array

Send an array from the current process to the process that has rank dst in the group.

参数:

• x (array) -- Input array.

• dst (int) -- Rank of the destination process in the group.

• group (Group) -- The group of processes that will participate in the send. If set to None the global group is used. Default: None.

• stream (Stream, optional) -- Stream or device. Defaults to None in which case the default stream of the default device is used.

返回:

An array identical to x which when evaluated the send is performed.

返回类型:

array