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

[benchmark] Add lite opencl gpu option support (#1944)

Browse Source 
[benchmark] add lite opencl gpu option support
This commit is contained in:
DefTruth
2023-05-15 20:41:10 +08:00
committed by GitHub
parent b12d330076
commit a5a7c962d0
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
benchmark/cpp/option.h
View File

@@ -53,6 +53,11 @@ static bool CreateRuntimeOption(fastdeploy::RuntimeOption* option,
if (config_info["use_fp16"] == "true") {
option->trt_option.enable_fp16 = true;
}
} else if (config_info["backend"] == "lite") {
option->UsePaddleLiteBackend();
if (config_info["use_fp16"] == "true") {
option->paddle_lite_option.enable_fp16 = true;
}
} else if (config_info["backend"] == "default") {
PrintBenchmarkInfo(config_info);
return true;