| /third_party/mindspore/tests/ut/cpp/utils/ |
| D | comm_manager_test.cc | 33 ASSERT_TRUE(CommManager::GetInstance().GetRankSize(string("1-2-3"), &rank_size)); in TEST_F()
34 ASSERT_TRUE(CommManager::GetInstance().GetRankSize(string("hccl_world_group"), &rank_size)); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/distribute/ |
| D | mpi_pycc.cc | 33 int MpiPycc::GetRankSize(const std::string &group) { return GetGroupSize(group); } in GetRankSize() function in mindspore::device::ascend::collective::MpiPycc
42 mpi_initializer.def("get_rank_size", &MpiPycc::GetRankSize, "get rank size"); in PYBIND11_MODULE()
|
| D | mpi_pycc.h | 35 static int GetRankSize(const std::string &group);
|
| D | ascend_collective.h | 51 int GetRankSize(const std::string &name = kHcclWorldGroup) const;
|
| D | ascend_collective.cc | 72 int HcclCollectiveGroup::GetRankSize(const std::string &name) const { in GetRankSize() function in mindspore::device::ascend::collective::HcclCollectiveGroup
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | comm_manager.cc | 104 bool CommManager::GetRankSize(const string &group, unsigned int *rank_size) const { in GetRankSize() function in mindspore::CommManager
110 *rank_size = static_cast<unsigned int>(HcclCollectiveGroup::instance().GetRankSize(group)); in GetRankSize()
164 bool CommManager::GetRankSize(const string &group, unsigned int *rank_size) const {
204 bool CommManager::GetRankSize(const string &group, unsigned int *rank_size) const {
|
| D | comm_manager.h | 37 bool GetRankSize(const string &group, unsigned int *rank_size) const;
|
| /third_party/mindspore/tests/ut/cpp/stub/hccl/ |
| D | collective_stub.cc | 27 int HcclCollectiveGroup::GetRankSize(const std::string &) const { return 0; } in GetRankSize() function in mindspore::device::ascend::collective::HcclCollectiveGroup
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | group_manager.cc | 167 (void)CommManager::GetInstance().GetRankSize(world_group_, &world_size); in CreateGroup()
262 Status GroupManager::GetRankSize(const std::string &name, uint32_t *const rank_size) { in GetRankSize() function in mindspore::parallel::GroupManager
268 bool ret = CommManager::GetInstance().GetRankSize(name, rank_size); in GetRankSize()
|
| D | group_manager.h | 62 Status GetRankSize(const std::string &name, uint32_t *rank_size);
|
| D | step_parallel.cc | 2896 if (!CommManager::GetInstance().GetRankSize(world_group, &world_rank_size)) { in GetCommInfo()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/mpi/ |
| D | mpi_adapter.h | 36 FUNC_EXPORT int GetRankSize() const { return rank_size_; } in GetRankSize() function
|
| D | mpi_export.cc | 34 return inst->GetRankSize(); in GetMPIRankSize()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | all_to_all_unify_mindir.cc | 48 uint32_t GetRankSize(const std::string &group) { in GetRankSize() function
113 uint32_t rank_size = GetRankSize(group); in CreateAllToAllvNode()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | pipeline_split.cc | 90 if (!CommManager::GetInstance().GetRankSize(world_group, &world_rank_size)) { in PipelineSplit()
|
| D | pipeline.cc | 1343 bool ret = CommManager::GetInstance().GetRankSize("", &rank_size); in GetHcclRankSize()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | special_op_eliminate.h | 655 if (!CommManager::GetInstance().GetRankSize(group, &num_of_devices)) { in operator()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_kernel_runtime.cc | 1085 auto rank_size = HcclCollectiveGroup::instance().GetRankSize(); in HcclInit()
|