get_rank_size
功能描述
返回MPI通信器中的总进程数。
函数原型
def get_rank_size()
返回值说明
- 成功:返回MPI通信器中的总进程数。
- 失败:抛出异常。
使用示例
from mx_rec.util.communication.hccl_ops import get_rank_size rank_size = get_rank_size()
父主题: 其他接口
返回MPI通信器中的总进程数。
def get_rank_size()
from mx_rec.util.communication.hccl_ops import get_rank_size rank_size = get_rank_size()