apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-15 05:01:00 +08:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

[Hackathon 181] Add TVM support for FastDeploy on macOS (#1969)

Browse Source 
* update for tvm backend

* update third_party

* update third_party

* update

* update

* update

* update

* update

* update

* update

* update

---------

Co-authored-by: DefTruth <31974251+DefTruth@users.noreply.github.com>
This commit is contained in:
Zheng-Bicheng
2023-05-25 19:59:02 +08:00
committed by GitHub
parent 49c033a828
commit 643730bf5f
20 changed files with 658 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
fastdeploy/runtime/enum_variables.cc
View File

@@ -32,8 +32,10 @@ std::ostream& operator<<(std::ostream& out, const Backend& backend) {
out << "Backend::POROS";
} else if (backend == Backend::LITE) {
out << "Backend::PDLITE";
} else if(backend == Backend::HORIZONNPU){
} else if (backend == Backend::HORIZONNPU) {
out << "Backend::HORIZONNPU";
} else if (backend == Backend::TVM) {
out << "Backend::TVM";
} else {
out << "UNKNOWN-Backend";
}
@@ -88,8 +90,9 @@ std::ostream& operator<<(std::ostream& out, const ModelFormat& format) {
out << "ModelFormat::TORCHSCRIPT";
} else if (format == ModelFormat::HORIZON) {
out << "ModelFormat::HORIZON";
}
else {
} else if (format == ModelFormat::TVMFormat) {
out << "ModelFormat::TVMFormat";
} else {
out << "UNKNOWN-ModelFormat";
}
return out;
@@ -123,6 +126,9 @@ std::vector<Backend> GetAvailableBackends() {
#endif
#ifdef ENABLE_SOPHGO_BACKEND
backends.push_back(Backend::SOPHGOTPU);
#endif
#ifdef ENABLE_TVM_BACKEND
backends.push_back(Backend::TVM);
#endif
return backends;
}