mlx.core.distributed.init

目录

mlx.core.distributed.init#

init(strict: bool = False, backend: str = 'any') → Group#

初始化通信后端并创建全局通信组。

示例

import mlx.core as mx

group = mx.distributed.init(backend="ring")
参数:

  • strict (bool, 可选) – 如果设置为 False，则在 mx.distributed.is_available() 返回 False 时返回一个单例组，否则抛出运行时错误。 默认值: False

  • backend (str, 可选) – 要初始化的分布式后端。可能的值有 mpi, ring, any。如果设置为 any，则会尝试所有可用的后端，第一个成功的后端将成为全局组，并在后续调用中返回。 默认值: any

返回:

表示所有已启动进程的组。

返回类型:

Group

目录