fixing key error in aggregate data

adamomainz · facebook-github-bot · commit a21b30e48ad3 · 2024-10-21T15:43:27.000-07:00
Summary: for some reason  OSS isnt happy with dict.get so im moving to this slightly less pythonic but more exact approach

Reviewed By: bertmaher, sfzhu93

Differential Revision: D64698791

fbshipit-source-id: 48cc4b6f7df61287efdc71c30176c2830dfde110
diff --git a/torchbenchmark/util/triton_op.py b/torchbenchmark/util/triton_op.py
@@ -364,6 +364,8 @@ def userbenchmark_dict(self) -> Dict[str, Any]:
                 agg_metric_name = (
                     f"tritonbench_{self.op_name}_{self.op_mode}[{provider}]_{metrics}"
                 )
+                if value is None:
+                    continue
                 agg_data[agg_metric_name] = agg_data.get(agg_metric_name, 0) + value
         final_agg_data = {k: v / num_rows for k, v in agg_data.items()}
         userbenchmark_metrics_dict.update(final_agg_data)

Original file line number	Diff line number	Diff line change
`@@ -364,6 +364,8 @@ def userbenchmark_dict(self) -> Dict[str, Any]:`
`364`	`364`	`agg_metric_name = (`
`365`	`365`	`f"tritonbench_{self.op_name}_{self.op_mode}[{provider}]_{metrics}"`
`366`	`366`	`)`
	`367`	`+ if value is None:`
	`368`	`+ continue`
`367`	`369`	`agg_data[agg_metric_name] = agg_data.get(agg_metric_name, 0) + value`
`368`	`370`	`final_agg_data = {k: v / num_rows for k, v in agg_data.items()}`
`369`	`371`	`userbenchmark_metrics_dict.update(final_agg_data)`