oneflow.env

Environment

oneflow.env.get_world_size()

返回当前进程组包含的进程数量。

返回值：

进程组中包含的进程数量。

oneflow.env.get_rank()

返回当前进程组的 rank 值。

返回值：

进程组的 rank 。

oneflow.env.get_local_rank()

返回当前机器的本地 rank 。本地 rank 在全局上不是独一的，其仅对于一个机器的单个进程是独一的。

返回值：

当前机器进程的本地 rank 。

oneflow.env.get_node_size()

返回当前进程组包含的机器数量。

返回值：

进程组中包含的机器数量。