获取group信息
调用group管理接口获取group相关信息。
get_rank_size:获取当前group的所有昇腾AI处理器数量。
from npu_bridge.npu_init import * rankSize = get_rank_size(“myGroup”)
get_local_rank_size:获取当前昇腾AI处理器所在Server内属于group内的昇腾AI处理器数量。
from npu_bridge.npu_init import * lcoalRankSize = get_local_rank_size(“myGroup”)
get_rank_id:获取当前昇腾AI处理器在group内的逻辑设备号rank id。
from npu_bridge.npu_init import * rankId = get_rank_id(“myGroup”)
get_local_rank_id:获取当前昇腾AI处理器所在Server内的逻辑设备号local rank id。
from npu_bridge.npu_init import * localRankId = get_local_rank_id(“myGroup”)
父主题: 集合通信接口使用指导