adding aggregates to servicelab

Summary: current aggregation does not seem to be working as expected. Adding another aggregation field before changing the previous over

Reviewed By: xuzhao9

Differential Revision: D64616616

fbshipit-source-id: 676f09035e0d4427e9b60e9ed8f8c790782f0aec

Author: adamomainz

torchbenchmark/util/triton_op.py

@@ -351,13 +351,23 @@ def userbenchmark_dict(self) -> Dict[str, Any]:
         # tritonbench_{op_name}_{op_mode}[{x_val}-{provider}-{metric}]
         userbenchmark_metrics_dict = {}
         headers, table = self._table()
+        num_rows = len(table)
+        agg_data = {}
         for row in table:
             x_val = row[0]
+
             for ind, value in enumerate(row[1:]):
                 header = headers[ind + 1]
                 provider, _dash, metrics = header.partition("-")
                 metric_name = f"tritonbench_{self.op_name}_{self.op_mode}[x_{x_val}-{provider}]_{metrics}"
                 userbenchmark_metrics_dict[metric_name] = value
+                agg_metric_name = (
+                    f"tritonbench_{self.op_name}_{self.op_mode}[{provider}]_{metrics}"
+                )
+                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)
+
         return userbenchmark_metrics_dict
 
     def get_y_vals(self, x_val, provider, metric_name: str):