mlx.core.distributed.Group

class Group

一个 mlx.core.distributed.Group 表示一组可以通信的独立 mlx 进程。

__init__(*args, **kwargs)

方法

__init__(*args, **kwargs)
rank(self)	获取此进程的 rank
size(self)	获取组的大小
split(self, color[, key])	根据提供的 color 将组拆分为子组。