Searched refs:world_rank_id (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/communication/ |
| D | _comm_helper.py | 323 world_rank_id = None
329 world_rank_id = hccl.get_world_rank_from_group_rank(group, group_rank_id)
334 return world_rank_id
338 def _get_group_rank_from_world_rank_helper(world_rank_id, group, backend): argument
355 if not isinstance(world_rank_id, int):
356 raise TypeError("world_rank_id should be int, but got type {}".format(type(world_rank_id)))
360 group_rank_id = hccl.get_group_rank_from_world_rank(world_rank_id, group)
|
| D | management.py | 335 def get_group_rank_from_world_rank(world_rank_id, group): argument
371 …return _get_group_rank_from_world_rank_helper(world_rank_id=world_rank_id, group=group, backend=Gl…
|
| D | _hccl_management.py | 251 def get_group_rank_from_world_rank(world_rank_id, group): argument
262 check_rank_id(world_rank_id)
264 c_world_rank_id = ctypes.c_uint(world_rank_id)
|
| /third_party/mindspore/tests/ut/python/hccl_test/manage/ |
| D | api.py | 80 def get_group_rank_from_world_rank(world_rank_id, group): argument
81 return world_rank_id
|