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

!47117 dependency for independent node 

Merge pull request !47117 from kisnwang/refactor-kernel-graph-execorder-generator
This commit is contained in:
i-robot 2022-12-23 02:48:20 +00:00 committed by Gitee
parent 9f291738fa ff5dc9963d
commit fcce7943b5
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mindspore/ccsrc/backend/common/session/exec_order_builder.cc
View File

@ -148,7 +148,7 @@ void ExecOrderBuilder::FindIndependentNodes() {
MS_EXCEPTION_IF_NULL(cnode);
bool independent = true;
auto &inputs = cnode->inputs();
for (auto iter = inputs.begin(); iter != inputs.end(); ++iter) {
for (auto iter = inputs.rbegin(); iter != inputs.rend(); ++iter) {
auto &input = *iter;
MS_EXCEPTION_IF_NULL(input);
if (IsTrivialNode(input)) {
@ -164,6 +164,13 @@ void ExecOrderBuilder::FindIndependentNodes() {
}
if (AnfUtils::IsRealKernel(node)) {
to_visit.push(input);
if (!independent_nodes_.empty()) {
auto inode = independent_nodes_.top();
node_output_edges_[input].emplace_back(inode);
node_input_edges_[inode].emplace_back(input);
node_input_num_[inode] += 1;
independent_nodes_.pop();
}
} else {
vnode_to_visit.push(input);
}