mlx_distributed_recv function

int mlx_distributed_recv(
  1. Pointer<mlx_array> res,
  2. Pointer<Int> shape,
  3. int shape_num,
  4. mlx_dtype_ dtype,
  5. int src,
  6. mlx_distributed_group group,
  7. mlx_stream s,
)

Implementation

int mlx_distributed_recv(
  ffi.Pointer<mlx_array> res,
  ffi.Pointer<ffi.Int> shape,
  int shape_num,
  mlx_dtype_ dtype,
  int src,
  mlx_distributed_group group,
  mlx_stream s,
) => _mlx_distributed_recv(res, shape, shape_num, dtype.value, src, group, s);