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

Add Huawei Ascend NPU deploy through PaddleLite CANN

Browse Source
This commit is contained in:
yunyaoXYY
2022-11-30 07:50:24 +00:00
parent 9d78b1d414
commit a5e3c1ecb3
18 changed files with 371 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
fastdeploy/fastdeploy_model.h
View File

@@ -45,6 +45,9 @@ class FASTDEPLOY_DECL FastDeployModel {
/** Model's valid timvx backends. This member defined all the timvx backends have successfully tested for the model
*/
std::vector<Backend> valid_timvx_backends = {};
/** Model's valid cann backends. This member defined all the cann backends have successfully tested for the model
*/
std::vector<Backend> valid_cann_backends = {};
/** Model's valid hardware backends. This member defined all the gpu backends have successfully tested for the model
*/
std::vector<Backend> valid_rknpu_backends = {};
@@ -129,6 +132,7 @@ class FASTDEPLOY_DECL FastDeployModel {
bool CreateIpuBackend();
bool CreateRKNPUBackend();
bool CreateTimVXBackend();
bool CreateCANNBackend();
std::shared_ptr<Runtime> runtime_;
bool runtime_initialized_ = false;