Skip to content

Commit 3ad4d0a

Browse files
authored
Update model name in dashboard to reflect api name and shape (#3358)
1 parent 4c16ab8 commit 3ad4d0a

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

benchmarks/dashboard/ci_microbenchmark_runner.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -68,8 +68,12 @@ def create_benchmark_result(
6868
else "unknown"
6969
)
7070

71-
# Format shape as M-K-N
72-
mkn_name = f"{shape[0]}-{shape[1]}-{shape[2]}" if len(shape) == 3 else "unknown"
71+
# Format shape as quant-type-M-K-N
72+
mkn_name = (
73+
f"{quant_type}-{shape[0]}-{shape[1]}-{shape[2]}"
74+
if len(shape) == 3
75+
else "unknown"
76+
)
7377

7478
return {
7579
"benchmark": {
@@ -83,7 +87,7 @@ def create_benchmark_result(
8387
},
8488
},
8589
"model": {
86-
"name": mkn_name, # name in M-K-N format
90+
"name": mkn_name, # name in quant-type-M-K-N format (e.g., float8wo-1024-1024-1024)
8791
"type": "micro-benchmark custom layer", # type
8892
"origins": ["torchao"],
8993
},

0 commit comments

Comments
 (0)