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

[ME] Do not associate code line information with nodes inserted by the compiler.

This commit is contained in:
Margaret_wangrui 2021-12-31 10:12:24 +08:00
parent 793b371d09
commit 0f6345e073
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mindspore/ccsrc/pipeline/jit/parse/function_block.cc
View File

@ -590,6 +590,9 @@ void FunctionBlock::AttachIsolatedNodesBeforeReturn() {
state = states[1]; state = states[1];
} else { } else {
state = func_graph_->NewCNode(std::move(states)); state = func_graph_->NewCNode(std::move(states));
if (state->debug_info()) {
state->debug_info()->set_location(nullptr);
}
} }
AnfNodePtr old_output = nullptr; AnfNodePtr old_output = nullptr;
@ -605,6 +608,12 @@ void FunctionBlock::AttachIsolatedNodesBeforeReturn() {
} }
AnfNodePtr stop_grad_node = func_graph_->NewCNode({NewValueNode(prim::kPrimStopGradient), state}); AnfNodePtr stop_grad_node = func_graph_->NewCNode({NewValueNode(prim::kPrimStopGradient), state});
CNodePtr depend_node = func_graph_->NewCNode({NewValueNode(prim::kPrimDepend), old_output, stop_grad_node}); CNodePtr depend_node = func_graph_->NewCNode({NewValueNode(prim::kPrimDepend), old_output, stop_grad_node});
if (stop_grad_node->debug_info()) {
stop_grad_node->debug_info()->set_location(nullptr);
}
if (depend_node->debug_info()) {
depend_node->debug_info()->set_location(nullptr);
}
// We add this attribute for @constexpr use scene, since we must infer them before other nodes. // We add this attribute for @constexpr use scene, since we must infer them before other nodes.
// That means isolated nodes will be evaluated first. It's not complete, but works in most scenes. // That means isolated nodes will be evaluated first. It's not complete, but works in most scenes.
depend_node->AddAttr(kAttrTopoSortRhsFirst, MakeValue(true)); depend_node->AddAttr(kAttrTopoSortRhsFirst, MakeValue(true));