Rearrange error checking in benchmark_model so that most informative check is done first.

Change: 128177141

1 files changed, 4 insertions, 3 deletions

diff --git a/tensorflow/tools/benchmark/benchmark_model.cc b/tensorflow/tools/benchmark/benchmark_model.cc
index ccb579fd03..cc590e07d4 100644
--- a/tensorflow/tools/benchmark/benchmark_model.cc
+++ b/tensorflow/tools/benchmark/benchmark_model.cc
@@ -118,15 +118,16 @@ Status RunBenchmark(DataType input_data_type, TensorShape input_shape,
   s = session->Run(run_options, input_tensors, output_names, {},
                    &output_tensors, &run_metadata);
 
+  if (!s.ok()) {
+    LOG(ERROR) << "Error during inference: " << s;
+  }
+
   assert(run_metadata.has_step_stats());
 
   const StepStats& step_stats = run_metadata.step_stats();
 
   stats->ProcessStepStats(step_stats);
 
-  if (!s.ok()) {
-    LOG(ERROR) << "Error during inference: " << s;
-  }
   return s;
 }