apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-06 00:57:33 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

【Hackthon_4th 180】Support HORIZON BPU Backend for FastDeploy (#1822)

Browse Source 
* add horizon backend and PPYOLOE examples

* 更改horizon头文件编码规范

* 更改horizon头文件编码规范

* 更改horizon头文件编码规范

* 增加horizon packages下载及自动安装

* Add UseHorizonNPUBackend Method

* 删除编译FD SDK后多余的头文件,同时更改部分规范

* Update horizon.md

* Update horizon.md

---------

Co-authored-by: DefTruth <31974251+DefTruth@users.noreply.github.com>
This commit is contained in:
seyosum
2023-05-06 16:10:37 +08:00
committed by GitHub
parent 9164796645
commit df8dd3e3ac
29 changed files with 1693 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
fastdeploy/runtime/enum_variables.cc
View File

@@ -32,6 +32,8 @@ 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){
out << "Backend::HORIZONNPU";
} else {
out << "UNKNOWN-Backend";
}
@@ -49,6 +51,9 @@ std::ostream& operator<<(std::ostream& out, const Device& d) {
case Device::RKNPU:
out << "Device::RKNPU";
break;
case Device::SUNRISENPU:
out << "Device::SUNRISENPU";
break;
case Device::SOPHGOTPUD:
out << "Device::SOPHGOTPUD";
break;
@@ -81,7 +86,10 @@ std::ostream& operator<<(std::ostream& out, const ModelFormat& format) {
out << "ModelFormat::SOPHGO";
} else if (format == ModelFormat::TORCHSCRIPT) {
out << "ModelFormat::TORCHSCRIPT";
} else {
} else if (format == ModelFormat::HORIZON) {
out << "ModelFormat::HORIZON";
}
else {
out << "UNKNOWN-ModelFormat";
}
return out;
@@ -110,6 +118,9 @@ std::vector<Backend> GetAvailableBackends() {
#ifdef ENABLE_RKNPU2_BACKEND
backends.push_back(Backend::RKNPU2);
#endif
#ifdef ENABLE_HORIZON_BACKEND
backends.push_back(Backend::HORIZONNPU);
#endif
#ifdef ENABLE_SOPHGO_BACKEND
backends.push_back(Backend::SOPHGOTPU);
#endif