| /third_party/mindspore/config/ |
| D | hccl_multi_machine_multi_rank.json | 16 "rank_id": "0", string
24 "rank_id": "1", string
32 "rank_id": "2", string
40 "rank_id": "3", string
48 "rank_id": "4", string
56 "rank_id": "5", string
64 "rank_id": "6", string
72 "rank_id": "7", string
80 "rank_id": "8", string
88 "rank_id": "9", string
[all …]
|
| D | hccl_single_machine_multi_rank.json | 16 "rank_id": "0", string
24 "rank_id": "1", string
32 "rank_id": "2", string
40 "rank_id": "3", string
48 "rank_id": "4", string
56 "rank_id": "5", string
64 "rank_id": "6", string
72 "rank_id": "7", string
104 "rank_id": "the first device must be 0 and then increase in order", string
|
| D | hccl_evb_multi_rank.json | 20 "rank_id": "0", string
30 "rank_id": "1", string
|
| /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 | 41 def rank_id(self): member in Hccl
45 def rank_id(self, rank_id): argument
|
| /third_party/mindspore/tests/ut/data/dump/gpu_dumps/golden/ |
| D | read_tensors_nonexist_node_expected.json | 8 "rank_id": 0, number
26 "rank_id": 0, number
44 "rank_id": 0, number
|
| D | watchpoints_expected.json | 20 "rank_id": 0, number
42 "rank_id": 0, number
73 "rank_id": 0, number
|
| D | read_tensors_base_stat_expected.json | 8 "rank_id": 0, number
47 "rank_id": 0, number
86 "rank_id": 0, number
123 "rank_id": 0, number
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | pipeline_split.cc | 48 uint32_t rank_id = 0; in GetRank() local
58 static int64_t InferStage(int64_t rank_id, int64_t stage_num, int64_t device_num) { in InferStage()
|
| /third_party/mindspore/tests/st/tbe_networks/ |
| D | hccl_2p_server_template.json | 20 "rank_id": "0", string
30 "rank_id": "1", string
|
| D | hccl_2p_evb_template.json | 20 "rank_id": "0", string
30 "rank_id": "1", string
|
| /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/mindspore/ccsrc/backend/kernel_compiler/cpu/ps/ |
| D | apply_momentum_ps_kernel.h | 30 ApplyMomentumPSKernel(size_t rank_id, size_t pserver_num, size_t worker_num) in ApplyMomentumPSKernel()
|
| D | sparse_apply_adam_ps_kernel.h | 31 SparseApplyAdamPSKernel(size_t rank_id, size_t pserver_num, size_t worker_num) in SparseApplyAdamPSKernel()
|
| D | embedding_look_up_ps_kernel.h | 30 EmbeddingLookUpPSKernel(size_t rank_id, size_t pserver_num, size_t worker_num) in EmbeddingLookUpPSKernel()
|
| D | sparse_apply_lazy_adam_ps_kernel.h | 30 SparseApplyLazyAdamPSKernel(size_t rank_id, size_t pserver_num, size_t worker_num) in SparseApplyLazyAdamPSKernel()
|
| D | sparse_apply_ftrl_ps_kernel.h | 31 SparseApplyFtrlPSKernel(size_t rank_id, size_t pserver_num, size_t worker_num) in SparseApplyFtrlPSKernel()
|
| D | pserver_kernel.h | 30 PServerKernel(size_t rank_id, size_t pserver_num, size_t worker_num) in PServerKernel()
|
| /third_party/mindspore/tests/st/broadcast/ |
| D | lenet_broadcast_auto_parallel.py | 32 rank_id = 0 variable
|
| /third_party/mindspore/mindspore/ccsrc/profiler/device/cpu/ |
| D | cpu_data_saver.cc | 40 auto rank_id = common::GetEnv("RANK_ID"); in WriteFile() local
|
| /third_party/mindspore/mindspore/ccsrc/ps/core/protos/ |
| D | comm.proto | 58 uint32 rank_id = 4; field
116 uint32 rank_id = 1; field
133 uint32 rank_id = 5; field
|
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | optimizer_info.h | 36 size_t rank_id) {} in ComputeMean()
|
| /third_party/mindspore/mindspore/ccsrc/debug/debugger/offline_debug/ |
| D | dbg_services.cc | 84 std::vector<std::uint32_t> rank_id; in AddWatchpoint() local
127 std::vector<std::uint32_t> rank_id; in AddWatchpoint() local
175 std::vector<unsigned int> rank_id; in CheckWatchpoints() local
237 std::vector<unsigned int> rank_id; in ReadTensorsUtil() local
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hccl_adapter/ |
| D | hccl_adapter.cc | 43 …tic std::map<std::string, std::string> GenHcclOptions(uint32_t device_id, std::string_view rank_id, in GenHcclOptions()
150 bool HcclAdapter::InitHccl(uint32_t device_id, std::string_view rank_id, std::string_view rank_file, in InitHccl()
319 bool HcclAdapter::InitKernelInfoStore(uint32_t device_id, std::string_view rank_id, std::string_vie… in InitKernelInfoStore()
389 bool HcclAdapter::InitHcclComm(std::string_view rank_id, std::string_view rank_file) { in InitHcclComm()
|
| /third_party/mindspore/mindspore/ccsrc/ps/core/ |
| D | abstract_node.cc | 177 bool AbstractNode::Send(const NodeRole &node_role, const uint32_t &rank_id, const DataPtr &data, si… in Send()
240 bool AbstractNode::Send(const NodeRole &node_role, const uint32_t &rank_id, const DataPtr &message,… in Send()
334 uint64_t AbstractNode::CollectiveSendAsync(const NodeRole &node_role, const uint32_t &rank_id, cons… in CollectiveSendAsync()
353 … uint64_t> AbstractNode::CollectiveReceiveAsync(const NodeRole &node_role, const uint32_t &rank_id, in CollectiveReceiveAsync()
858 const std::shared_ptr<TcpClient> &AbstractNode::GetOrCreateTcpClient(const uint32_t &rank_id) { in GetOrCreateTcpClient()
981 uint32_t rank_id = meta->rank_id(); in RunReceiveCallback() local
1009 uint64_t AbstractNode::NextExpectedRankRequestId(const uint32_t &rank_id) { in NextExpectedRankRequestId()
1021 uint64_t AbstractNode::NextActualRankRequestId(const uint32_t &rank_id) { in NextActualRankRequestId()
|