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

!8894 Remove return node checking for SuccDeeperSimple() including function. 

From: @zh_qh
Reviewed-by: @stsuteng
Signed-off-by: @stsuteng
This commit is contained in:
mindspore-ci-bot 2020-11-23 14:05:07 +08:00 committed by Gitee
parent 2eee1c06f0 7bc0ac619b
commit f241fbcb8e
1 changed files with 6 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
mindspore/core/ir/graph_utils.cc
View File

@ -34,18 +34,17 @@
namespace mindspore {
std::vector<AnfNodePtr> TopoSort(const AnfNodePtr &root, const SuccFunc &succ, const IncludeFunc &include) {
std::vector<AnfNodePtr> res;
if (root == nullptr) {
return res;
}
size_t seen = NewSeenGeneration();
std::deque<AnfNodePtr> todo(1024);
std::vector<AnfNodePtr> res;
todo.clear();
todo.push_back(root);
while (!todo.empty()) {
AnfNodePtr node = todo.back();
if (node == nullptr) {
todo.pop_back();
continue;
}
if (node->extra_seen_ == seen) { // We use extra_seen_ as finish flag
todo.pop_back();
continue;
@ -65,10 +64,8 @@ std::vector<AnfNodePtr> TopoSort(const AnfNodePtr &root, const SuccFunc &succ, c
node->seen_ = seen;
if (incl == FOLLOW) {
auto succs = succ(node);
(void)std::copy_if(succs.begin(), succs.end(), std::back_inserter(todo), [seen](const AnfNodePtr &next) {
return next != nullptr && next->seen_ != seen &&
(next->func_graph() == nullptr || next->func_graph()->get_return() != next);
});
(void)std::copy_if(succs.begin(), succs.end(), std::back_inserter(todo),
[seen](const AnfNodePtr &next) { return next != nullptr && next->seen_ != seen; });
} else if (incl > EXCLUDE) { // Not NOFOLLOW or EXCLUDE
MS_LOG(EXCEPTION) << "The result of include(node) must be one of: \"follow\", \"nofollow\", \"exclude\"";
}