Distributed Group#

typedef struct mlx_distributed_group_ mlx_distributed_group#

A MLX distributed group object.

int mlx_distributed_group_rank(mlx_distributed_group group)#

Get the rank.

int mlx_distributed_group_size(mlx_distributed_group group)#

Get the group size.

mlx_distributed_group mlx_distributed_group_split(mlx_distributed_group group, int color, int key)#

Split the group.

bool mlx_distributed_is_available()#

Check if distributed is available.

mlx_distributed_group mlx_distributed_init(bool strict)#

Initialize distributed.

struct mlx_distributed_group_#
#include <distributed_group.h>

A MLX distributed group object.