diff --git a/sherpa-onnx/csrc/ascend/offline-paraformer-model-ascend.cc b/sherpa-onnx/csrc/ascend/offline-paraformer-model-ascend.cc index 8acd020d72..413b05382b 100644 --- a/sherpa-onnx/csrc/ascend/offline-paraformer-model-ascend.cc +++ b/sherpa-onnx/csrc/ascend/offline-paraformer-model-ascend.cc @@ -86,6 +86,9 @@ class OfflineParaformerModelAscend::Impl { // TODO(fangjun): Support multi clients std::lock_guard lock(mutex_); + aclError ret_set_ctx = aclrtSetCurrentContext(*context_); + SHERPA_ONNX_ASCEND_CHECK(ret_set_ctx, "Failed to call aclrtSetCurrentContext"); + features = ApplyLFR(std::move(features)); if (features.empty()) { return {}; diff --git a/sherpa-onnx/csrc/ascend/offline-sense-voice-model-ascend.cc b/sherpa-onnx/csrc/ascend/offline-sense-voice-model-ascend.cc index 7143d1ce29..ab356b9c0c 100644 --- a/sherpa-onnx/csrc/ascend/offline-sense-voice-model-ascend.cc +++ b/sherpa-onnx/csrc/ascend/offline-sense-voice-model-ascend.cc @@ -56,6 +56,9 @@ class OfflineSenseVoiceModelAscend::Impl { // TODO(fangjun): Support multi clients std::lock_guard lock(mutex_); + aclError ret_set_ctx = aclrtSetCurrentContext(*context_); + SHERPA_ONNX_ASCEND_CHECK(ret_set_ctx, "Failed to call aclrtSetCurrentContext"); + features = ApplyLFR(std::move(features)); if (features.empty()) { return {}; diff --git a/sherpa-onnx/csrc/ascend/offline-whisper-model-ascend.cc b/sherpa-onnx/csrc/ascend/offline-whisper-model-ascend.cc index 38e4ab9848..be69ae1da4 100644 --- a/sherpa-onnx/csrc/ascend/offline-whisper-model-ascend.cc +++ b/sherpa-onnx/csrc/ascend/offline-whisper-model-ascend.cc @@ -88,6 +88,9 @@ class OfflineWhisperModelAscend::Impl { // TODO(fangjun): Support multi clients std::lock_guard lock(mutex_); + aclError ret_set_ctx = aclrtSetCurrentContext(*context_); + SHERPA_ONNX_ASCEND_CHECK(ret_set_ctx, "Failed to call aclrtSetCurrentContext"); + OfflineWhisperDecoderResult r; if (features.empty()) { diff --git a/sherpa-onnx/csrc/ascend/offline-zipformer-ctc-model-ascend.cc b/sherpa-onnx/csrc/ascend/offline-zipformer-ctc-model-ascend.cc index 626b634a14..cd6d34d90f 100644 --- a/sherpa-onnx/csrc/ascend/offline-zipformer-ctc-model-ascend.cc +++ b/sherpa-onnx/csrc/ascend/offline-zipformer-ctc-model-ascend.cc @@ -51,6 +51,9 @@ class OfflineZipformerCtcModelAscend::Impl { // TODO(fangjun): Support multi clients std::lock_guard lock(mutex_); + aclError ret_set_ctx = aclrtSetCurrentContext(*context_); + SHERPA_ONNX_ASCEND_CHECK(ret_set_ctx, "Failed to call aclrtSetCurrentContext"); + int32_t num_frames = features.size() / feat_dim_; if (num_frames != max_num_frames_) {