Support CoreML for macOS (#151)

2023-05-12 15:57:44 +08:00
parent de1880948b
commit cea718e3d8
22 changed files with 216 additions and 87 deletions
--- a/sherpa-onnx/csrc/offline-nemo-enc-dec-ctc-model.cc
+++ b/sherpa-onnx/csrc/offline-nemo-enc-dec-ctc-model.cc
@@ -6,6 +6,7 @@

 #include "sherpa-onnx/csrc/macros.h"
 #include "sherpa-onnx/csrc/onnx-utils.h"
+#include "sherpa-onnx/csrc/session.h"
 #include "sherpa-onnx/csrc/text-utils.h"
 #include "sherpa-onnx/csrc/transpose.h"

@@ -16,11 +17,8 @@ class OfflineNemoEncDecCtcModel::Impl {
  explicit Impl(const OfflineModelConfig &config)
      : config_(config),
        env_(ORT_LOGGING_LEVEL_ERROR),
-        sess_opts_{},
+        sess_opts_(GetSessionOptions(config)),
        allocator_{} {
-    sess_opts_.SetIntraOpNumThreads(config_.num_threads);
-    sess_opts_.SetInterOpNumThreads(config_.num_threads);
-
    Init();
  }