Fix extra N/A row

Change-Id: I588bf159711b04f8caeca3f1bd62b1d17f3b0718
Jeffwhen · Oct 26, 2022 · 24575d0 · 24575d0
1 parent a104ee3
commit 24575d0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/tpu_perf/run.py b/python/tpu_perf/run.py
@@ -153,7 +153,8 @@ def run_model(tree, config, name, b, profile_path, bmodel, stat_f, extra):
             logging.warning(
                 f'Profile exists but no GOPs in config.yaml, {config["name"]}')
             row.append('N/A')
-            row.extend(['N/A'] * (2 if option_cmodel_stats else 1))
+            if option_cmodel_stats:
+                row.append('N/A')
         else:
             row.append(f'{calc_mac_util(real_time):.2%}')
             if option_cmodel_stats: