audit: conservatively count predictor arithmetic

1 files changed, 7 insertions, 2 deletions

diff --git a/experiments/conv_run.py b/experiments/conv_run.py
index 2e2787b..c25c48c 100644
--- a/experiments/conv_run.py
+++ b/experiments/conv_run.py
@@ -191,7 +191,11 @@ def work_report(net, mode, counters):
             * net.mixed_elementwise_ops_per_example()
             + (counters["predictor_warmup_examples"]
                + counters["predictor_update_examples"])
-            * net.predictor_elementwise_ops_per_example)
+            * net.predictor_elementwise_ops_per_example
+            + counters["traffic_calibration_examples"]
+            * net.traffic_calibration_elementwise_ops_per_example
+            + counters["traffic_audit_examples"]
+            * net.predictor_audit_elementwise_ops_per_example)
     else:
         elementwise_operations = 0
     components = {
@@ -234,7 +238,8 @@ def work_report(net, mode, counters):
             "equals one forward-weight MAC count; BP reverse is estimated as one "
             "weight-gradient plus one activation-gradient convolution per forward "
             "convolution; mixed-traffic/predictor elementwise arithmetic is reported "
-            "separately and is not folded into MACs"),
+            "as a conservative operation estimate separately and is not folded "
+            "into MACs"),
     }