Searched refs:dest_rank (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/ |
| D | hcom_util.cc | 240 bool HcomUtil::GetHcomDestRank(const AnfNodePtr &anf_node, uint32_t *dest_rank) { in GetHcomDestRank() argument
242 MS_EXCEPTION_IF_NULL(dest_rank); in GetHcomDestRank()
246 *dest_rank = static_cast<uint32_t>(GetValue<int64_t>(primitive->GetAttr("dest_rank"))); in GetHcomDestRank()
|
| D | hcom_util.h | 71 static bool GetHcomDestRank(const AnfNodePtr &anf_node, uint32_t *dest_rank);
|
| /third_party/mindspore/tests/st/nccl/ |
| D | test_nccl_send_recv_op.py | 42 self.send = Send(sr_tag=0, dest_rank=rank+size//2, group=NCCL_WORLD_COMM_GROUP)
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/ |
| D | pipeline_transformer.cc | 557 auto dest_rank = global_rank_ + (user_node_stage - node_stage) * per_stage_rank_num_; in InsertSend() local
559 if (send_tag_map.find(dest_rank) != send_tag_map.end()) { in InsertSend()
560 send_tag = send_tag_map[dest_rank] + 1; in InsertSend()
561 send_tag_map[dest_rank] += 1; in InsertSend()
564 send_tag_map[dest_rank] = 0; in InsertSend()
|
| /third_party/mindspore/mindspore/boost/ |
| D | adasum.py | 223 send = Send(sr_tag=send_tag, dest_rank=dest_target, group="hccl_world_group")
234 send = Send(sr_tag=send_tag, dest_rank=dest_target, group="hccl_world_group")
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | _inner_ops.py | 421 …def __init__(self, sr_tag, dest_rank, group=GlobalComm.WORLD_COMM_GROUP, group_back=GlobalComm.WOR… argument
422 self.rank = dest_rank
|