feat: add directml support (#1153)

2024-07-22 18:50:48 +03:00
parent ea1d81bdfe
commit d32a46169f
6 changed files with 218 additions and 7 deletions
--- a/sherpa-onnx/csrc/provider.cc
+++ b/sherpa-onnx/csrc/provider.cc
@@ -26,6 +26,8 @@ Provider StringToProvider(std::string s) {
    return Provider::kNNAPI;
  } else if (s == "trt") {
    return Provider::kTRT;
+  } else if (s == "directml") {
+    return Provider::kDirectML;
  } else {
    SHERPA_ONNX_LOGE("Unsupported string: %s. Fallback to cpu", s.c_str());
    return Provider::kCPU;