mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-06 17:17:14 +08:00
【Hackthon_4th 180】Support HORIZON BPU Backend for FastDeploy (#1822)
* 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:
@@ -38,6 +38,7 @@ enum Backend {
|
||||
LITE, ///< Paddle Lite, support Paddle format model, ARM CPU only
|
||||
RKNPU2, ///< RKNPU2, support RKNN format model, Rockchip NPU only
|
||||
SOPHGOTPU, ///< SOPHGOTPU, support SOPHGO format model, Sophgo TPU only
|
||||
HORIZONNPU, ///< HORIZONNPU, support Horizon format model, Horizon NPU
|
||||
};
|
||||
|
||||
/**
|
||||
@@ -60,7 +61,8 @@ enum FASTDEPLOY_DECL Device {
|
||||
KUNLUNXIN,
|
||||
ASCEND,
|
||||
SOPHGOTPUD,
|
||||
DIRECTML
|
||||
DIRECTML,
|
||||
SUNRISENPU,
|
||||
};
|
||||
|
||||
/*! Deep learning model format */
|
||||
@@ -71,6 +73,7 @@ enum ModelFormat {
|
||||
RKNN, ///< Model with RKNN format
|
||||
TORCHSCRIPT, ///< Model with TorchScript format
|
||||
SOPHGO, ///< Model with SOPHGO format
|
||||
HORIZON, ///< Model with HORIZON format
|
||||
};
|
||||
|
||||
/// Describle all the supported backends for specified model format
|
||||
@@ -80,6 +83,7 @@ static std::map<ModelFormat, std::vector<Backend>>
|
||||
Backend::ORT, Backend::OPENVINO, Backend::TRT}},
|
||||
{ModelFormat::ONNX, {Backend::ORT, Backend::OPENVINO, Backend::TRT}},
|
||||
{ModelFormat::RKNN, {Backend::RKNPU2}},
|
||||
{ModelFormat::HORIZON, {Backend::HORIZONNPU}},
|
||||
{ModelFormat::TORCHSCRIPT, {Backend::POROS}},
|
||||
{ModelFormat::SOPHGO, {Backend::SOPHGOTPU}}
|
||||
};
|
||||
@@ -91,6 +95,7 @@ static std::map<Device, std::vector<Backend>>
|
||||
Backend::OPENVINO, Backend::POROS}},
|
||||
{Device::GPU, {Backend::PDINFER, Backend::ORT, Backend::TRT, Backend::POROS}},
|
||||
{Device::RKNPU, {Backend::RKNPU2}},
|
||||
{Device::SUNRISENPU, {Backend::HORIZONNPU}},
|
||||
{Device::IPU, {Backend::PDINFER}},
|
||||
{Device::TIMVX, {Backend::LITE}},
|
||||
{Device::KUNLUNXIN, {Backend::LITE}},
|
||||
|
Reference in New Issue
Block a user