Searched refs:group_rank_id (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/communication/ |
| D | _comm_helper.py | 307 def _get_world_rank_from_group_rank_helper(group, group_rank_id, backend): argument
324 if not isinstance(group_rank_id, int):
325 raise TypeError("group_rank_id should be int, but got type {}".format(type(group_rank_id)))
329 world_rank_id = hccl.get_world_rank_from_group_rank(group, group_rank_id)
354 group_rank_id = None
360 group_rank_id = hccl.get_group_rank_from_world_rank(world_rank_id, group)
365 return group_rank_id
|
| D | management.py | 296 def get_world_rank_from_group_rank(group, group_rank_id): argument
332 …return _get_world_rank_from_group_rank_helper(group=group, group_rank_id=group_rank_id, backend=Gl…
|
| D | _hccl_management.py | 229 def get_world_rank_from_group_rank(group, group_rank_id): argument
240 check_rank_id(group_rank_id)
242 c_group_rank_id = ctypes.c_uint(group_rank_id)
|
| /third_party/mindspore/tests/ut/python/hccl_test/manage/ |
| D | api.py | 75 def get_world_rank_from_group_rank(group, group_rank_id): argument
76 return group_rank_id
|