From f75412664634494b3a6ebefd1f70d63a0f3e18c5 Mon Sep 17 00:00:00 2001 From: YurenHao0426 Date: Wed, 22 Jul 2026 15:14:59 -0500 Subject: audit: conservatively count predictor arithmetic --- experiments/conv_run.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) (limited to 'experiments/conv_run.py') 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"), } -- cgit v1.2.3