Browse Source

check invalid internal output

tags/v0.6.0-beta
kswang 5 years ago
parent
commit
11f78fb8ad
1 changed files with 10 additions and 0 deletions
  1. +10
    -0
      mindspore/ccsrc/backend/session/session_basic.cc

+ 10
- 0
mindspore/ccsrc/backend/session/session_basic.cc View File

@@ -978,6 +978,16 @@ CNodePtr SessionBasic::ConstructOutput(const AnfNodePtrList &outputs, const std:
bool internal_output = true;
std::string kernel_target = GetCNodeTarget(front_real_kernel.first);
for (auto user : users) {
auto cnode = user.first->cast<CNodePtr>();
if (cnode == nullptr) {
internal_output = false;
break;
}
auto prim = cnode->input(kAnfPrimitiveIndex);
if (prim == nullptr || !prim->isa<ValueNode>()) {
internal_output = false;
break;
}
if (!AnfAlgo::IsRealKernel(user.first) || kernel_target != GetCNodeTarget(user.first)) {
internal_output = false;
break;


Loading…
Cancel
Save