Support TensorRT provider (#921)

Signed-off-by: manickavela1998@gmail.com <manickavela1998@gmail.com>
Signed-off-by: manickavela1998@gmail.com <manickavela.arumugam@uniphore.com>
This commit is contained in:
Manix
2024-06-06 08:15:28 +05:30
committed by GitHub
parent 7e0931c762
commit 69347ffc8f
3 changed files with 64 additions and 1 deletions

View File

@@ -24,6 +24,8 @@ Provider StringToProvider(std::string s) {
return Provider::kXnnpack;
} else if (s == "nnapi") {
return Provider::kNNAPI;
} else if (s == "trt") {
return Provider::kTRT;
} else {
SHERPA_ONNX_LOGE("Unsupported string: %s. Fallback to cpu", s.c_str());
return Provider::kCPU;