File size: 4,768 Bytes
38770a7 8c41d43 38770a7 8c41d43 38770a7 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 | {
"op": "com.microsoft.MatMulBnb4",
"cases": [
{
"name": "bnb4-nf4-decode-m1-k2048-n2048-b64",
"preset": "smoke",
"vars": { "M": 1, "K": 2048, "N": 2048, "blockSize": 64 },
"attrs": { "K": 2048, "N": 2048, "block_size": 64, "quant_type": 1 },
"inputs": {
"aT": { "shape": [1, 2048], "dtype": "float32", "dist": "normal", "seed": 410, "scale": 0.2 },
"bT": { "shape": [2097152], "dtype": "uint8", "dist": "linearMod", "mod": 256, "step": 7 },
"absmaxT": {
"shape": [65536],
"dtype": "float32",
"dist": "uniform",
"seed": 411,
"offset": 0.04,
"scale": 0.01,
"signed": false
}
},
"outputs": { "yT": { "shape": [1, 2048], "dtype": "float32" } },
"bench": {
"metrics": [
{
"type": "bandwidth",
"value": "numel(shapes.bT) * 4 + (numel(shapes.aT) + numel(shapes.absmaxT) + numel(shapes.yT)) * 4"
}
]
}
},
{
"name": "bnb4-nf4-prefill-m128-k2048-n2048-b64",
"preset": "smoke",
"vars": { "M": 128, "K": 2048, "N": 2048, "blockSize": 64 },
"attrs": { "K": 2048, "N": 2048, "block_size": 64, "quant_type": 1 },
"tunableSpace": { "PORTABLE_TILE_K": [16, 32, 64] },
"inputs": {
"aT": { "shape": [128, 2048], "dtype": "float32", "dist": "normal", "seed": 412, "scale": 0.2 },
"bT": { "shape": [2097152], "dtype": "uint8", "dist": "linearMod", "mod": 256, "step": 7 },
"absmaxT": {
"shape": [65536],
"dtype": "float32",
"dist": "uniform",
"seed": 413,
"offset": 0.04,
"scale": 0.01,
"signed": false
}
},
"outputs": { "yT": { "shape": [128, 2048], "dtype": "float32" } },
"bench": {
"primary": true,
"metrics": [{ "type": "gflops", "value": "2 * dim(shapes.aT, 0) * args.K * args.N" }]
}
},
{
"name": "bnb4-fp4-prefill-m128-k2048-n2048-b64",
"preset": "smoke",
"vars": { "M": 128, "K": 2048, "N": 2048, "blockSize": 64 },
"attrs": { "K": 2048, "N": 2048, "block_size": 64, "quant_type": 0 },
"tunableSpace": { "PORTABLE_TILE_K": [16, 32, 64] },
"inputs": {
"aT": { "shape": [128, 2048], "dtype": "float32", "dist": "normal", "seed": 414, "scale": 0.2 },
"bT": { "shape": [2097152], "dtype": "uint8", "dist": "linearMod", "mod": 256, "step": 7 },
"absmaxT": {
"shape": [65536],
"dtype": "float32",
"dist": "uniform",
"seed": 415,
"offset": 0.04,
"scale": 0.01,
"signed": false
}
},
"outputs": { "yT": { "shape": [128, 2048], "dtype": "float32" } },
"bench": { "metrics": [{ "type": "gflops", "value": "2 * dim(shapes.aT, 0) * args.K * args.N" }] }
},
{
"name": "bnb4-nf4-prefill-m512-k4096-n4096-b64",
"preset": "model",
"vars": { "M": 512, "K": 4096, "N": 4096, "blockSize": 64 },
"attrs": { "K": 4096, "N": 4096, "block_size": 64, "quant_type": 1 },
"inputs": {
"aT": { "shape": [512, 4096], "dtype": "float32", "dist": "normal", "seed": 416, "scale": 0.2 },
"bT": { "shape": [8388608], "dtype": "uint8", "dist": "linearMod", "mod": 256, "step": 7 },
"absmaxT": {
"shape": [262144],
"dtype": "float32",
"dist": "uniform",
"seed": 417,
"offset": 0.04,
"scale": 0.01,
"signed": false
}
},
"outputs": { "yT": { "shape": [512, 4096], "dtype": "float32" } },
"bench": { "metrics": [{ "type": "gflops", "value": "2 * dim(shapes.aT, 0) * args.K * args.N" }] },
"tunableSpace": { "PORTABLE_TILE_K": [16, 32, 64] }
},
{
"name": "bnb4-nf4-prefill-m512-k4096-n4096-b64-f16",
"preset": "model",
"vars": { "M": 512, "K": 4096, "N": 4096, "blockSize": 64 },
"attrs": { "K": 4096, "N": 4096, "block_size": 64, "quant_type": 1 },
"inputs": {
"aT": { "shape": [512, 4096], "dtype": "float16", "dist": "normal", "seed": 416, "scale": 0.2 },
"bT": { "shape": [8388608], "dtype": "uint8", "dist": "linearMod", "mod": 256, "step": 7 },
"absmaxT": {
"shape": [262144],
"dtype": "float16",
"dist": "uniform",
"seed": 417,
"offset": 0.04,
"scale": 0.01,
"signed": false
}
},
"outputs": { "yT": { "shape": [512, 4096], "dtype": "float16" } },
"bench": { "metrics": [{ "type": "gflops", "value": "2 * dim(shapes.aT, 0) * args.K * args.N" }] },
"tunableSpace": { "PORTABLE_TILE_K": [16, 32, 64] }
}
]
}
|