Fix metrics collection with --bypass-fail

Summary: Fixes an issue where computing the average with `--bypass-fail` will fail if a subset of the rows are None as it will revert `avg_row[col_num]` to None.

Reviewed By: xuzhao9

Differential Revision: D77949601

fbshipit-source-id: 3ff3d0491c4e3ff5ba2e652e4fad154067fa9707

Author: njriasan

tritonbench/utils/triton_op.py

@@ -351,11 +351,10 @@ def select_metric(backend, m):
                 if len(avg_row) <= col_num:
                     avg_row.append(next_val if isinstance(next_val, Number) else None)
                 else:
-                    avg_row[col_num] = (
-                        avg_row[col_num] + next_val
-                        if isinstance(next_val, Number)
-                        else None
-                    )
+                    if avg_row[col_num] is None:
+                        avg_row[col_num] = next_val
+                    elif isinstance(next_val, Number):
+                        avg_row[col_num] = avg_row[col_num] + next_val
                 col_num += 1
             for backend in backends:
                 if x_val == "hashes" and len(hashes) > 0:
@@ -381,11 +380,10 @@ def select_metric(backend, m):
                             metric_val if isinstance(metric_val, Number) else None
                         )
                     else:
-                        avg_row[col_num] = (
-                            avg_row[col_num] + metric_val
-                            if isinstance(metric_val, Number)
-                            else None
-                        )
+                        if avg_row[col_num] is None:
+                            avg_row[col_num] = metric_val
+                        elif isinstance(metric_val, Number):
+                            avg_row[col_num] = avg_row[col_num] + metric_val
                     col_num += 1
             table.append(row)