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

Filter the func_graph node where the current node is located when make Interpret Node

This commit is contained in:
Margaret_wangrui 2021-11-30 10:24:40 +08:00
parent 7605e9f653
commit 74b80995eb
1 changed files with 14 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
mindspore/ccsrc/pipeline/jit/parse/parse.cc
View File

@ -1869,7 +1869,20 @@ AnfNodePtr Parser::HandleInterpret(const FunctionBlockPtr &block, const AnfNodeP
auto global_dict_node = NewValueNode(globals_converted_value);
// Prepare local parameters.
auto [keys, values] = block->local_py_params();
auto local_dict_node = ParseDictByKeysAndValues(block, keys, values);
// Filter the func_graph node where the current node is located.
auto current_fg = value_node->func_graph();
std::vector<AnfNodePtr> filter_keys;
std::vector<AnfNodePtr> filter_values;
for (size_t index = 0; index < values.size(); ++index) {
auto value = values[index];
auto fg = GetValueNode<FuncGraphPtr>(value);
if (fg == current_fg) {
continue;
}
(void)filter_keys.emplace_back(keys[index]);
(void)filter_values.emplace_back(value);
}
auto local_dict_node = ParseDictByKeysAndValues(block, filter_keys, filter_values);
// Update the valued node if it need interpreting.
constexpr int recursive_level = 2;
MS_LOG(INFO) << "[" << block->func_graph()->ToString() << "] script_text: `" << script_text