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

!19226 fix the multi outputs of value node 

Merge pull request !19226 from limingqi107/bug_fix
This commit is contained in:
i-robot 2021-07-02 01:09:55 +00:00 committed by Gitee
parent ea82e018e8 279091fa75
commit d251d09034
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mindspore/ccsrc/backend/session/anf_runtime_algorithm.cc
View File

@ -323,13 +323,17 @@ std::vector<KernelWithIndex> AnfRuntimeAlgorithm::GetAllOutputWithIndex(const An
const std::vector<PrimitivePtr> return_types = {prim::kPrimDepend, prim::kPrimMakeTuple};
size_t outputs_num = 1;
if (IsRealCNodeKernel(node)) {
// Value node may be tuple which has multi outputs.
if (IsRealCNodeKernel(node) || node->isa<ValueNode>()) {
outputs_num = AnfAlgo::GetOutputTensorNum(node);
}
// The output may be the tuple of node, so need visit all the outputs of node.
for (size_t i = 0; i < outputs_num; ++i) {
const auto &output_with_index = AnfAlgo::VisitKernelWithReturnType(node, i, false, return_types);
auto output_with_index = AnfAlgo::VisitKernelWithReturnType(node, i, false, return_types);
MS_EXCEPTION_IF_NULL(output_with_index.first);
if (node->isa<ValueNode>()) {
output_with_index.second = i;
}
// The depend and makeTuple node need recurse.
if (AnfAlgo::CheckPrimitiveType(output_with_index.first, prim::kPrimDepend) ||