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,11 +143,11 @@ 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;
-      }
+    }
+    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;