Home
last modified time | relevance | path

Searched refs:src_rank (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore/profiler/parser/
Dhccl_parser.py328 src_rank = item.get("args").get("src rank")
330 if src_rank is None or dst_rank is None:
335 if int(src_rank) == int('0xffffffff', 16):
336 src_rank = dst_rank
339 dst_rank = src_rank
343 item["args"]["dst rank"] = src_rank
344 src_dst_key = str(dst_rank) + '-' + str(src_rank)
346 src_dst_key = str(src_rank) + '-' + str(dst_rank)
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/
Dindex_add_gpu_kernel.h80 int64_t src_rank = src_shape.size(); in Init() local
83 axis += src_rank; in Init()
90 for (int64_t i = axis + 1; i >= 0 && i < src_rank; i++) { in Init()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/
Dhcom_util.cc226 bool HcomUtil::GetHcomSrcRank(const AnfNodePtr &anf_node, uint32_t *src_rank) { in GetHcomSrcRank() argument
228 MS_EXCEPTION_IF_NULL(src_rank); in GetHcomSrcRank()
232 *src_rank = static_cast<uint32_t>(GetValue<int64_t>(primitive->GetAttr("src_rank"))); in GetHcomSrcRank()
Dhcom_util.h70 static bool GetHcomSrcRank(const AnfNodePtr &anf_node, uint32_t *src_rank);
/third_party/mindspore/tests/st/nccl/
Dtest_nccl_send_recv_op.py51 … self.recv = Receive(sr_tag=0, src_rank=rank-size//2, shape=[3, 3, 3, 3], dtype=mstype.float32,
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dpipeline_transformer.cc624 auto src_rank = global_rank_ - (user_node_stage - node_stage) * per_stage_rank_num_; in InsertReceive() local
626 if (recv_tag_map.find(src_rank) != recv_tag_map.end()) { in InsertReceive()
627 recv_tag = recv_tag_map[src_rank] + 1; in InsertReceive()
628 recv_tag_map[src_rank] += 1; in InsertReceive()
631 recv_tag_map[src_rank] = 0; in InsertReceive()
/third_party/mindspore/mindspore/boost/
Dadasum.py226 recv = Receive(sr_tag=recv_tag, src_rank=dest_target, shape=shape, dtype=dtype,
237 recv = Receive(sr_tag=recv_tag, src_rank=dest_target, shape=shape, dtype=dtype,
/third_party/mindspore/mindspore/ops/operations/
D_inner_ops.py477 def __init__(self, sr_tag, src_rank, shape, dtype, group=GlobalComm.WORLD_COMM_GROUP, argument
479 self.rank = src_rank