| /third_party/mindspore/tests/ut/cpp/device/ |
| D | hccl_adapter_test.cc | 95 uint32_t rank_size = 2; in TEST_F() local
118 uint32_t rank_size = 2; in TEST_F() local
141 uint32_t rank_size = 4; in TEST_F() local
165 uint32_t rank_size = 4; in TEST_F() local
189 uint32_t rank_size = 2; in TEST_F() local
212 uint32_t rank_size = 2; in TEST_F() local
235 uint32_t rank_size = 4; in TEST_F() local
259 uint32_t rank_size = 4; in TEST_F() local
283 uint32_t rank_size = 4; in TEST_F() local
304 uint32_t rank_size = 2; in TEST_F() local
[all …]
|
| /third_party/mindspore/tests/st/model_zoo_tests/wide_and_deep/python_file_for_ci/ |
| D | datasets.py | 34 line_per_sample=1000, rank_size=None, rank_id=None): argument
70 line_per_sample=1000, rank_size=None, rank_id=None): argument
111 data_type=DataType.TFRECORD, line_per_sample=1000, rank_size=None, rank_id=None): argument
|
| /third_party/mindspore/tests/ut/python/hccl_test/manage/ |
| D | api.py | 49 def rank_size(self): member in Hccl
53 def rank_size(self, size): member in Hccl
85 def create_group(group, rank_size, rank_ids): argument
|
| /third_party/mindspore/mindspore/ccsrc/fl/server/ |
| D | collective_ops_impl.cc | 40 uint32_t rank_size = server_num_; in RingAllReduce() local
142 uint32_t rank_size = server_num_; in ReduceBroadcastAllReduce() local
219 uint32_t rank_size = server_num_; in AllReduce() local
|
| /third_party/mindspore/tests/st/model_zoo_tests/DeepFM/src/ |
| D | dataset.py | 181 line_per_sample=1000, rank_size=None, rank_id=None): argument
222 line_per_sample=1000, rank_size=None, rank_id=None): argument
271 rank_size=None, rank_id=None): argument
|
| /third_party/mindspore/tests/ut/cpp/utils/ |
| D | comm_manager_test.cc | 32 unsigned int rank_size = 0; in TEST_F() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | all_to_all_unify_mindir.cc | 49 uint32_t rank_size; in GetRankSize() local
113 uint32_t rank_size = GetRankSize(group); in CreateAllToAllvNode() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | concat_outputs_for_all_gather.cc | 90 … const std::vector<AnfNodePtr> &new_tuple_getitems, int64_t rank_size) { in InsertConcatForOutput()
143 auto rank_size = AnfAlgo::GetNodeAttr<int64_t>(node, kAttrRankSize); in Process() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/ |
| D | concat_outputs_for_all_gather.cc | 92 … const std::vector<AnfNodePtr> &new_tuple_getitems, int64_t rank_size) { in InsertConcatForOutput()
150 auto rank_size = AnfAlgo::GetNodeAttr<int64_t>(node, kAttrRankSize); in Process() local
|
| D | split_inputs_for_reduce_scatter.cc | 108 auto rank_size = AnfAlgo::GetNodeAttr<int64_t>(node, kAttrRankSize); in Process() local
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | comm_manager.cc | 71 auto rank_size = rank_id_list.size(); in CreateGroupSync() local
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hccl_adapter/ |
| D | all_to_all_v_calc_param.cc | 37 AllToAllvCalcParam::AllToAllvCalcParam(const CNodeWeakPtr &cnode, uint32_t rank_size) in AllToAllvCalcParam()
|
| D | converter.cc | 131 uint32_t rank_size = 0; in SetAllToAllvAttr() local
|
| /third_party/mindspore/tests/st/tbe_networks/ |
| D | test_resnet_cifar_8p.py | 45 def create_dataset(repeat_num=1, training=True, batch_size=32, rank_id=0, rank_size=1, argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/ |
| D | communication_op_fusion.cc | 50 int64_t rank_size = 1; in GenerateKernelBuildInfo() local
337 int64_t rank_size = 1; in CreateFusedCommunicationOp() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/ |
| D | hccl_kernel.cc | 167 int64_t rank_size = 1; in GetOutputSizeList() local
|
| D | hcom_util.cc | 148 int64_t rank_size; in GetHcomCount() local
|
| /third_party/mindspore/mindspore/ccsrc/utils/context/ |
| D | context_extends.cc | 68 uint32_t rank_size = 1; in OpenTsd() local
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | group_manager.cc | 262 Status GroupManager::GetRankSize(const std::string &name, uint32_t *const rank_size) { in GetRankSize()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | prim_others.cc | 468 auto rank_size = GetValue<int>(primitive->GetAttr(kRankSize)); in InferImplAllGather() local
492 auto rank_size = GetValue<int>(primitive->GetAttr(kRankSize)); in InferImplReduceScatter() local
|
| /third_party/mindspore/mindspore/nn/layer/ |
| D | normalization.py | 179 def _check_rank_ids(self, process_groups, rank_size): argument
|
| /third_party/mindspore/mindspore/profiler/parser/ |
| D | integrator.py | 1055 def __init__(self, profiling_dir, device_id, rank_id, rank_size): argument
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_kernel_runtime.cc | 1085 auto rank_size = HcclCollectiveGroup::instance().GetRankSize(); in HcclInit() local
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | pipeline.cc | 1342 uint32_t rank_size = 0; in GetHcclRankSize() local
|