We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4c16ab8 commit 3ad4d0aCopy full SHA for 3ad4d0a
benchmarks/dashboard/ci_microbenchmark_runner.py
@@ -68,8 +68,12 @@ def create_benchmark_result(
68
else "unknown"
69
)
70
71
- # Format shape as M-K-N
72
- 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 = (
73
+ f"{quant_type}-{shape[0]}-{shape[1]}-{shape[2]}"
74
+ if len(shape) == 3
75
+ else "unknown"
76
+ )
77
78
return {
79
"benchmark": {
@@ -83,7 +87,7 @@ def create_benchmark_result(
83
87
},
84
88
85
89
"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)
91
"type": "micro-benchmark custom layer", # type
92
"origins": ["torchao"],
93
0 commit comments