p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!53 [Auto parallel] Remove the redundant work in star elimination 

Merge pull request !53 from Xiaoda/modify_star_and_triangle_elimination
This commit is contained in:
mindspore-ci-bot 2020-04-01 11:43:31 +08:00 committed by Gitee
parent c46e267cfd c080ec7874
commit 569e5b056e
1 changed files with 5 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
mindspore/ccsrc/parallel/auto_parallel/graph_costmodel.cc
View File

@ -1210,19 +1210,16 @@ void CostGraph::CreateStarEliminationCostList(std::vector<std::shared_ptr<Edge>>
MS_EXCEPTION_IF_NULL(succ_node_cost);
for (auto& succ_edge_cost : succ_edge_clist) {
MS_EXCEPTION_IF_NULL(succ_edge_cost);
if ((succ_node_cost->memory_cost_ < DEVICE_MEMORY_CAPACITY) &&
(succ_edge_cost->memory_cost_ < DEVICE_MEMORY_CAPACITY)) {
succ_nodes_stras[k] = succ_node_stra;
succ_edges_costs[k] = succ_edge_cost;
succ_nodes_costs[k] = succ_node_cost;
recursive(k + 1);
}
succ_nodes_stras[k] = succ_node_stra;
succ_edges_costs[k] = succ_edge_cost;
succ_nodes_costs[k] = succ_node_cost;
recursive(k + 1);
}
}
}
};
recursive(0);
recursive(1);
}
std::vector<std::shared_ptr<Edge>> CostGraph::EliminationStar(const OperatorInfoPtr& merged_op) {