def get_local_rank_id(group="hccl_world_group")
获取device在group中对应的local rank序号。
参数名 |
输入/输出 |
描述 |
|---|---|---|
group |
输入 |
String类型，最大长度为128字节，含结束符。 group名称，可以为用户自定义group，如果用户不传，默认为"hccl_world_group"。 |
int类型，返回device所在服务器内的local rank id序号。
from npu_bridge.npu_init import * create_group(“myGroup”, 4, [0, 1, 2, 3]) localRankId = get_local_rank_id(“myGroup”) #rankId = 0