h3-blackwell-runtime/benchmarks/gb10-cute-p0-attn-qkv-128rows.json

84 lines
1.7 KiB
JSON
Raw Normal View History

{
"device": "NVIDIA GB10",
"torch": "2.9.1+cu130",
"cutlass_dsl": "4.6.2",
"metadata": {
"width": 1344,
"height": 768,
"frames": 124,
"steps": 12,
"sampler_step": 1,
"seed": 440420,
"text_tokens": 100,
"hidden_shape": [
37810,
5376
],
"segments": [
[
0,
100,
1
],
[
100,
514,
2
],
[
514,
37810,
0
]
]
},
"linear": "attn_qkv_proj",
"rows": 128,
"mnk": [
128,
21504,
5376
],
"tile_shape_mnk": [
128,
128,
128
],
"swap_nibbles": false,
"reencode_fp4": true,
"zero_a": false,
"cute_shapes": {
"a": "(128, 5376, 1)",
"b": "(21504, 5376, 1)",
"sfa": "(128, 336, 1)",
"sfb": "(21504, 336, 1)",
"c": "(128, 21504, 1)"
},
"tensor_scales": {
"a": 0.0205078125,
"b": 0.00141143798828125,
"alpha": 2.8945505619049072e-05
},
"raw_output": {
"dtype": "torch.bfloat16",
"checksum": -640210368.0,
"max_abs": 6455296.0,
"nonzero": 2752509,
"finite": true
},
"raw_blockscaled_parity": {
"reference_checksum": -640210368.0,
"candidate_checksum": -640210368.0,
"equal": true,
"max_abs": 0.0,
"mean_abs": 0.0
},
"reference_checksum": -18469.02734375,
"candidate_checksum": -18533.14453125,
"equal": false,
"max_abs": 1.0,
"mean_abs": 0.018117837607860565,
"relative_l2": 0.002810996025800705,
"numerical_note": "The stock SM121 kernel's validated BF16 epilogue rounds before the external global-scale product. Exact H3 integration requires applying alpha in a custom epilogue before BF16 conversion."
}