diff --git a/benchmarks/dashboard/ci_microbenchmark_runner.py b/benchmarks/dashboard/ci_microbenchmark_runner.py index 29971692ba..2379462bd8 100644 --- a/benchmarks/dashboard/ci_microbenchmark_runner.py +++ b/benchmarks/dashboard/ci_microbenchmark_runner.py @@ -68,8 +68,12 @@ def create_benchmark_result( else "unknown" ) - # Format shape as M-K-N - mkn_name = f"{shape[0]}-{shape[1]}-{shape[2]}" if len(shape) == 3 else "unknown" + # Format shape as quant-type-M-K-N + mkn_name = ( + f"{quant_type}-{shape[0]}-{shape[1]}-{shape[2]}" + if len(shape) == 3 + else "unknown" + ) return { "benchmark": { @@ -83,7 +87,7 @@ def create_benchmark_result( }, }, "model": { - "name": mkn_name, # name in M-K-N format + "name": mkn_name, # name in quant-type-M-K-N format (e.g., float8wo-1024-1024-1024) "type": "micro-benchmark custom layer", # type "origins": ["torchao"], },