mlx_distributed_group_split function

  1. @Native<mlx_distributed_group Function(mlx_distributed_group, Int, Int)>(mlx_distributed_group, ffi.Int, ffi.Int)>()
mlx_distributed_group mlx_distributed_group_split(
  1. mlx_distributed_group group,
  2. int color,
  3. int key
)

Split the group.

Implementation

@ffi.Native<
  mlx_distributed_group Function(mlx_distributed_group, ffi.Int, ffi.Int)
>()
external mlx_distributed_group mlx_distributed_group_split(
  mlx_distributed_group group,
  int color,
  int key,
);