Searched defs:src_rank (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore-src/source/tests/st/nontask_sink/ |
| D | test_collective_scatter.py | 25 src_rank = 0 variable
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/ |
| D | hcom_send.cc | 58 …uint32_t src_rank = LongToUint(mindspore::parallel::ParallelContext::GetInstance()->global_rank())… in SendShapeForDynamic() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/communication/ |
| D | test_comm.py | 94 def __init__(self, input_channel, out_channel, src_rank): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pipeline_transformer/ |
| D | fold_pipeline_transformer.cc | 306 int64_t src_rank) { in ComputeRecvTag()
332 auto src_rank = global_rank_ - (user_node_stage - node_stage) * per_stage_rank_num_; in InsertReceive() local
|
| D | pipeline_transformer.cc | 1053 auto src_rank = global_rank_ - (user_node_stage - node_stage) * per_stage_rank_num_; in InsertReceive() local
1571 auto src_rank = *src_rank_ptr; in GenNewRecvFromOld() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/ |
| D | comm_ops.py | 1197 def __init__(self, src_rank=0, group=GlobalComm.WORLD_COMM_GROUP): argument
1514 def __init__(self, sr_tag, src_rank, shape, dtype, group=GlobalComm.WORLD_COMM_GROUP, argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/embedding_cache/ |
| D | embedding_cache_prefetch_actor.cc | 42 std::string GenerateInterProcessEdge(const std::string &src_role, uint32_t src_rank, const std::str… in GenerateInterProcessEdge()
|