h3-blackwell-runtime/benchmarks/gb10-cute-p1-stream-a-attn-qkv-128rows-alpha.json

87 lines
1.6 KiB
JSON
Raw Permalink 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,
"fuse_alpha": true,
"stream_a": true,
"cute_shapes": {
"a": "(128, 5376, 1)",
"b": "(21504, 5376, 1)",
"sfa": "(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": -18469.02734375,
"max_abs": 187.0,
"nonzero": 2752509,
"finite": true
},
"raw_blockscaled_parity": {
"applicable": false,
"reference_checksum": -640210368.0,
"candidate_checksum": -18469.02734375,
"equal": false,
"max_abs": 6455109.0,
"mean_abs": 455697.25
},
"reference_checksum": -18469.02734375,
"candidate_checksum": -18469.02734375,
"equal": true,
"max_abs": 0.0,
"mean_abs": 0.0,
"relative_l2": 0.0,
"numerical_note": "The experimental epilogue applies the FP32 global-scale product before BF16 conversion."
}