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

!26756 [AutoPar] [D-Rec] Adding cost graph edges for Rec search in order to support the initialization for Reshape 

Merge pull request !26756 from daniel_hg/Gather_version_update
This commit is contained in:
i-robot 2021-11-27 22:36:18 +00:00 committed by Gitee
parent d028302fa1 1bc9cad97d
commit 42d0054bcd
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mindspore/ccsrc/frontend/parallel/step_auto_parallel.cc
View File

@ -1088,6 +1088,7 @@ Status ParallelStrategyRecSearch(const std::vector<AnfNodePtr> &all_nodes, const
}
}
ReshapeCostCompute(all_nodes);
ConstructCostGraphEdges(all_nodes);
auto ops = entire_costgraph->GetOperators();
std::vector<std::vector<std::string>> input_tensor_names = entire_costgraph->get_inputs_tensor_name_list();