FireRedASR2 supported axcl backend

sherpa-onnx/csrc/CMakeLists.txt

@@ -221,6 +221,7 @@ if(SHERPA_ONNX_ENABLE_AXCL)
     ./axcl/axcl-manager.cc
     ./axcl/axcl-model.cc
     ./axcl/offline-sense-voice-model-axcl.cc
+    ./axcl/offline-fire-red-asr-ctc-model-axcl.cc
     ./axcl/utils.cc
   )
 endif()

sherpa-onnx/csrc/axcl/axcl-model.cc

@@ -208,6 +208,31 @@ class AxclModel::Impl {
     return {};
   }
 
+  std::vector<uint8_t> GetOutputTensorDataRaw(const std::string &name) const {
+    for (size_t i = 0; i < output_tensor_names_.size(); ++i) {
+      if (output_tensor_names_[i] == name) {
+        size_t bytes = output_tensors_[i].Size();
+        std::vector<uint8_t> out(bytes);
+
+        auto ret = axclrtMemcpy(out.data(), output_tensors_[i].Get(), bytes,
+                                AXCL_MEMCPY_DEVICE_TO_HOST);
+        if (ret != 0) {
+          SHERPA_ONNX_LOGE(
+              "Failed to call axclrtMemcpy(). tensor name: '%s', return code: "
+              "%d",
+              name.c_str(), static_cast<int32_t>(ret));
+          return {};
+        }
+
+        return out;
+      }
+    }
+
+    SHERPA_ONNX_LOGE("Found no tensor with name: '%s'", name.c_str());
+
+    return {};
+  }
+
   bool Run() const {
     uint32_t group = 0;
     auto ret =
@@ -434,6 +459,11 @@ std::vector<float> AxclModel::GetOutputTensorData(
   return impl_->GetOutputTensorData(name);
 }
 
+std::vector<uint8_t> AxclModel::GetOutputTensorDataRaw(
+    const std::string &name) const {
+  return impl_->GetOutputTensorDataRaw(name);
+}
+
 bool AxclModel::Run() const { return impl_->Run(); }
 
 bool AxclModel::IsInitialized() const { return impl_->IsInitialized(); }

sherpa-onnx/csrc/axcl/axcl-model.h

@@ -36,6 +36,8 @@ class AxclModel {
 
   std::vector<float> GetOutputTensorData(const std::string &name) const;
 
+  std::vector<uint8_t> GetOutputTensorDataRaw(const std::string &name) const;
+
   bool Run() const;
   bool IsInitialized() const;