mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!20171 fix inputs size error 

Merge pull request !20171 from huanghui/fix-export-paras-size-error
This commit is contained in:
i-robot 2021-07-14 01:59:34 +00:00 committed by Gitee
parent 20534c699c 3dde001d83
commit 85f773abbc
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mindspore/ccsrc/debug/trace.cc
View File

@ -70,7 +70,7 @@ std::string GetGraphParamString(const FuncGraphPtr &graph, abstract::AbstractBas
oss << "graph:" << graph->ToString() << " with args[";
auto params = graph->parameters();
if (params.size() < args_spec_list.size()) {
MS_EXCEPTION(ValueError) << "The size of parameters less than args_spec_list's size.";
MS_EXCEPTION(TypeError) << "The size of parameters less than args_spec_list's size.";
}
for (size_t i = 0; i < args_spec_list.size(); i++) {
auto parameter = params[i];
@ -105,6 +105,9 @@ void DumpInferStack(std::ostringstream &oss) {
continue;
}
auto args_spec_list = context->args_spec_list();
if (graph->parameters().size() < args_spec_list.size()) {
continue;
}
oss << " #" << index++ << " " << GetGraphParamString(graph, args_spec_list) << "\n";
}
}