Searched refs:kAttrFusion (Results 1 – 12 of 12) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/ |
| D | adjust_depend_for_parallel_optimizer_recompute_all_gather.cc | 41 …bool is_fusion = AnfAlgo::HasNodeAttr(kAttrFusion, cnode) && AnfAlgo::GetNodeAttr<int64_t>(cnode, … in Run()
45 int64_t fusion_id = AnfAlgo::GetNodeAttr<int64_t>(cnode, kAttrFusion); in Run()
59 int64_t unrecompute_fusion_id = AnfAlgo::GetNodeAttr<int64_t>(cnode, kAttrFusion); in Run()
87 int64_t current_fusion_id = AnfAlgo::GetNodeAttr<int64_t>(adjust_node, kAttrFusion); in IncreaseAllgatherFusionId()
89 AnfAlgo::SetNodeAttr(kAttrFusion, MakeValue(destination_fusion_id), adjust_node); in IncreaseAllgatherFusionId()
92 int64_t current_fusion_id = AnfAlgo::GetNodeAttr<int64_t>(adjust_node, kAttrFusion); in IncreaseAllgatherFusionId()
94 AnfAlgo::SetNodeAttr(kAttrFusion, MakeValue(destination_fusion_id), adjust_node); in IncreaseAllgatherFusionId()
|
| D | communication_op_fusion.cc | 90 ValuePtr attr_fusion = primitive->GetAttr(kAttrFusion); in GetFusionGroupKey()
371 const std::vector<std::string> kHcclFusionAttrs = {kAttrFusion, kAttrGroup, kAttrGroupBack, in CreateFusedCommunicationOp()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/ |
| D | insert_depend_for_all_gather.cc | 35 …if (AnfAlgo::GetCNodeName(cnode) == kAllGatherOpName && AnfAlgo::HasNodeAttr(kAttrFusion, cnode) && in Run()
36 AnfAlgo::GetNodeAttr<int64_t>(cnode, kAttrFusion) > 0 && !is_recompute) { in Run()
37 all_gather_node[AnfAlgo::GetNodeAttr<int64_t>(cnode, kAttrFusion)] = node; in Run()
|
| D | split_inputs_for_reduce_scatter.cc | 75 AnfAlgo::SetNodeAttr(kAttrFusion, MakeValue(1L), reduce_scatter); in RearrangeInputsForReduceScatter()
93 AnfAlgo::SetNodeAttr(kAttrFusion, MakeValue(0L), node); in Process()
96 if (!AnfAlgo::HasNodeAttr(kAttrFusion, cnode) || !AnfAlgo::HasNodeAttr(kAttrRankSize, cnode)) { in Process()
99 auto fusion = AnfAlgo::GetNodeAttr<int64_t>(cnode, kAttrFusion); in Process()
|
| D | concat_outputs_for_all_gather.cc | 139 if (!AnfAlgo::HasNodeAttr(kAttrFusion, cnode) || !AnfAlgo::HasNodeAttr(kAttrRankSize, cnode)) { in Process()
142 auto fusion = AnfAlgo::GetNodeAttr<int64_t>(cnode, kAttrFusion); in Process()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/ |
| D | hcom_util.cc | 158 …if (AnfAlgo::HasNodeAttr(kAttrFusion, cnode) && AnfAlgo::GetNodeAttr<int64_t>(anf_node, kAttrFusio… in GetHcomCount()
165 …if (AnfAlgo::HasNodeAttr(kAttrFusion, cnode) && AnfAlgo::GetNodeAttr<int64_t>(anf_node, kAttrFusio… in GetHcomCount()
|
| D | hccl_kernel.cc | 172 if (AnfAlgo::HasNodeAttr(kAttrFusion, cnode)) { in GetOutputSizeList()
173 fusion = AnfAlgo::GetNodeAttr<int64_t>(cnode, kAttrFusion); in GetOutputSizeList()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | concat_outputs_for_all_gather.cc | 132 if (!AnfAlgo::HasNodeAttr(kAttrFusion, cnode) || !AnfAlgo::HasNodeAttr(kAttrRankSize, cnode)) { in Process()
135 auto fusion = AnfAlgo::GetNodeAttr<int64_t>(cnode, kAttrFusion); in Process()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | recompute.cc | 357 …int64_t fusion_id = prim->HasAttr(kAttrFusion) ? GetValue<int64_t>(prim->GetAttr(kAttrFusion)) : 0; in NewRecomputedNode()
361 new_prim->set_attr(kAttrFusion, MakeValue(fusion_id + fusion_id_increasement_size)); in NewRecomputedNode()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | bn_split.cc | 235 AnfAlgo::SetNodeAttr(kAttrFusion, MakeValue(opid), allreduce); in CreateAllReduceAndMul()
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | utils.h | 386 constexpr auto kAttrFusion = "fusion"; variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | anf_runtime_algorithm.cc | 1666 ValuePtr attr_fusion = primitive->GetAttr(kAttrFusion); in IsFusedCommunicationOp()
|