昇腾社区首页
开发者
资源
支持
积分兑换 NEW

get_rank_size

功能描述

返回MPI通信器中的总进程数。

函数原型

1
def get_rank_size()

返回值说明

  • 成功：返回MPI通信器中的总进程数。
  • 失败：抛出异常。

使用示例

1
2
from rec_sdk_common.communication.hccl.hccl_info import get_rank_size
rank_size = get_rank_size()
父主题： 其他接口