Searched refs:get_group_rank_from_world_rank (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/tests/ut/python/communication/ |
| D | test_management_api.py | 85 D.get_group_rank_from_world_rank(0, D.HCCL_WORLD_COMM_GROUP)
90 D.get_group_rank_from_world_rank('0', '0-1')
95 D.get_group_rank_from_world_rank(0, '0-1')
139 assert D.get_group_rank_from_world_rank(0, '0-1') == 0
|
| /third_party/mindspore/mindspore/communication/ |
| D | __init__.py | 20 get_group_rank_from_world_rank, create_group, HCCL_WORLD_COMM_GROUP, NCCL_WORLD_COMM_GROUP, \
|
| D | management.py | 335 def get_group_rank_from_world_rank(world_rank_id, group): function
|
| D | _hccl_management.py | 251 def get_group_rank_from_world_rank(world_rank_id, group): function
|
| D | _comm_helper.py | 360 group_rank_id = hccl.get_group_rank_from_world_rank(world_rank_id, group)
|
| /third_party/mindspore/tests/ut/python/hccl_test/manage/ |
| D | api.py | 80 def get_group_rank_from_world_rank(world_rank_id, group): function
|