mlx.core.distributed.recv_like

目录

mlx.core.distributed.recv_like#

recv_like(x: array, src: int, *, group: Group | None = None, stream: None | Stream | Device = None) array#

从 rank 为 src 的进程接收一个形状和类型与 x 相似的数组。

这等价于调用 mx.distributed.recv(x.shape, x.dtype, src)

参数:
  • x (array) – 定义接收数组的形状和数据类型的数组。

  • src (int) – 组中源进程的 rank。

  • group (Group) – 参与接收的进程组。如果设置为 None,则使用全局组。默认值: None

  • stream (Stream, 可选) – 流或设备。默认为 None,在这种情况下使用默认设备的默认流。

返回值:

src 接收到的数组。

返回类型:

array