make npu non 4d op error instead of output 0

4 years ago · ff0551fef7
parent b6f8959c64
commit ff0551fef7
1 changed files with 5 additions and 5 deletions
--- a/mindspore/lite/src/runtime/agent/npu/npu_executor.cc
+++ b/mindspore/lite/src/runtime/agent/npu/npu_executor.cc
@ -143,13 +143,13 @@ int NPUExecutor::Run(const std::vector<Tensor *> &in_tensors, const std::vector<
        outputs_visited[index] = true;
        break;
      }
+    }
    if (index == out_tensors.size()) {
      MS_LOG(ERROR) << "Can't find corresponding ms lite tensor of " << i << " output tensor for npu executor "
                    << model_name_;
      return RET_ERROR;
    }
  }
-  }
  return RET_OK;
 }