Searched refs:get_local_rank_size (Results 1 – 4 of 4) sorted by relevance
21 get_local_rank, get_local_rank_size, destroy_group
296 size = hccl.get_local_rank_size()298 size = hccl.get_local_rank_size(group)
263 def get_local_rank_size(group=GlobalComm.WORLD_COMM_GROUP): function
190 def get_local_rank_size(group="hccl_world_group"): function