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

!26744 Bugfix in ConvertNonscalarTensorToParameter 

Merge pull request !26744 from DeshiChen/1124_valuenode_parameter
This commit is contained in:
i-robot 2021-11-25 10:56:36 +00:00 committed by Gitee
parent 11768ec145 174e4ea3ce
commit 3c3d10fa31
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mindspore/ccsrc/backend/optimizer/graph_kernel/core/graph_builder.cc
View File

@ -18,6 +18,7 @@
#include <algorithm>
#include <memory>
#include <tuple>
#include <set>
#include <unordered_set>
#include <utility>
#include <vector>
@ -103,7 +104,7 @@ void EliminateMakeTuple(const FuncGraphPtr &fg) {
bool ConvertNonscalarTensorToParameter(const FuncGraphPtr &fg, AnfNodePtrList *inputs_ptr) {
auto cnodes = fg->GetOrderedCnodes();
AnfNodePtrList value_nodes;
std::set<AnfNodePtr> value_nodes;
for (const auto &cnode : cnodes) {
auto &inputs = cnode->inputs();
for (size_t i = 1; i < inputs.size(); ++i) {
@ -112,7 +113,7 @@ bool ConvertNonscalarTensorToParameter(const FuncGraphPtr &fg, AnfNodePtrList *i
if (tensor == nullptr || tensor->DataSize() == 1) {
continue;
}
value_nodes.push_back(tnode);
value_nodes.insert(tnode);
}
}
if (value_nodes.empty()) return false;