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

Add NNAdapter interface for paddlelite

Browse Source
This commit is contained in:
yunyaoXYY
2022-12-01 13:01:13 +00:00
parent e1aa652366
commit e957f8ac4e
5 changed files with 122 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
fastdeploy/backends/lite/lite_backend.cc
View File

@@ -81,8 +81,8 @@ void LiteBackend::BuildOption(const LiteBackendOption& option) {
valid_places.push_back(
paddle::lite_api::Place{TARGET(kARM), PRECISION(kInt8)});
}else if(option_.enable_cann){
config_.set_nnadapter_device_names({"huawei_ascend_npu"});
config_.set_nnadapter_context_properties("HUAWEI_ASCEND_NPU_SELECTED_DEVICE_IDS=0");
config_.set_nnadapter_device_names(option_.nnadapter_device_names);
config_.set_nnadapter_context_properties(option_.nnadapter_context_properties);
valid_places.push_back(
paddle::lite_api::Place{TARGET(kNNAdapter), PRECISION(kInt8)});
valid_places.push_back(