Searched refs:node_role (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/ps/core/ |
| D | abstract_node.h | 70 bool Broadcast(const NodeRole &node_role, const DataPtr &message, size_t size, int command,
92 …bool Send(const NodeRole &node_role, const uint32_t &rank_id, const DataPtr &data, size_t len, int…
94 …bool Send(const NodeRole &node_role, const std::vector<uint32_t> &rank_ids, const std::vector<Data…
96 …bool Send(const NodeRole &node_role, const uint32_t &rank_id, const DataPtr &message, size_t len, …
98 …bool Send(const NodeRole &node_role, const std::vector<uint32_t> &rank_ids, const std::vector<Data…
102 …uint64_t CollectiveSendAsync(const NodeRole &node_role, const uint32_t &rank_id, const void *data,…
103 …std::pair<uint32_t, uint64_t> CollectiveReceiveAsync(const NodeRole &node_role, const uint32_t &ra…
|
| D | abstract_node.cc | 65 bool AbstractNode::Broadcast(const NodeRole &node_role, const DataPtr &message, size_t size, int co… in Broadcast() argument
68 if (node_role != NodeRole::SERVER) { in Broadcast()
177 bool AbstractNode::Send(const NodeRole &node_role, const uint32_t &rank_id, const DataPtr &data, si… in Send() argument
184 if (!CommUtil::ValidateRankId(node_role, rank_id, worker_num_, server_num_)) { in Send()
200 bool AbstractNode::Send(const NodeRole &node_role, const std::vector<uint32_t> &rank_ids, in Send() argument
214 if (!CommUtil::ValidateRankId(node_role, rank_ids.at(it), worker_num_, server_num_)) { in Send()
240 bool AbstractNode::Send(const NodeRole &node_role, const uint32_t &rank_id, const DataPtr &message,… in Send() argument
248 if (!CommUtil::ValidateRankId(node_role, rank_id, worker_num_, server_num_)) { in Send()
280 bool AbstractNode::Send(const NodeRole &node_role, const std::vector<uint32_t> &rank_ids, in Send() argument
307 if (!CommUtil::ValidateRankId(node_role, rank_ids.at(it), worker_num_, server_num_)) { in Send()
[all …]
|
| D | comm_util.cc | 138 bool CommUtil::ValidateRankId(const enum NodeRole &node_role, const uint32_t &rank_id, const int32_… in ValidateRankId() argument
140 if (node_role == NodeRole::SERVER && (rank_id > IntToUint(total_server_num) - 1)) { in ValidateRankId()
142 } else if (node_role == NodeRole::WORKER && (rank_id > IntToUint(total_worker_num) - 1)) { in ValidateRankId()
|
| D | comm_util.h | 104 …static bool ValidateRankId(const enum NodeRole &node_role, const uint32_t &rank_id, const int32_t …
|