Searched refs:HCCL_WORLD_COMM_GROUP (Results 1 – 5 of 5) sorted by relevance
/third_party/mindspore/mindspore/communication/ |
D | _comm_helper.py | 52 HCCL_WORLD_COMM_GROUP = "hccl_world_group" variable 97 WORLD_COMM_GROUP = HCCL_WORLD_COMM_GROUP 175 group = HCCL_WORLD_COMM_GROUP 204 if group == HCCL_WORLD_COMM_GROUP: 234 if group == HCCL_WORLD_COMM_GROUP: 267 if group == HCCL_WORLD_COMM_GROUP: 295 if group == HCCL_WORLD_COMM_GROUP: 327 if group == HCCL_WORLD_COMM_GROUP: 358 if group == HCCL_WORLD_COMM_GROUP: 420 if group == HCCL_WORLD_COMM_GROUP:
|
D | management.py | 20 _create_group_helper, _destroy_group_helper, HCCL_WORLD_COMM_GROUP, NCCL_WORLD_COMM_GROUP, \ 30 DEFAULT_WORLD_COMM_GROUP = HCCL_WORLD_COMM_GROUP 137 GlobalComm.WORLD_COMM_GROUP = HCCL_WORLD_COMM_GROUP
|
D | __init__.py | 20 get_group_rank_from_world_rank, create_group, HCCL_WORLD_COMM_GROUP, NCCL_WORLD_COMM_GROUP, \
|
/third_party/mindspore/tests/ut/python/communication/ |
D | test_management_api.py | 70 D.get_world_rank_from_group_rank(D.HCCL_WORLD_COMM_GROUP, 0) 85 D.get_group_rank_from_world_rank(0, D.HCCL_WORLD_COMM_GROUP) 105 D.destroy_group(D.HCCL_WORLD_COMM_GROUP)
|
D | test_comm.py | 23 from mindspore.communication.management import HCCL_WORLD_COMM_GROUP, NCCL_WORLD_COMM_GROUP, Global… 81 self.allgather = AllGather(group=HCCL_WORLD_COMM_GROUP)
|