def get_group_rank_from_world_rank(world_rank_id, group)
从world rank id，获取该进程在group中的group rank id。
参数名 |
输入/输出 |
描述 |
|---|---|---|
world_rank_id |
输入 |
int类型。 进程在"hccl_world_group"中的rank id。 |
group |
输入 |
String类型，最大长度为128字节，含结束符。 group名称，可以为用户自定义group或者"hccl_world_group"。 |
int类型，正常返回进程在group中的rank id。
from npu_bridge.npu_init import * create_group(“myGroup”, 4, [0, 1, 2, 3]) groupRankId = get_group_rank_from_world_rank (8, “myGroup”) #groupRankId = 1