From a5a7c962d0f96c724476a88f16aed657faad66a8 Mon Sep 17 00:00:00 2001 From: DefTruth <31974251+DefTruth@users.noreply.github.com> Date: Mon, 15 May 2023 20:41:10 +0800 Subject: [PATCH] [benchmark] Add lite opencl gpu option support (#1944) [benchmark] add lite opencl gpu option support --- benchmark/cpp/option.h | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/benchmark/cpp/option.h b/benchmark/cpp/option.h index f57d3ece2..f5f2b021a 100755 --- a/benchmark/cpp/option.h +++ b/benchmark/cpp/option.h @@ -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;