Add C++ API for streaming zipformer ASR on RK NPU (#1908)

2025-02-24 19:07:37 +08:00
parent bafd1103d0
commit 4d79e6a007
73 changed files with 1909 additions and 120 deletions
--- a/sherpa-onnx/csrc/rknn/macros.h
+++ b/sherpa-onnx/csrc/rknn/macros.h
@@ -0,0 +1,19 @@
+// sherpa-onnx/csrc/macros.h
+//
+// Copyright      2025  Xiaomi Corporation
+
+#ifndef SHERPA_ONNX_CSRC_RKNN_MACROS_H_
+#define SHERPA_ONNX_CSRC_RKNN_MACROS_H_
+
+#include "sherpa-onnx/csrc/macros.h"
+
+#define SHERPA_ONNX_RKNN_CHECK(ret, msg, ...)      \
+  do {                                             \
+    if (ret != RKNN_SUCC) {                        \
+      SHERPA_ONNX_LOGE("Return code is: %d", ret); \
+      SHERPA_ONNX_LOGE(msg, ##__VA_ARGS__);        \
+      SHERPA_ONNX_EXIT(-1);                        \
+    }                                              \
+  } while (0)
+
+#endif  // SHERPA_ONNX_CSRC_RKNN_MACROS_H_