provider.cc
932 字节
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
// sherpa-onnx/csrc/provider.cc
//
// Copyright (c) 2023 Xiaomi Corporation
#include "sherpa-onnx/csrc/provider.h"
#include <algorithm>
#include <cctype>
#include "sherpa-onnx/csrc/macros.h"
namespace sherpa_onnx {
Provider StringToProvider(std::string s) {
std::transform(s.cbegin(), s.cend(), s.begin(),
[](unsigned char c) { return std::tolower(c); });
if (s == "cpu") {
return Provider::kCPU;
} else if (s == "cuda") {
return Provider::kCUDA;
} else if (s == "coreml") {
return Provider::kCoreML;
} else if (s == "xnnpack") {
return Provider::kXnnpack;
} else if (s == "nnapi") {
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;
}
}
} // namespace sherpa_onnx