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

[benchmark] optimize benchmark bin options (#1984)

Browse Source 
* [benchmark] optimize benchmark bin options

* [benchmark] optimize benchmark bin options
This commit is contained in:
DefTruth
2023-05-24 14:56:00 +08:00
committed by GitHub
parent 9a5b291f60
commit 2672e2b881
4 changed files with 106 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
benchmark/cpp/option.h
View File

@@ -16,6 +16,22 @@
#include "fastdeploy/vision.h"
static void UpdateBaseCustomFlags(
std::unordered_map<std::string, std::string>& config_info) {
if (FLAGS_warmup > -1) {
config_info["warmup"] = std::to_string(FLAGS_warmup);
}
if (FLAGS_repeat > -1) {
config_info["repeat"] = std::to_string(FLAGS_repeat);
}
if (FLAGS_device_id > -1) {
config_info["device_id"] = std::to_string(FLAGS_device_id);
}
if (FLAGS_use_fp16) {
config_info["use_fp16"] = "true";
}
}
static bool CreateRuntimeOption(fastdeploy::RuntimeOption* option,
int argc, char* argv[], bool remove_flags) {
google::ParseCommandLineFlags(&argc, &argv, remove_flags);
@@ -23,14 +39,10 @@ static bool CreateRuntimeOption(fastdeploy::RuntimeOption* option,
std::unordered_map<std::string, std::string> config_info;
fastdeploy::benchmark::ResultManager::LoadBenchmarkConfig(
FLAGS_config_path, &config_info);
UpdateBaseCustomFlags(config_info);
int warmup = std::stoi(config_info["warmup"]);
int repeat = std::stoi(config_info["repeat"]);
if (FLAGS_warmup != -1) {
warmup = FLAGS_warmup;
}
if (FLAGS_repeat != -1) {
repeat = FLAGS_repeat;
}
if (config_info["profile_mode"] == "runtime") {
option->EnableProfiling(config_info["include_h2d_d2h"] == "true",
repeat, warmup);