From 6c8f1ee32a02fc4aa858a00741667b303c269758 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Wed, 31 Jul 2019 13:38:40 -0700 Subject: [PATCH] This allows us to grep & process the output easier. PiperOrigin-RevId: 260989162 --- tensorflow/lite/tools/benchmark/logging.h | 17 +++++++++++++---- 1 file changed, 13 insertions(+), 4 deletions(-) diff --git a/tensorflow/lite/tools/benchmark/logging.h b/tensorflow/lite/tools/benchmark/logging.h index 42ccb2c9663..808090bf21f 100644 --- a/tensorflow/lite/tools/benchmark/logging.h +++ b/tensorflow/lite/tools/benchmark/logging.h @@ -46,10 +46,19 @@ class LoggingWrapper { std::stringstream& Stream() { return stream_; } ~LoggingWrapper() { if (should_log_) { - std::cerr << stream_.str() << std::endl; - if (severity_ == LogSeverity::FATAL) { - std::flush(std::cerr); - std::abort(); + switch (severity_) { + case LogSeverity::INFO: + case LogSeverity::WARN: + std::cout << stream_.str() << std::endl; + break; + case LogSeverity::ERROR: + std::cerr << stream_.str() << std::endl; + break; + case LogSeverity::FATAL: + std::cerr << stream_.str() << std::endl; + std::flush(std::cerr); + std::abort(); + break; } } }