From ff0551fef750d71aff2855d21a016b7bcad8ddb0 Mon Sep 17 00:00:00 2001 From: zhaozhenlong Date: Thu, 11 Mar 2021 20:11:30 +0800 Subject: [PATCH] make npu non 4d op error instead of output 0 --- mindspore/lite/src/runtime/agent/npu/npu_executor.cc | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/mindspore/lite/src/runtime/agent/npu/npu_executor.cc b/mindspore/lite/src/runtime/agent/npu/npu_executor.cc index ea079dc3ef..830e2b8db5 100644 --- a/mindspore/lite/src/runtime/agent/npu/npu_executor.cc +++ b/mindspore/lite/src/runtime/agent/npu/npu_executor.cc @@ -143,11 +143,11 @@ int NPUExecutor::Run(const std::vector &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; - } + } + 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;