提交 fb00dae1 编写于 作者: X Xiaoda Zhang 提交者: 高东海

change star elimination: remove some redundant and checking works

上级 7f6cb987
...@@ -1210,19 +1210,16 @@ void CostGraph::CreateStarEliminationCostList(std::vector<std::shared_ptr<Edge>> ...@@ -1210,19 +1210,16 @@ void CostGraph::CreateStarEliminationCostList(std::vector<std::shared_ptr<Edge>>
MS_EXCEPTION_IF_NULL(succ_node_cost); MS_EXCEPTION_IF_NULL(succ_node_cost);
for (auto& succ_edge_cost : succ_edge_clist) { for (auto& succ_edge_cost : succ_edge_clist) {
MS_EXCEPTION_IF_NULL(succ_edge_cost); MS_EXCEPTION_IF_NULL(succ_edge_cost);
if ((succ_node_cost->memory_cost_ < DEVICE_MEMORY_CAPACITY) && succ_nodes_stras[k] = succ_node_stra;
(succ_edge_cost->memory_cost_ < DEVICE_MEMORY_CAPACITY)) { succ_edges_costs[k] = succ_edge_cost;
succ_nodes_stras[k] = succ_node_stra; succ_nodes_costs[k] = succ_node_cost;
succ_edges_costs[k] = succ_edge_cost; recursive(k + 1);
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) { std::vector<std::shared_ptr<Edge>> CostGraph::EliminationStar(const OperatorInfoPtr& merged_op) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册