[Benchmark] Optimize log info for use_fp16 flag on CPU/XPU

This commit is contained in:
DefTruth
2023-03-13 13:38:10 +08:00
committed by GitHub
parent 5a708f5163
commit 57decf349e

View File

@@ -62,8 +62,8 @@ static void PrintBenchmarkInfo(std::unordered_map<std::string,
ss << "device: " << config_info["device"] << std::endl; ss << "device: " << config_info["device"] << std::endl;
if (config_info["device"] == "gpu") { if (config_info["device"] == "gpu") {
ss << "device_id: " << config_info["device_id"] << std::endl; ss << "device_id: " << config_info["device_id"] << std::endl;
ss << "use_fp16: " << config_info["use_fp16"] << std::endl;
} }
ss << "use_fp16: " << config_info["use_fp16"] << std::endl;
ss << "backend: " << config_info["backend"] << std::endl; ss << "backend: " << config_info["backend"] << std::endl;
if (config_info["device"] == "cpu") { if (config_info["device"] == "cpu") {
ss << "cpu_thread_nums: " << config_info["cpu_thread_nums"] << std::endl; ss << "cpu_thread_nums: " << config_info["cpu_thread_nums"] << std::endl;