Searched refs:world_group (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | pipeline_split.cc | 31 std::string world_group; in GetWorldGroup() local
34 world_group = parallel::HCCL_WORLD_GROUP; in GetWorldGroup()
36 world_group = parallel::NCCL_WORLD_GROUP; in GetWorldGroup()
40 return world_group; in GetWorldGroup()
46 auto world_group = GetWorldGroup(); in GetRank() local
50 if (!CommManager::GetInstance().GetRankID(world_group, &rank_id)) { in GetRank()
86 auto world_group = GetWorldGroup(); in PipelineSplit() local
90 if (!CommManager::GetInstance().GetRankSize(world_group, &world_rank_size)) { in PipelineSplit()
|
| /third_party/boost/libs/mpi/example/ |
| D | hello_world_groups.cpp | 31 mpi::group world_group = world.group(); in main() local
32 mpi::group group_a = world_group.include(group_a_ranks,group_a_ranks+2); in main()
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | comm_manager.cc | 216 std::string world_group; in GetRank() local
219 world_group = parallel::HCCL_WORLD_GROUP; in GetRank()
221 world_group = parallel::NCCL_WORLD_GROUP; in GetRank()
242 if (!CommManager::GetInstance().GetRankID(world_group, &rank_id)) { in GetRank()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | device_manager.h | 87 std::string world_group() const { return gm_.world_group(); } in world_group() function
|
| D | group_manager.h | 64 std::string world_group() const { return world_group_; } in world_group() function
|
| D | step_parallel.h | 53 std::string world_group; member
|
| D | step_parallel.cc | 2884 std::string world_group; in GetCommInfo() local
2887 world_group = HCCL_WORLD_GROUP; in GetCommInfo()
2890 world_group = NCCL_WORLD_GROUP; in GetCommInfo()
2896 if (!CommManager::GetInstance().GetRankSize(world_group, &world_rank_size)) { in GetCommInfo()
2915 if (!CommManager::GetInstance().GetRankID(world_group, &rank_id)) { in GetCommInfo()
2921 CommInfo comm_info{device_num, global_rank, world_group, communication_backend}; in GetCommInfo()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/distribution/ |
| D | mpi_wrapper.cc | 132 NcclGroupInfo world_group = {rank_size_, rank_id_, unique_id, nullptr, world_group_ranks}; in Init() local
133 NCCLWrapper::instance().AddGroupInfo(NCCL_WORLD_GROUP, &world_group); in Init()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | session_basic.cc | 2734 std::string world_group; in GetRankId() local
2737 world_group = kHcclWorldGroup; in GetRankId()
2739 world_group = kNcclWorldGroup; in GetRankId()
2744 if (!CommManager::GetInstance().GetRankID(world_group, &rank_id)) { in GetRankId()
|