Home
last modified time | relevance | path

Searched refs:rank_size_t (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/
Dsplit_inputs_for_reduce_scatter.cc28 size_t rank_size_t = LongToSize(rank_size); in InsertSplitForInput() local
37 for (size_t j = 0; j < rank_size_t; j++) { in InsertSplitForInput()
39 output_node_shape[0] /= rank_size_t; in InsertSplitForInput()
65 size_t rank_size_t = LongToSize(rank_size); in RearrangeInputsForReduceScatter() local
66 for (size_t i = 0; i < rank_size_t; i++) { in RearrangeInputsForReduceScatter()
67 for (size_t j = 0, idx = i; j < inputs_size; j++, idx += rank_size_t) { in RearrangeInputsForReduceScatter()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Dcommunication_op_fusion.cc54 size_t rank_size_t = LongToSize(rank_size); in GenerateKernelBuildInfo() local
55 if (rank_size_t == 0) { in GenerateKernelBuildInfo()
64 for (size_t rank_index = 0; rank_index < rank_size_t; ++rank_index) { in GenerateKernelBuildInfo()
71 shape[0] /= rank_size_t; in GenerateKernelBuildInfo()
341 size_t rank_size_t = LongToSize(rank_size); in CreateFusedCommunicationOp() local
342 if (rank_size_t == 0) { in CreateFusedCommunicationOp()
345 size_t output_num = node_num * rank_size_t; in CreateFusedCommunicationOp()
349 for (size_t i = 0; i < rank_size_t; ++i) { in CreateFusedCommunicationOp()
355 shape[0] /= rank_size_t; in CreateFusedCommunicationOp()