adding gbps to fp8 gemm rowwise

Summary: TSIA

Reviewed By: bertmaher

Differential Revision: D70415169

fbshipit-source-id: ff7fc85584efd1366dfad450efcd5d6de2054419

Author: adamomainz

tritonbench/operators/fp8_gemm_rowwise/operator.py

@@ -204,6 +204,18 @@ def flops(
         flops = m * k * 2 * n
         return flops
 
+    @register_metric()
+    def gbps(self, fn, example_inputs: Any, metrics: BenchmarkOperatorMetrics) -> float:
+        def nbytes(t):
+            return t.numel() * t.element_size()
+
+        a, b, _, _ = example_inputs
+        c = fn()
+        c = c[0] if isinstance(c, tuple) else c
+
+        gb = (nbytes(a) + nbytes(b) + nbytes(c)) / 1e9
+        return gb / metrics.latency * 1e3
+
     @register_x_val(label="(M, N, K)")
     def get_x_val(self, example_inputs) -> Tuple[int, int, int]:
         xq, wq, _, _ = example_inputs