{ "fixtureArrays": { "block_column_indices_t_pattern": [0, 0, 1, 0, 1, 2, 0, 1, 2, 3, 0, 1, 2, 3, 4, 0, 1, 2, 3, 4, 5, 0, 1, 2, 3, 4, 5, 6, 0, 1, 2, 3, 4, 5, 6, 7, 0, 1, 2, 3, 4, 5, 6, 7, 8, 0, 2, 3, 4, 5, 6, 7, 8, 9, 0, 3, 4, 5, 6, 7, 8, 9, 10, 0, 4, 5, 6, 7, 8, 9, 10, 11, 0, 4, 5, 6, 7, 8, 9, 10, 11, 12, 0, 4, 6, 7, 8, 9, 10, 11, 12, 13, 0, 4, 7, 8, 9, 10, 11, 12, 13, 14, 0, 4, 8, 9, 10, 11, 12, 13, 14, 15, 0, 0, 1, 0, 1, 2, 0, 1, 2, 3, 0, 1, 2, 3, 4, 0, 1, 2, 3, 4, 5, 0, 1, 2, 3, 4, 5, 6, 0, 1, 2, 3, 4, 5, 6, 7, 1, 2, 3, 4, 5, 6, 7, 8, 2, 3, 4, 5, 6, 7, 8, 9, 3, 4, 5, 6, 7, 8, 9, 10, 3, 4, 5, 6, 7, 8, 9, 10, 11, 3, 5, 6, 7, 8, 9, 10, 11, 12, 3, 6, 7, 8, 9, 10, 11, 12, 13, 3, 7, 8, 9, 10, 11, 12, 13, 14, 3, 7, 8, 9, 10, 11, 12, 13, 14, 15, -1, -1, -1, -1, -1, -1, 0, 0, 1, 0, 1, 2, 0, 1, 2, 3, 0, 1, 2, 3, 4, 0, 1, 2, 3, 4, 5, 0, 1, 2, 3, 4, 5, 6, 0, 1, 2, 3, 4, 5, 6, 7, 1, 2, 3, 4, 5, 6, 7, 8, 2, 3, 4, 5, 6, 7, 8, 9, 2, 3, 4, 5, 6, 7, 8, 9, 10, 2, 4, 5, 6, 7, 8, 9, 10, 11, 2, 5, 6, 7, 8, 9, 10, 11, 12, 2, 6, 7, 8, 9, 10, 11, 12, 13, 2, 6, 7, 8, 9, 10, 11, 12, 13, 14, 2, 6, 8, 9, 10, 11, 12, 13, 14, 15, -1, -1, -1, -1, 0, 0, 1, 0, 1, 2, 0, 1, 2, 3, 0, 1, 2, 3, 4, 0, 1, 2, 3, 4, 5, 0, 1, 2, 3, 4, 5, 6, 0, 1, 2, 3, 4, 5, 6, 7, 1, 2, 3, 4, 5, 6, 7, 8, 1, 2, 3, 4, 5, 6, 7, 8, 9, 1, 3, 4, 5, 6, 7, 8, 9, 10, 1, 4, 5, 6, 7, 8, 9, 10, 11, 1, 5, 6, 7, 8, 9, 10, 11, 12, 1, 5, 6, 7, 8, 9, 10, 11, 12, 13, 1, 5, 7, 8, 9, 10, 11, 12, 13, 14, 1, 5, 8, 9, 10, 11, 12, 13, 14, 15, -1, -1], "sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT": [0, 1, 3, 6, 10, 15, 21, 28, 36, 45, 54, 63, 72, 82, 92, 102, 112, 0, 1, 3, 6, 10, 15, 21, 28, 36, 44, 52, 60, 69, 78, 87, 96, 106, 0, 1, 3, 6, 10, 15, 21, 28, 36, 44, 52, 61, 70, 79, 88, 98, 108, 0, 1, 3, 6, 10, 15, 21, 28, 36, 44, 53, 62, 71, 80, 90, 100, 110], "boundary-value-partition-float32-d520-s1_input_blockColIndicesT": [0, 0, 1, 1, 2, 1, 2, 3, -1, 0, 0, 1, 0, 1, 2, 0, 2, 3] }, "tunableSpace": { "WORKGROUP_SIZE": [32, 64, 128, 256], "APPEND_WORKGROUP_SIZE": [64, 128, 256] }, "cases": [ { "name": "sparse-prompt-b1-s1024-h32kv8-d128-blk64", "preset": "smoke", "vars": { "dtype": "float32", "batch": 1, "seq": 1024, "heads": 32, "kvHeads": 8, "headDim": 128, "qkPairs": 13254656, "attendedKeys": 1024, "dtypeBytes": 4 }, "attrs": { "num_heads": 32, "kv_num_heads": 8, "sparse_block_size": 64 }, "inputs": { "queryT": { "shape": [1, 1024, 4096], "dtype": "float32", "dist": "normal", "seed": 9300, "scale": 1 }, "keyT": { "shape": [1, 1024, 1024], "dtype": "float32", "dist": "normal", "seed": 9301, "scale": 1 }, "valueT": { "shape": [1, 1024, 1024], "dtype": "float32", "dist": "normal", "seed": 9302, "scale": 1 }, "pastKeyT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9303, "scale": 1 }, "pastValueT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9304, "scale": 1 }, "blockRowIndicesT": { "shape": [4, 17], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT" } } }, "blockColIndicesT": { "shape": [4, 112], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/block_column_indices_t_pattern" } } }, "totalSequenceLengthT": { "shape": [1], "dtype": "int32", "data": { "kind": "values", "values": [1024] } }, "keyTotalSequenceLengthsT": { "shape": [1], "dtype": "int32", "dist": "constant", "value": 1024 } }, "outputs": { "outputT": { "shape": [1, 1024, 4096], "dtype": "float32" } }, "bench": { "metrics": [{ "type": "gflops", "value": "4 * args.qkPairs * args.headDim" }] } }, { "name": "sparse-decode-b8-past1023-h32kv8-d128-blk64", "preset": "smoke", "vars": { "dtype": "float32", "batch": 8, "seq": 1, "heads": 32, "kvHeads": 8, "headDim": 128, "qkPairs": 163840, "attendedKeys": 640, "dtypeBytes": 4 }, "attrs": { "num_heads": 32, "kv_num_heads": 8, "sparse_block_size": 64 }, "inputs": { "queryT": { "shape": [8, 1, 4096], "dtype": "float32", "dist": "normal", "seed": 9310, "scale": 1 }, "keyT": { "shape": [8, 1, 1024], "dtype": "float32", "dist": "normal", "seed": 9311, "scale": 1 }, "valueT": { "shape": [8, 1, 1024], "dtype": "float32", "dist": "normal", "seed": 9312, "scale": 1 }, "pastKeyT": { "shape": [8, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9313, "scale": 1 }, "pastValueT": { "shape": [8, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9314, "scale": 1 }, "blockRowIndicesT": { "shape": [4, 17], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT" } } }, "blockColIndicesT": { "shape": [4, 112], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/block_column_indices_t_pattern" } } }, "totalSequenceLengthT": { "shape": [1], "dtype": "int32", "data": { "kind": "values", "values": [1024] } }, "keyTotalSequenceLengthsT": { "shape": [8], "dtype": "int32", "dist": "constant", "value": 1024 } }, "outputs": { "outputT": { "shape": [8, 1, 4096], "dtype": "float32" } }, "bench": { "metrics": [ { "type": "bandwidth", "value": "2 * args.batch * args.kvHeads * args.attendedKeys * args.headDim * args.dtypeBytes" } ] } }, { "name": "sparse-decode-f16-b8-past1023", "preset": "model", "vars": { "dtype": "float16", "batch": 8, "seq": 1, "heads": 32, "kvHeads": 8, "headDim": 128, "qkPairs": 163840, "attendedKeys": 640, "dtypeBytes": 2 }, "attrs": { "num_heads": 32, "kv_num_heads": 8, "sparse_block_size": 64 }, "inputs": { "queryT": { "shape": [8, 1, 4096], "dtype": "float16", "dist": "normal", "seed": 9320, "scale": 1 }, "keyT": { "shape": [8, 1, 1024], "dtype": "float16", "dist": "normal", "seed": 9321, "scale": 1 }, "valueT": { "shape": [8, 1, 1024], "dtype": "float16", "dist": "normal", "seed": 9322, "scale": 1 }, "pastKeyT": { "shape": [8, 8, 1024, 128], "dtype": "float16", "dist": "normal", "seed": 9323, "scale": 1 }, "pastValueT": { "shape": [8, 8, 1024, 128], "dtype": "float16", "dist": "normal", "seed": 9324, "scale": 1 }, "blockRowIndicesT": { "shape": [4, 17], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT" } } }, "blockColIndicesT": { "shape": [4, 112], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/block_column_indices_t_pattern" } } }, "totalSequenceLengthT": { "shape": [1], "dtype": "int32", "data": { "kind": "values", "values": [1024] } }, "keyTotalSequenceLengthsT": { "shape": [8], "dtype": "int32", "dist": "constant", "value": 1024 } }, "outputs": { "outputT": { "shape": [8, 1, 4096], "dtype": "float16" } }, "bench": { "metrics": [ { "type": "bandwidth", "value": "2 * args.batch * args.kvHeads * args.attendedKeys * args.headDim * args.dtypeBytes" } ] } }, { "name": "sparse-decode-packed-rotary-b8-past1023", "preset": "model", "vars": { "dtype": "float32", "batch": 8, "seq": 1, "heads": 32, "kvHeads": 8, "headDim": 128, "qkPairs": 163840, "attendedKeys": 640, "dtypeBytes": 4 }, "attrs": { "num_heads": 32, "kv_num_heads": 8, "sparse_block_size": 64, "do_rotary": 1 }, "inputs": { "queryT": { "shape": [8, 1, 6144], "dtype": "float32", "dist": "normal", "seed": 9330, "scale": 1 }, "pastKeyT": { "shape": [8, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9333, "scale": 1 }, "pastValueT": { "shape": [8, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9334, "scale": 1 }, "blockRowIndicesT": { "shape": [4, 17], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT" } } }, "blockColIndicesT": { "shape": [4, 112], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/block_column_indices_t_pattern" } } }, "totalSequenceLengthT": { "shape": [1], "dtype": "int32", "data": { "kind": "values", "values": [1024] } }, "keyTotalSequenceLengthsT": { "shape": [8], "dtype": "int32", "dist": "constant", "value": 1024 }, "cosCacheT": { "shape": [1024, 64], "dtype": "float32", "dist": "normal", "seed": 9335, "scale": 1 }, "sinCacheT": { "shape": [1024, 64], "dtype": "float32", "dist": "normal", "seed": 9336, "scale": 1 } }, "outputs": { "outputT": { "shape": [8, 1, 4096], "dtype": "float32" } }, "bench": { "metrics": [ { "type": "bandwidth", "value": "2 * args.batch * args.kvHeads * args.attendedKeys * args.headDim * args.dtypeBytes" } ] } }, { "name": "sparse-chunked-b2-s128-past896", "preset": "model", "vars": { "dtype": "float32", "batch": 2, "seq": 128, "heads": 32, "kvHeads": 8, "headDim": 128, "qkPairs": 4919296, "attendedKeys": 704, "dtypeBytes": 4 }, "attrs": { "num_heads": 32, "kv_num_heads": 8, "sparse_block_size": 64 }, "inputs": { "queryT": { "shape": [2, 128, 4096], "dtype": "float32", "dist": "normal", "seed": 9340, "scale": 1 }, "keyT": { "shape": [2, 128, 1024], "dtype": "float32", "dist": "normal", "seed": 9341, "scale": 1 }, "valueT": { "shape": [2, 128, 1024], "dtype": "float32", "dist": "normal", "seed": 9342, "scale": 1 }, "pastKeyT": { "shape": [2, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9343, "scale": 1 }, "pastValueT": { "shape": [2, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9344, "scale": 1 }, "blockRowIndicesT": { "shape": [4, 17], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT" } } }, "blockColIndicesT": { "shape": [4, 112], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/block_column_indices_t_pattern" } } }, "totalSequenceLengthT": { "shape": [1], "dtype": "int32", "data": { "kind": "values", "values": [1024] } }, "keyTotalSequenceLengthsT": { "shape": [2], "dtype": "int32", "dist": "constant", "value": 1024 } }, "outputs": { "outputT": { "shape": [2, 128, 4096], "dtype": "float32" } }, "bench": { "metrics": [{ "type": "gflops", "value": "4 * args.qkPairs * args.headDim" }] } }, { "name": "boundary-sgmat_capacity_d96_s64_packed0_rotary0", "attrs": { "num_heads": 2, "kv_num_heads": 2, "sparse_block_size": 64 }, "inputs": { "queryT": { "dtype": "float32", "shape": [2, 64, 192], "data": { "kind": "fillFloat32", "sinStep": 0.13, "cosStep": 0.29, "scale": 0.5 } }, "keyT": { "dtype": "float32", "shape": [2, 64, 192], "data": { "kind": "fillFloat32", "sinStep": 0.17, "cosStep": 0.31, "scale": 0.5 } }, "valueT": { "dtype": "float32", "shape": [2, 64, 192], "data": { "kind": "fillFloat32", "sinStep": 0.19, "cosStep": 0.23, "scale": 0.5 } }, "pastKeyT": { "dtype": "float32", "shape": [2, 2, 256, 96], "data": { "kind": "fillFloat32", "sinStep": 0.011, "cosStep": 0.017, "scale": 0.25 } }, "pastValueT": { "dtype": "float32", "shape": [2, 2, 256, 96], "data": { "kind": "fillFloat32", "sinStep": 0.019, "cosStep": 0.013, "scale": 0.25 } }, "blockRowIndicesT": { "dtype": "int32", "shape": [2, 5], "data": { "kind": "values", "values": [0, 1, 3, 5, 7, 0, 1, 2, 4, 5] } }, "blockColIndicesT": { "dtype": "int32", "shape": [2, 7], "data": { "kind": "values", "values": [0, 0, 1, 0, 2, 1, 3, 0, 1, 0, 2, 3, 0, 0] } }, "totalSequenceLengthT": { "dtype": "int32", "shape": [1], "data": { "kind": "values", "values": [256] } }, "keyTotalSequenceLengthsT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [164, 132] } } }, "outputs": { "outputT": { "dtype": "float32", "shape": [2, 64, 192] } }, "preset": "model" }, { "name": "boundary-sgmat_capacity_d96_s64_packed0_rotary1", "attrs": { "num_heads": 2, "kv_num_heads": 2, "sparse_block_size": 64, "do_rotary": 1, "rotary_interleaved": 0 }, "inputs": { "queryT": { "dtype": "float32", "shape": [2, 64, 192], "data": { "kind": "fillFloat32", "sinStep": 0.13, "cosStep": 0.29, "scale": 0.5 } }, "keyT": { "dtype": "float32", "shape": [2, 64, 192], "data": { "kind": "fillFloat32", "sinStep": 0.17, "cosStep": 0.31, "scale": 0.5 } }, "valueT": { "dtype": "float32", "shape": [2, 64, 192], "data": { "kind": "fillFloat32", "sinStep": 0.19, "cosStep": 0.23, "scale": 0.5 } }, "pastKeyT": { "dtype": "float32", "shape": [2, 2, 256, 96], "data": { "kind": "fillFloat32", "sinStep": 0.011, "cosStep": 0.017, "scale": 0.25 } }, "pastValueT": { "dtype": "float32", "shape": [2, 2, 256, 96], "data": { "kind": "fillFloat32", "sinStep": 0.019, "cosStep": 0.013, "scale": 0.25 } }, "blockRowIndicesT": { "dtype": "int32", "shape": [2, 5], "data": { "kind": "values", "values": [0, 1, 3, 5, 7, 0, 1, 2, 4, 5] } }, "blockColIndicesT": { "dtype": "int32", "shape": [2, 7], "data": { "kind": "values", "values": [0, 0, 1, 0, 2, 1, 3, 0, 1, 0, 2, 3, 0, 0] } }, "totalSequenceLengthT": { "dtype": "int32", "shape": [1], "data": { "kind": "values", "values": [256] } }, "keyTotalSequenceLengthsT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [164, 132] } }, "cosCacheT": { "dtype": "float32", "shape": [256, 48], "data": { "kind": "fillFloat32", "sinStep": 0.01, "cosStep": 0.013, "scale": 0.5 } }, "sinCacheT": { "dtype": "float32", "shape": [256, 48], "data": { "kind": "fillFloat32", "sinStep": 0.017, "cosStep": 0.013, "scale": 0.5 } } }, "outputs": { "outputT": { "dtype": "float32", "shape": [2, 64, 192] } }, "preset": "model" }, { "name": "boundary-sgmat_capacity_d96_s64_packed1_rotary0", "attrs": { "num_heads": 2, "kv_num_heads": 2, "sparse_block_size": 64 }, "inputs": { "queryT": { "dtype": "float32", "shape": [2, 64, 576], "data": { "kind": "fillFloat32", "sinStep": 0.13, "cosStep": 0.29, "scale": 0.5 } }, "pastKeyT": { "dtype": "float32", "shape": [2, 2, 256, 96], "data": { "kind": "fillFloat32", "sinStep": 0.011, "cosStep": 0.017, "scale": 0.25 } }, "pastValueT": { "dtype": "float32", "shape": [2, 2, 256, 96], "data": { "kind": "fillFloat32", "sinStep": 0.019, "cosStep": 0.013, "scale": 0.25 } }, "blockRowIndicesT": { "dtype": "int32", "shape": [2, 5], "data": { "kind": "values", "values": [0, 1, 3, 5, 7, 0, 1, 2, 4, 5] } }, "blockColIndicesT": { "dtype": "int32", "shape": [2, 7], "data": { "kind": "values", "values": [0, 0, 1, 0, 2, 1, 3, 0, 1, 0, 2, 3, 0, 0] } }, "totalSequenceLengthT": { "dtype": "int32", "shape": [1], "data": { "kind": "values", "values": [256] } }, "keyTotalSequenceLengthsT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [164, 132] } } }, "outputs": { "outputT": { "dtype": "float32", "shape": [2, 64, 192] } }, "preset": "model" }, { "name": "boundary-sgmat_capacity_d96_s64_packed1_rotary1", "attrs": { "num_heads": 2, "kv_num_heads": 2, "sparse_block_size": 64, "do_rotary": 1, "rotary_interleaved": 1 }, "inputs": { "queryT": { "dtype": "float32", "shape": [2, 64, 576], "data": { "kind": "fillFloat32", "sinStep": 0.13, "cosStep": 0.29, "scale": 0.5 } }, "pastKeyT": { "dtype": "float32", "shape": [2, 2, 256, 96], "data": { "kind": "fillFloat32", "sinStep": 0.011, "cosStep": 0.017, "scale": 0.25 } }, "pastValueT": { "dtype": "float32", "shape": [2, 2, 256, 96], "data": { "kind": "fillFloat32", "sinStep": 0.019, "cosStep": 0.013, "scale": 0.25 } }, "blockRowIndicesT": { "dtype": "int32", "shape": [2, 5], "data": { "kind": "values", "values": [0, 1, 3, 5, 7, 0, 1, 2, 4, 5] } }, "blockColIndicesT": { "dtype": "int32", "shape": [2, 7], "data": { "kind": "values", "values": [0, 0, 1, 0, 2, 1, 3, 0, 1, 0, 2, 3, 0, 0] } }, "totalSequenceLengthT": { "dtype": "int32", "shape": [1], "data": { "kind": "values", "values": [256] } }, "keyTotalSequenceLengthsT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [164, 132] } }, "cosCacheT": { "dtype": "float32", "shape": [256, 48], "data": { "kind": "fillFloat32", "sinStep": 0.01, "cosStep": 0.013, "scale": 0.5 } }, "sinCacheT": { "dtype": "float32", "shape": [256, 48], "data": { "kind": "fillFloat32", "sinStep": 0.017, "cosStep": 0.013, "scale": 0.5 } } }, "outputs": { "outputT": { "dtype": "float32", "shape": [2, 64, 192] } }, "preset": "model" }, { "name": "boundary-sgmat_capacity_d96_s65_packed0_rotary0", "attrs": { "num_heads": 2, "kv_num_heads": 2, "sparse_block_size": 64 }, "inputs": { "queryT": { "dtype": "float32", "shape": [2, 65, 192], "data": { "kind": "fillFloat32", "sinStep": 0.13, "cosStep": 0.29, "scale": 0.5 } }, "keyT": { "dtype": "float32", "shape": [2, 65, 192], "data": { "kind": "fillFloat32", "sinStep": 0.17, "cosStep": 0.31, "scale": 0.5 } }, "valueT": { "dtype": "float32", "shape": [2, 65, 192], "data": { "kind": "fillFloat32", "sinStep": 0.19, "cosStep": 0.23, "scale": 0.5 } }, "pastKeyT": { "dtype": "float32", "shape": [2, 2, 256, 96], "data": { "kind": "fillFloat32", "sinStep": 0.011, "cosStep": 0.017, "scale": 0.25 } }, "pastValueT": { "dtype": "float32", "shape": [2, 2, 256, 96], "data": { "kind": "fillFloat32", "sinStep": 0.019, "cosStep": 0.013, "scale": 0.25 } }, "blockRowIndicesT": { "dtype": "int32", "shape": [2, 5], "data": { "kind": "values", "values": [0, 1, 3, 5, 7, 0, 1, 2, 4, 5] } }, "blockColIndicesT": { "dtype": "int32", "shape": [2, 7], "data": { "kind": "values", "values": [0, 0, 1, 0, 2, 1, 3, 0, 1, 0, 2, 3, 0, 0] } }, "totalSequenceLengthT": { "dtype": "int32", "shape": [1], "data": { "kind": "values", "values": [256] } }, "keyTotalSequenceLengthsT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [165, 133] } } }, "outputs": { "outputT": { "dtype": "float32", "shape": [2, 65, 192] } }, "preset": "model" }, { "name": "boundary-sgmat_capacity_d96_s65_packed0_rotary1", "attrs": { "num_heads": 2, "kv_num_heads": 2, "sparse_block_size": 64, "do_rotary": 1, "rotary_interleaved": 0 }, "inputs": { "queryT": { "dtype": "float32", "shape": [2, 65, 192], "data": { "kind": "fillFloat32", "sinStep": 0.13, "cosStep": 0.29, "scale": 0.5 } }, "keyT": { "dtype": "float32", "shape": [2, 65, 192], "data": { "kind": "fillFloat32", "sinStep": 0.17, "cosStep": 0.31, "scale": 0.5 } }, "valueT": { "dtype": "float32", "shape": [2, 65, 192], "data": { "kind": "fillFloat32", "sinStep": 0.19, "cosStep": 0.23, "scale": 0.5 } }, "pastKeyT": { "dtype": "float32", "shape": [2, 2, 256, 96], "data": { "kind": "fillFloat32", "sinStep": 0.011, "cosStep": 0.017, "scale": 0.25 } }, "pastValueT": { "dtype": "float32", "shape": [2, 2, 256, 96], "data": { "kind": "fillFloat32", "sinStep": 0.019, "cosStep": 0.013, "scale": 0.25 } }, "blockRowIndicesT": { "dtype": "int32", "shape": [2, 5], "data": { "kind": "values", "values": [0, 1, 3, 5, 7, 0, 1, 2, 4, 5] } }, "blockColIndicesT": { "dtype": "int32", "shape": [2, 7], "data": { "kind": "values", "values": [0, 0, 1, 0, 2, 1, 3, 0, 1, 0, 2, 3, 0, 0] } }, "totalSequenceLengthT": { "dtype": "int32", "shape": [1], "data": { "kind": "values", "values": [256] } }, "keyTotalSequenceLengthsT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [165, 133] } }, "cosCacheT": { "dtype": "float32", "shape": [256, 48], "data": { "kind": "fillFloat32", "sinStep": 0.01, "cosStep": 0.013, "scale": 0.5 } }, "sinCacheT": { "dtype": "float32", "shape": [256, 48], "data": { "kind": "fillFloat32", "sinStep": 0.017, "cosStep": 0.013, "scale": 0.5 } } }, "outputs": { "outputT": { "dtype": "float32", "shape": [2, 65, 192] } }, "preset": "model" }, { "name": "boundary-sgmat_capacity_d96_s65_packed1_rotary0", "attrs": { "num_heads": 2, "kv_num_heads": 2, "sparse_block_size": 64 }, "inputs": { "queryT": { "dtype": "float32", "shape": [2, 65, 576], "data": { "kind": "fillFloat32", "sinStep": 0.13, "cosStep": 0.29, "scale": 0.5 } }, "pastKeyT": { "dtype": "float32", "shape": [2, 2, 256, 96], "data": { "kind": "fillFloat32", "sinStep": 0.011, "cosStep": 0.017, "scale": 0.25 } }, "pastValueT": { "dtype": "float32", "shape": [2, 2, 256, 96], "data": { "kind": "fillFloat32", "sinStep": 0.019, "cosStep": 0.013, "scale": 0.25 } }, "blockRowIndicesT": { "dtype": "int32", "shape": [2, 5], "data": { "kind": "values", "values": [0, 1, 3, 5, 7, 0, 1, 2, 4, 5] } }, "blockColIndicesT": { "dtype": "int32", "shape": [2, 7], "data": { "kind": "values", "values": [0, 0, 1, 0, 2, 1, 3, 0, 1, 0, 2, 3, 0, 0] } }, "totalSequenceLengthT": { "dtype": "int32", "shape": [1], "data": { "kind": "values", "values": [256] } }, "keyTotalSequenceLengthsT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [165, 133] } } }, "outputs": { "outputT": { "dtype": "float32", "shape": [2, 65, 192] } }, "preset": "model" }, { "name": "boundary-sgmat_capacity_d96_s65_packed1_rotary1", "attrs": { "num_heads": 2, "kv_num_heads": 2, "sparse_block_size": 64, "do_rotary": 1, "rotary_interleaved": 1 }, "inputs": { "queryT": { "dtype": "float32", "shape": [2, 65, 576], "data": { "kind": "fillFloat32", "sinStep": 0.13, "cosStep": 0.29, "scale": 0.5 } }, "pastKeyT": { "dtype": "float32", "shape": [2, 2, 256, 96], "data": { "kind": "fillFloat32", "sinStep": 0.011, "cosStep": 0.017, "scale": 0.25 } }, "pastValueT": { "dtype": "float32", "shape": [2, 2, 256, 96], "data": { "kind": "fillFloat32", "sinStep": 0.019, "cosStep": 0.013, "scale": 0.25 } }, "blockRowIndicesT": { "dtype": "int32", "shape": [2, 5], "data": { "kind": "values", "values": [0, 1, 3, 5, 7, 0, 1, 2, 4, 5] } }, "blockColIndicesT": { "dtype": "int32", "shape": [2, 7], "data": { "kind": "values", "values": [0, 0, 1, 0, 2, 1, 3, 0, 1, 0, 2, 3, 0, 0] } }, "totalSequenceLengthT": { "dtype": "int32", "shape": [1], "data": { "kind": "values", "values": [256] } }, "keyTotalSequenceLengthsT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [165, 133] } }, "cosCacheT": { "dtype": "float32", "shape": [256, 48], "data": { "kind": "fillFloat32", "sinStep": 0.01, "cosStep": 0.013, "scale": 0.5 } }, "sinCacheT": { "dtype": "float32", "shape": [256, 48], "data": { "kind": "fillFloat32", "sinStep": 0.017, "cosStep": 0.013, "scale": 0.5 } } }, "outputs": { "outputT": { "dtype": "float32", "shape": [2, 65, 192] } }, "preset": "model" }, { "name": "boundary-sgmat_prompt_separate_blk64_duplicate_csr", "attrs": { "num_heads": 4, "kv_num_heads": 2, "sparse_block_size": 64 }, "inputs": { "queryT": { "dtype": "float32", "shape": [1, 192, 256], "data": { "kind": "fillFloat32", "sinStep": 0.13, "cosStep": 0.29, "scale": 0.5 } }, "keyT": { "dtype": "float32", "shape": [1, 192, 128], "data": { "kind": "fillFloat32", "sinStep": 0.17, "cosStep": 0.31, "scale": 0.5 } }, "valueT": { "dtype": "float32", "shape": [1, 192, 128], "data": { "kind": "fillFloat32", "sinStep": 0.19, "cosStep": 0.23, "scale": 0.5 } }, "pastKeyT": { "dtype": "float32", "shape": [1, 2, 192, 64], "data": { "kind": "constant", "value": 0.0 } }, "pastValueT": { "dtype": "float32", "shape": [1, 2, 192, 64], "data": { "kind": "constant", "value": 0.0 } }, "blockRowIndicesT": { "dtype": "int32", "shape": [2, 4], "data": { "kind": "values", "values": [0, 1, 3, 5, 0, 1, 2, 3] } }, "blockColIndicesT": { "dtype": "int32", "shape": [2, 5], "data": { "kind": "values", "values": [0, 0, 1, 1, 1, 0, 1, 2, 0, 0] } }, "totalSequenceLengthT": { "dtype": "int32", "shape": [1], "data": { "kind": "values", "values": [192] } }, "keyTotalSequenceLengthsT": { "dtype": "int32", "shape": [1], "data": { "kind": "values", "values": [192] } } }, "outputs": { "outputT": { "dtype": "float32", "shape": [1, 192, 256] } }, "preset": "model" }, { "name": "boundary-sgmat_prompt_tail_s100_explicit_scale", "attrs": { "num_heads": 2, "kv_num_heads": 1, "sparse_block_size": 64, "scale": 0.25 }, "inputs": { "queryT": { "dtype": "float32", "shape": [1, 100, 64], "data": { "kind": "fillFloat32", "sinStep": 0.13, "cosStep": 0.29, "scale": 0.5 } }, "keyT": { "dtype": "float32", "shape": [1, 100, 32], "data": { "kind": "fillFloat32", "sinStep": 0.17, "cosStep": 0.31, "scale": 0.5 } }, "valueT": { "dtype": "float32", "shape": [1, 100, 32], "data": { "kind": "fillFloat32", "sinStep": 0.19, "cosStep": 0.23, "scale": 0.5 } }, "pastKeyT": { "dtype": "float32", "shape": [1, 1, 128, 32], "data": { "kind": "constant", "value": 0.0 } }, "pastValueT": { "dtype": "float32", "shape": [1, 1, 128, 32], "data": { "kind": "constant", "value": 0.0 } }, "blockRowIndicesT": { "dtype": "int32", "shape": [1, 3], "data": { "kind": "values", "values": [0, 1, 3] } }, "blockColIndicesT": { "dtype": "int32", "shape": [1, 3], "data": { "kind": "values", "values": [0, 0, 1] } }, "totalSequenceLengthT": { "dtype": "int32", "shape": [1], "data": { "kind": "values", "values": [100] } }, "keyTotalSequenceLengthsT": { "dtype": "int32", "shape": [1], "data": { "kind": "values", "values": [100] } } }, "outputs": { "outputT": { "dtype": "float32", "shape": [1, 100, 64] } }, "preset": "model" }, { "name": "boundary-sgmat_prompt_blk128_subtiles", "attrs": { "num_heads": 2, "kv_num_heads": 1, "sparse_block_size": 128 }, "inputs": { "queryT": { "dtype": "float32", "shape": [1, 256, 256], "data": { "kind": "fillFloat32", "sinStep": 0.13, "cosStep": 0.29, "scale": 0.5 } }, "keyT": { "dtype": "float32", "shape": [1, 256, 128], "data": { "kind": "fillFloat32", "sinStep": 0.17, "cosStep": 0.31, "scale": 0.5 } }, "valueT": { "dtype": "float32", "shape": [1, 256, 128], "data": { "kind": "fillFloat32", "sinStep": 0.19, "cosStep": 0.23, "scale": 0.5 } }, "pastKeyT": { "dtype": "float32", "shape": [1, 1, 256, 128], "data": { "kind": "constant", "value": 0.0 } }, "pastValueT": { "dtype": "float32", "shape": [1, 1, 256, 128], "data": { "kind": "constant", "value": 0.0 } }, "blockRowIndicesT": { "dtype": "int32", "shape": [1, 3], "data": { "kind": "values", "values": [0, 1, 2] } }, "blockColIndicesT": { "dtype": "int32", "shape": [1, 2], "data": { "kind": "values", "values": [0, 1] } }, "totalSequenceLengthT": { "dtype": "int32", "shape": [1], "data": { "kind": "values", "values": [256] } }, "keyTotalSequenceLengthsT": { "dtype": "int32", "shape": [1], "data": { "kind": "values", "values": [256] } } }, "outputs": { "outputT": { "dtype": "float32", "shape": [1, 256, 256] } }, "preset": "model" }, { "name": "boundary-sgmat_parallelism_d96_s128", "attrs": { "num_heads": 4, "kv_num_heads": 2, "sparse_block_size": 64 }, "inputs": { "queryT": { "dtype": "float32", "shape": [2, 128, 384], "data": { "kind": "fillFloat32", "sinStep": 0.13, "cosStep": 0.29, "scale": 0.5 } }, "keyT": { "dtype": "float32", "shape": [2, 128, 192], "data": { "kind": "fillFloat32", "sinStep": 0.17, "cosStep": 0.31, "scale": 0.5 } }, "valueT": { "dtype": "float32", "shape": [2, 128, 192], "data": { "kind": "fillFloat32", "sinStep": 0.19, "cosStep": 0.23, "scale": 0.5 } }, "pastKeyT": { "dtype": "float32", "shape": [2, 2, 256, 96], "data": { "kind": "fillFloat32", "sinStep": 0.011, "cosStep": 0.017, "scale": 0.25 } }, "pastValueT": { "dtype": "float32", "shape": [2, 2, 256, 96], "data": { "kind": "fillFloat32", "sinStep": 0.019, "cosStep": 0.013, "scale": 0.25 } }, "blockRowIndicesT": { "dtype": "int32", "shape": [2, 5], "data": { "kind": "values", "values": [0, 1, 3, 5, 7, 0, 1, 2, 4, 5] } }, "blockColIndicesT": { "dtype": "int32", "shape": [2, 7], "data": { "kind": "values", "values": [0, 0, 1, 0, 2, 1, 3, 0, 1, 0, 2, 3, 0, 0] } }, "totalSequenceLengthT": { "dtype": "int32", "shape": [1], "data": { "kind": "values", "values": [256] } }, "keyTotalSequenceLengthsT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [256, 256] } } }, "outputs": { "outputT": { "dtype": "float32", "shape": [2, 128, 384] } }, "preset": "model" }, { "name": "boundary-sgmat_parallelism_d96_s256", "attrs": { "num_heads": 4, "kv_num_heads": 2, "sparse_block_size": 64 }, "inputs": { "queryT": { "dtype": "float32", "shape": [2, 256, 384], "data": { "kind": "fillFloat32", "sinStep": 0.13, "cosStep": 0.29, "scale": 0.5 } }, "keyT": { "dtype": "float32", "shape": [2, 256, 192], "data": { "kind": "fillFloat32", "sinStep": 0.17, "cosStep": 0.31, "scale": 0.5 } }, "valueT": { "dtype": "float32", "shape": [2, 256, 192], "data": { "kind": "fillFloat32", "sinStep": 0.19, "cosStep": 0.23, "scale": 0.5 } }, "pastKeyT": { "dtype": "float32", "shape": [2, 2, 256, 96], "data": { "kind": "fillFloat32", "sinStep": 0.011, "cosStep": 0.017, "scale": 0.25 } }, "pastValueT": { "dtype": "float32", "shape": [2, 2, 256, 96], "data": { "kind": "fillFloat32", "sinStep": 0.019, "cosStep": 0.013, "scale": 0.25 } }, "blockRowIndicesT": { "dtype": "int32", "shape": [2, 5], "data": { "kind": "values", "values": [0, 1, 3, 5, 7, 0, 1, 2, 4, 5] } }, "blockColIndicesT": { "dtype": "int32", "shape": [2, 7], "data": { "kind": "values", "values": [0, 0, 1, 0, 2, 1, 3, 0, 1, 0, 2, 3, 0, 0] } }, "totalSequenceLengthT": { "dtype": "int32", "shape": [1], "data": { "kind": "values", "values": [256] } }, "keyTotalSequenceLengthsT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [256, 256] } } }, "outputs": { "outputT": { "dtype": "float32", "shape": [2, 256, 384] } }, "preset": "model" }, { "name": "sparse-low-parallelism-s64-h8-d32-past960", "preset": "model", "attrs": { "num_heads": 8, "kv_num_heads": 4, "sparse_block_size": 64 }, "inputs": { "queryT": { "shape": [1, 64, 256], "dtype": "float32", "dist": "normal", "seed": 9300, "scale": 1 }, "keyT": { "shape": [1, 64, 128], "dtype": "float32", "dist": "normal", "seed": 9301, "scale": 1 }, "valueT": { "shape": [1, 64, 128], "dtype": "float32", "dist": "normal", "seed": 9302, "scale": 1 }, "pastKeyT": { "shape": [1, 4, 1024, 32], "dtype": "float32", "dist": "normal", "seed": 9303, "scale": 1 }, "pastValueT": { "shape": [1, 4, 1024, 32], "dtype": "float32", "dist": "normal", "seed": 9304, "scale": 1 }, "blockRowIndicesT": { "shape": [4, 17], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT" } } }, "blockColIndicesT": { "shape": [4, 112], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/block_column_indices_t_pattern" } } }, "totalSequenceLengthT": { "shape": [1], "dtype": "int32", "data": { "kind": "values", "values": [1024] } }, "keyTotalSequenceLengthsT": { "shape": [1], "dtype": "int32", "dist": "constant", "value": 1024 } }, "outputs": { "outputT": { "shape": [1, 64, 256], "dtype": "float32" } } }, { "name": "sparse-low-parallelism-s64-h8-d96-past960", "preset": "model", "attrs": { "num_heads": 8, "kv_num_heads": 4, "sparse_block_size": 64 }, "inputs": { "queryT": { "shape": [1, 64, 768], "dtype": "float32", "dist": "normal", "seed": 9300, "scale": 1 }, "keyT": { "shape": [1, 64, 384], "dtype": "float32", "dist": "normal", "seed": 9301, "scale": 1 }, "valueT": { "shape": [1, 64, 384], "dtype": "float32", "dist": "normal", "seed": 9302, "scale": 1 }, "pastKeyT": { "shape": [1, 4, 1024, 96], "dtype": "float32", "dist": "normal", "seed": 9303, "scale": 1 }, "pastValueT": { "shape": [1, 4, 1024, 96], "dtype": "float32", "dist": "normal", "seed": 9304, "scale": 1 }, "blockRowIndicesT": { "shape": [4, 17], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT" } } }, "blockColIndicesT": { "shape": [4, 112], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/block_column_indices_t_pattern" } } }, "totalSequenceLengthT": { "shape": [1], "dtype": "int32", "data": { "kind": "values", "values": [1024] } }, "keyTotalSequenceLengthsT": { "shape": [1], "dtype": "int32", "dist": "constant", "value": 1024 } }, "outputs": { "outputT": { "shape": [1, 64, 768], "dtype": "float32" } } }, { "name": "sparse-low-parallelism-s64-h8-d128-past960", "preset": "model", "attrs": { "num_heads": 8, "kv_num_heads": 4, "sparse_block_size": 64 }, "inputs": { "queryT": { "shape": [1, 64, 1024], "dtype": "float32", "dist": "normal", "seed": 9300, "scale": 1 }, "keyT": { "shape": [1, 64, 512], "dtype": "float32", "dist": "normal", "seed": 9301, "scale": 1 }, "valueT": { "shape": [1, 64, 512], "dtype": "float32", "dist": "normal", "seed": 9302, "scale": 1 }, "pastKeyT": { "shape": [1, 4, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9303, "scale": 1 }, "pastValueT": { "shape": [1, 4, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9304, "scale": 1 }, "blockRowIndicesT": { "shape": [4, 17], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT" } } }, "blockColIndicesT": { "shape": [4, 112], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/block_column_indices_t_pattern" } } }, "totalSequenceLengthT": { "shape": [1], "dtype": "int32", "data": { "kind": "values", "values": [1024] } }, "keyTotalSequenceLengthsT": { "shape": [1], "dtype": "int32", "dist": "constant", "value": 1024 } }, "outputs": { "outputT": { "shape": [1, 64, 1024], "dtype": "float32" } } }, { "name": "sparse-crossover-s128-h8-d96-past128", "preset": "model", "attrs": { "num_heads": 8, "kv_num_heads": 4, "sparse_block_size": 64 }, "inputs": { "queryT": { "shape": [1, 128, 768], "dtype": "float32", "dist": "normal", "seed": 9300, "scale": 1 }, "keyT": { "shape": [1, 128, 384], "dtype": "float32", "dist": "normal", "seed": 9301, "scale": 1 }, "valueT": { "shape": [1, 128, 384], "dtype": "float32", "dist": "normal", "seed": 9302, "scale": 1 }, "pastKeyT": { "shape": [1, 4, 1024, 96], "dtype": "float32", "dist": "normal", "seed": 9303, "scale": 1 }, "pastValueT": { "shape": [1, 4, 1024, 96], "dtype": "float32", "dist": "normal", "seed": 9304, "scale": 1 }, "blockRowIndicesT": { "shape": [4, 17], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT" } } }, "blockColIndicesT": { "shape": [4, 112], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/block_column_indices_t_pattern" } } }, "totalSequenceLengthT": { "shape": [1], "dtype": "int32", "data": { "kind": "values", "values": [256] } }, "keyTotalSequenceLengthsT": { "shape": [1], "dtype": "int32", "dist": "constant", "value": 256 } }, "outputs": { "outputT": { "shape": [1, 128, 768], "dtype": "float32" } } }, { "name": "boundary-value-partition-float32-d520-s1", "attrs": { "num_heads": 4, "kv_num_heads": 2, "sparse_block_size": 16 }, "inputs": { "queryT": { "dtype": "float32", "shape": [2, 1, 2080], "data": { "kind": "fillFloat32", "sinStep": 0.13, "cosStep": 0.29, "scale": 0.5 } }, "keyT": { "dtype": "float32", "shape": [2, 1, 1040], "data": { "kind": "fillFloat32", "sinStep": 0.17, "cosStep": 0.31, "scale": 0.5 } }, "valueT": { "dtype": "float32", "shape": [2, 1, 1040], "data": { "kind": "fillFloat32", "sinStep": 0.19, "cosStep": 0.23, "scale": 0.5 } }, "pastKeyT": { "dtype": "float32", "shape": [2, 2, 64, 520], "data": { "kind": "fillFloat32", "sinStep": 0.011, "cosStep": 0.017, "scale": 0.25 } }, "pastValueT": { "dtype": "float32", "shape": [2, 2, 64, 520], "data": { "kind": "fillFloat32", "sinStep": 0.019, "cosStep": 0.013, "scale": 0.25 } }, "blockRowIndicesT": { "dtype": "int32", "shape": [2, 5], "data": { "kind": "values", "values": [0, 1, 3, 5, 8, 0, 1, 3, 6, 9] } }, "blockColIndicesT": { "dtype": "int32", "shape": [2, 9], "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/boundary-value-partition-float32-d520-s1_input_blockColIndicesT" } } }, "totalSequenceLengthT": { "dtype": "int32", "shape": [1], "data": { "kind": "values", "values": [41] } }, "keyTotalSequenceLengthsT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [41, 34] } } }, "outputs": { "outputT": { "dtype": "float32", "shape": [2, 1, 2080] } }, "preset": "model" }, { "name": "boundary-value-partition-float32-d520-s5", "attrs": { "num_heads": 4, "kv_num_heads": 2, "sparse_block_size": 16 }, "inputs": { "queryT": { "dtype": "float32", "shape": [2, 5, 2080], "data": { "kind": "fillFloat32", "sinStep": 0.13, "cosStep": 0.29, "scale": 0.5 } }, "keyT": { "dtype": "float32", "shape": [2, 5, 1040], "data": { "kind": "fillFloat32", "sinStep": 0.17, "cosStep": 0.31, "scale": 0.5 } }, "valueT": { "dtype": "float32", "shape": [2, 5, 1040], "data": { "kind": "fillFloat32", "sinStep": 0.19, "cosStep": 0.23, "scale": 0.5 } }, "pastKeyT": { "dtype": "float32", "shape": [2, 2, 64, 520], "data": { "kind": "fillFloat32", "sinStep": 0.011, "cosStep": 0.017, "scale": 0.25 } }, "pastValueT": { "dtype": "float32", "shape": [2, 2, 64, 520], "data": { "kind": "fillFloat32", "sinStep": 0.019, "cosStep": 0.013, "scale": 0.25 } }, "blockRowIndicesT": { "dtype": "int32", "shape": [2, 5], "data": { "kind": "values", "values": [0, 1, 3, 5, 8, 0, 1, 3, 6, 9] } }, "blockColIndicesT": { "dtype": "int32", "shape": [2, 9], "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/boundary-value-partition-float32-d520-s1_input_blockColIndicesT" } } }, "totalSequenceLengthT": { "dtype": "int32", "shape": [1], "data": { "kind": "values", "values": [41] } }, "keyTotalSequenceLengthsT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [41, 34] } } }, "outputs": { "outputT": { "dtype": "float32", "shape": [2, 5, 2080] } }, "preset": "model" }, { "name": "boundary-value-owner-float32-d144-wg32-s1", "attrs": { "num_heads": 4, "kv_num_heads": 2, "sparse_block_size": 16 }, "inputs": { "queryT": { "dtype": "float32", "shape": [2, 1, 576], "data": { "kind": "fillFloat32", "sinStep": 0.13, "cosStep": 0.29, "scale": 0.5 } }, "keyT": { "dtype": "float32", "shape": [2, 1, 288], "data": { "kind": "fillFloat32", "sinStep": 0.17, "cosStep": 0.31, "scale": 0.5 } }, "valueT": { "dtype": "float32", "shape": [2, 1, 288], "data": { "kind": "fillFloat32", "sinStep": 0.19, "cosStep": 0.23, "scale": 0.5 } }, "pastKeyT": { "dtype": "float32", "shape": [2, 2, 64, 144], "data": { "kind": "fillFloat32", "sinStep": 0.011, "cosStep": 0.017, "scale": 0.25 } }, "pastValueT": { "dtype": "float32", "shape": [2, 2, 64, 144], "data": { "kind": "fillFloat32", "sinStep": 0.019, "cosStep": 0.013, "scale": 0.25 } }, "blockRowIndicesT": { "dtype": "int32", "shape": [2, 5], "data": { "kind": "values", "values": [0, 1, 3, 5, 8, 0, 1, 3, 6, 9] } }, "blockColIndicesT": { "dtype": "int32", "shape": [2, 9], "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/boundary-value-partition-float32-d520-s1_input_blockColIndicesT" } } }, "totalSequenceLengthT": { "dtype": "int32", "shape": [1], "data": { "kind": "values", "values": [41] } }, "keyTotalSequenceLengthsT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [41, 34] } } }, "outputs": { "outputT": { "dtype": "float32", "shape": [2, 1, 576] } }, "tunables": { "WORKGROUP_SIZE": 32 }, "preset": "model" }, { "name": "boundary-value-owner-float32-d144-wg32-s5", "attrs": { "num_heads": 4, "kv_num_heads": 2, "sparse_block_size": 16 }, "inputs": { "queryT": { "dtype": "float32", "shape": [2, 5, 576], "data": { "kind": "fillFloat32", "sinStep": 0.13, "cosStep": 0.29, "scale": 0.5 } }, "keyT": { "dtype": "float32", "shape": [2, 5, 288], "data": { "kind": "fillFloat32", "sinStep": 0.17, "cosStep": 0.31, "scale": 0.5 } }, "valueT": { "dtype": "float32", "shape": [2, 5, 288], "data": { "kind": "fillFloat32", "sinStep": 0.19, "cosStep": 0.23, "scale": 0.5 } }, "pastKeyT": { "dtype": "float32", "shape": [2, 2, 64, 144], "data": { "kind": "fillFloat32", "sinStep": 0.011, "cosStep": 0.017, "scale": 0.25 } }, "pastValueT": { "dtype": "float32", "shape": [2, 2, 64, 144], "data": { "kind": "fillFloat32", "sinStep": 0.019, "cosStep": 0.013, "scale": 0.25 } }, "blockRowIndicesT": { "dtype": "int32", "shape": [2, 5], "data": { "kind": "values", "values": [0, 1, 3, 5, 8, 0, 1, 3, 6, 9] } }, "blockColIndicesT": { "dtype": "int32", "shape": [2, 9], "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/boundary-value-partition-float32-d520-s1_input_blockColIndicesT" } } }, "totalSequenceLengthT": { "dtype": "int32", "shape": [1], "data": { "kind": "values", "values": [41] } }, "keyTotalSequenceLengthsT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [41, 34] } } }, "outputs": { "outputT": { "dtype": "float32", "shape": [2, 5, 576] } }, "tunables": { "WORKGROUP_SIZE": 32 }, "preset": "model" }, { "name": "sparse-prompt-tail-b1-s127-h32kv8-d128-blk64", "preset": "model", "vars": { "dtype": "float32", "batch": 1, "seq": 127, "heads": 32, "kvHeads": 8, "headDim": 128, "qkPairs": 260096, "attendedKeys": 1024, "dtypeBytes": 4 }, "attrs": { "num_heads": 32, "kv_num_heads": 8, "sparse_block_size": 64 }, "inputs": { "queryT": { "shape": [1, 127, 4096], "dtype": "float32", "dist": "normal", "seed": 9300, "scale": 1 }, "keyT": { "shape": [1, 127, 1024], "dtype": "float32", "dist": "normal", "seed": 9301, "scale": 1 }, "valueT": { "shape": [1, 127, 1024], "dtype": "float32", "dist": "normal", "seed": 9302, "scale": 1 }, "pastKeyT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9303, "scale": 1 }, "pastValueT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9304, "scale": 1 }, "blockRowIndicesT": { "shape": [4, 17], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT" } } }, "blockColIndicesT": { "shape": [4, 112], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/block_column_indices_t_pattern" } } }, "totalSequenceLengthT": { "shape": [1], "dtype": "int32", "data": { "kind": "values", "values": [127] } }, "keyTotalSequenceLengthsT": { "shape": [1], "dtype": "int32", "dist": "constant", "value": 127 } }, "outputs": { "outputT": { "shape": [1, 127, 4096], "dtype": "float32" } }, "bench": { "metrics": [{ "type": "gflops", "value": "4 * args.qkPairs * args.headDim" }] } }, { "name": "sparse-prompt-tail-b1-s129-h32kv8-d128-blk64", "preset": "model", "vars": { "dtype": "float32", "batch": 1, "seq": 129, "heads": 32, "kvHeads": 8, "headDim": 128, "qkPairs": 268320, "attendedKeys": 1024, "dtypeBytes": 4 }, "attrs": { "num_heads": 32, "kv_num_heads": 8, "sparse_block_size": 64 }, "inputs": { "queryT": { "shape": [1, 129, 4096], "dtype": "float32", "dist": "normal", "seed": 9300, "scale": 1 }, "keyT": { "shape": [1, 129, 1024], "dtype": "float32", "dist": "normal", "seed": 9301, "scale": 1 }, "valueT": { "shape": [1, 129, 1024], "dtype": "float32", "dist": "normal", "seed": 9302, "scale": 1 }, "pastKeyT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9303, "scale": 1 }, "pastValueT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9304, "scale": 1 }, "blockRowIndicesT": { "shape": [4, 17], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT" } } }, "blockColIndicesT": { "shape": [4, 112], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/block_column_indices_t_pattern" } } }, "totalSequenceLengthT": { "shape": [1], "dtype": "int32", "data": { "kind": "values", "values": [129] } }, "keyTotalSequenceLengthsT": { "shape": [1], "dtype": "int32", "dist": "constant", "value": 129 } }, "outputs": { "outputT": { "shape": [1, 129, 4096], "dtype": "float32" } }, "bench": { "metrics": [{ "type": "gflops", "value": "4 * args.qkPairs * args.headDim" }] } }, { "name": "sparse-prompt-tail-b1-s511-h32kv8-d128-blk64", "preset": "model", "vars": { "dtype": "float32", "batch": 1, "seq": 511, "heads": 32, "kvHeads": 8, "headDim": 128, "qkPairs": 4186112, "attendedKeys": 1024, "dtypeBytes": 4 }, "attrs": { "num_heads": 32, "kv_num_heads": 8, "sparse_block_size": 64 }, "inputs": { "queryT": { "shape": [1, 511, 4096], "dtype": "float32", "dist": "normal", "seed": 9300, "scale": 1 }, "keyT": { "shape": [1, 511, 1024], "dtype": "float32", "dist": "normal", "seed": 9301, "scale": 1 }, "valueT": { "shape": [1, 511, 1024], "dtype": "float32", "dist": "normal", "seed": 9302, "scale": 1 }, "pastKeyT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9303, "scale": 1 }, "pastValueT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9304, "scale": 1 }, "blockRowIndicesT": { "shape": [4, 17], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT" } } }, "blockColIndicesT": { "shape": [4, 112], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/block_column_indices_t_pattern" } } }, "totalSequenceLengthT": { "shape": [1], "dtype": "int32", "data": { "kind": "values", "values": [511] } }, "keyTotalSequenceLengthsT": { "shape": [1], "dtype": "int32", "dist": "constant", "value": 511 } }, "outputs": { "outputT": { "shape": [1, 511, 4096], "dtype": "float32" } }, "bench": { "metrics": [{ "type": "gflops", "value": "4 * args.qkPairs * args.headDim" }] } }, { "name": "sparse-prompt-tail-b1-s513-h32kv8-d128-blk64", "preset": "model", "vars": { "dtype": "float32", "batch": 1, "seq": 513, "heads": 32, "kvHeads": 8, "headDim": 128, "qkPairs": 4217376, "attendedKeys": 1024, "dtypeBytes": 4 }, "attrs": { "num_heads": 32, "kv_num_heads": 8, "sparse_block_size": 64 }, "inputs": { "queryT": { "shape": [1, 513, 4096], "dtype": "float32", "dist": "normal", "seed": 9300, "scale": 1 }, "keyT": { "shape": [1, 513, 1024], "dtype": "float32", "dist": "normal", "seed": 9301, "scale": 1 }, "valueT": { "shape": [1, 513, 1024], "dtype": "float32", "dist": "normal", "seed": 9302, "scale": 1 }, "pastKeyT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9303, "scale": 1 }, "pastValueT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9304, "scale": 1 }, "blockRowIndicesT": { "shape": [4, 17], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT" } } }, "blockColIndicesT": { "shape": [4, 112], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/block_column_indices_t_pattern" } } }, "totalSequenceLengthT": { "shape": [1], "dtype": "int32", "data": { "kind": "values", "values": [513] } }, "keyTotalSequenceLengthsT": { "shape": [1], "dtype": "int32", "dist": "constant", "value": 513 } }, "outputs": { "outputT": { "shape": [1, 513, 4096], "dtype": "float32" } }, "bench": { "metrics": [{ "type": "gflops", "value": "4 * args.qkPairs * args.headDim" }] } }, { "name": "sparse-prompt-tail-b1-s1023-h32kv8-d128-blk64", "preset": "model", "vars": { "dtype": "float32", "batch": 1, "seq": 1023, "heads": 32, "kvHeads": 8, "headDim": 128, "qkPairs": 13234176, "attendedKeys": 1024, "dtypeBytes": 4 }, "attrs": { "num_heads": 32, "kv_num_heads": 8, "sparse_block_size": 64 }, "inputs": { "queryT": { "shape": [1, 1023, 4096], "dtype": "float32", "dist": "normal", "seed": 9300, "scale": 1 }, "keyT": { "shape": [1, 1023, 1024], "dtype": "float32", "dist": "normal", "seed": 9301, "scale": 1 }, "valueT": { "shape": [1, 1023, 1024], "dtype": "float32", "dist": "normal", "seed": 9302, "scale": 1 }, "pastKeyT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9303, "scale": 1 }, "pastValueT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9304, "scale": 1 }, "blockRowIndicesT": { "shape": [4, 17], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT" } } }, "blockColIndicesT": { "shape": [4, 112], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/block_column_indices_t_pattern" } } }, "totalSequenceLengthT": { "shape": [1], "dtype": "int32", "data": { "kind": "values", "values": [1023] } }, "keyTotalSequenceLengthsT": { "shape": [1], "dtype": "int32", "dist": "constant", "value": 1023 } }, "outputs": { "outputT": { "shape": [1, 1023, 4096], "dtype": "float32" } }, "bench": { "metrics": [{ "type": "gflops", "value": "4 * args.qkPairs * args.headDim" }] } }, { "name": "hybrid-neighbor-s65-h32kv8-d128", "preset": "model", "vars": { "dtype": "float32", "batch": 1, "seq": 65, "heads": 32, "kvHeads": 8, "headDim": 128, "qkPairs": 68640, "attendedKeys": 65, "dtypeBytes": 4 }, "attrs": { "num_heads": 32, "kv_num_heads": 8, "sparse_block_size": 64 }, "inputs": { "queryT": { "shape": [1, 65, 4096], "dtype": "float32", "dist": "normal", "seed": 9300, "scale": 1 }, "keyT": { "shape": [1, 65, 1024], "dtype": "float32", "dist": "normal", "seed": 9301, "scale": 1 }, "valueT": { "shape": [1, 65, 1024], "dtype": "float32", "dist": "normal", "seed": 9302, "scale": 1 }, "pastKeyT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9303, "scale": 1 }, "pastValueT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9304, "scale": 1 }, "blockRowIndicesT": { "shape": [4, 17], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT" } } }, "blockColIndicesT": { "shape": [4, 112], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/block_column_indices_t_pattern" } } }, "totalSequenceLengthT": { "shape": [1], "dtype": "int32", "data": { "kind": "values", "values": [65] } }, "keyTotalSequenceLengthsT": { "shape": [1], "dtype": "int32", "dist": "constant", "value": 65 } }, "outputs": { "outputT": { "shape": [1, 65, 4096], "dtype": "float32" } }, "bench": { "metrics": [{ "type": "gflops", "value": "4 * args.qkPairs * args.headDim" }] } }, { "name": "hybrid-neighbor-s66-h32kv8-d128", "preset": "model", "vars": { "dtype": "float32", "batch": 1, "seq": 66, "heads": 32, "kvHeads": 8, "headDim": 128, "qkPairs": 70752, "attendedKeys": 66, "dtypeBytes": 4 }, "attrs": { "num_heads": 32, "kv_num_heads": 8, "sparse_block_size": 64 }, "inputs": { "queryT": { "shape": [1, 66, 4096], "dtype": "float32", "dist": "normal", "seed": 9300, "scale": 1 }, "keyT": { "shape": [1, 66, 1024], "dtype": "float32", "dist": "normal", "seed": 9301, "scale": 1 }, "valueT": { "shape": [1, 66, 1024], "dtype": "float32", "dist": "normal", "seed": 9302, "scale": 1 }, "pastKeyT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9303, "scale": 1 }, "pastValueT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9304, "scale": 1 }, "blockRowIndicesT": { "shape": [4, 17], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT" } } }, "blockColIndicesT": { "shape": [4, 112], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/block_column_indices_t_pattern" } } }, "totalSequenceLengthT": { "shape": [1], "dtype": "int32", "data": { "kind": "values", "values": [66] } }, "keyTotalSequenceLengthsT": { "shape": [1], "dtype": "int32", "dist": "constant", "value": 66 } }, "outputs": { "outputT": { "shape": [1, 66, 4096], "dtype": "float32" } }, "bench": { "metrics": [{ "type": "gflops", "value": "4 * args.qkPairs * args.headDim" }] } }, { "name": "hybrid-neighbor-s67-h32kv8-d128", "preset": "model", "vars": { "dtype": "float32", "batch": 1, "seq": 67, "heads": 32, "kvHeads": 8, "headDim": 128, "qkPairs": 72896, "attendedKeys": 67, "dtypeBytes": 4 }, "attrs": { "num_heads": 32, "kv_num_heads": 8, "sparse_block_size": 64 }, "inputs": { "queryT": { "shape": [1, 67, 4096], "dtype": "float32", "dist": "normal", "seed": 9300, "scale": 1 }, "keyT": { "shape": [1, 67, 1024], "dtype": "float32", "dist": "normal", "seed": 9301, "scale": 1 }, "valueT": { "shape": [1, 67, 1024], "dtype": "float32", "dist": "normal", "seed": 9302, "scale": 1 }, "pastKeyT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9303, "scale": 1 }, "pastValueT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9304, "scale": 1 }, "blockRowIndicesT": { "shape": [4, 17], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT" } } }, "blockColIndicesT": { "shape": [4, 112], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/block_column_indices_t_pattern" } } }, "totalSequenceLengthT": { "shape": [1], "dtype": "int32", "data": { "kind": "values", "values": [67] } }, "keyTotalSequenceLengthsT": { "shape": [1], "dtype": "int32", "dist": "constant", "value": 67 } }, "outputs": { "outputT": { "shape": [1, 67, 4096], "dtype": "float32" } }, "bench": { "metrics": [{ "type": "gflops", "value": "4 * args.qkPairs * args.headDim" }] } }, { "name": "hybrid-neighbor-s68-h32kv8-d128", "preset": "model", "vars": { "dtype": "float32", "batch": 1, "seq": 68, "heads": 32, "kvHeads": 8, "headDim": 128, "qkPairs": 75072, "attendedKeys": 68, "dtypeBytes": 4 }, "attrs": { "num_heads": 32, "kv_num_heads": 8, "sparse_block_size": 64 }, "inputs": { "queryT": { "shape": [1, 68, 4096], "dtype": "float32", "dist": "normal", "seed": 9300, "scale": 1 }, "keyT": { "shape": [1, 68, 1024], "dtype": "float32", "dist": "normal", "seed": 9301, "scale": 1 }, "valueT": { "shape": [1, 68, 1024], "dtype": "float32", "dist": "normal", "seed": 9302, "scale": 1 }, "pastKeyT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9303, "scale": 1 }, "pastValueT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9304, "scale": 1 }, "blockRowIndicesT": { "shape": [4, 17], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT" } } }, "blockColIndicesT": { "shape": [4, 112], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/block_column_indices_t_pattern" } } }, "totalSequenceLengthT": { "shape": [1], "dtype": "int32", "data": { "kind": "values", "values": [68] } }, "keyTotalSequenceLengthsT": { "shape": [1], "dtype": "int32", "dist": "constant", "value": 68 } }, "outputs": { "outputT": { "shape": [1, 68, 4096], "dtype": "float32" } }, "bench": { "metrics": [{ "type": "gflops", "value": "4 * args.qkPairs * args.headDim" }] } }, { "name": "hybrid-neighbor-s130-h32kv8-d128", "preset": "model", "vars": { "dtype": "float32", "batch": 1, "seq": 130, "heads": 32, "kvHeads": 8, "headDim": 128, "qkPairs": 272480, "attendedKeys": 130, "dtypeBytes": 4 }, "attrs": { "num_heads": 32, "kv_num_heads": 8, "sparse_block_size": 64 }, "inputs": { "queryT": { "shape": [1, 130, 4096], "dtype": "float32", "dist": "normal", "seed": 9300, "scale": 1 }, "keyT": { "shape": [1, 130, 1024], "dtype": "float32", "dist": "normal", "seed": 9301, "scale": 1 }, "valueT": { "shape": [1, 130, 1024], "dtype": "float32", "dist": "normal", "seed": 9302, "scale": 1 }, "pastKeyT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9303, "scale": 1 }, "pastValueT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9304, "scale": 1 }, "blockRowIndicesT": { "shape": [4, 17], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT" } } }, "blockColIndicesT": { "shape": [4, 112], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/block_column_indices_t_pattern" } } }, "totalSequenceLengthT": { "shape": [1], "dtype": "int32", "data": { "kind": "values", "values": [130] } }, "keyTotalSequenceLengthsT": { "shape": [1], "dtype": "int32", "dist": "constant", "value": 130 } }, "outputs": { "outputT": { "shape": [1, 130, 4096], "dtype": "float32" } }, "bench": { "metrics": [{ "type": "gflops", "value": "4 * args.qkPairs * args.headDim" }] } }, { "name": "hybrid-neighbor-s131-h32kv8-d128", "preset": "model", "vars": { "dtype": "float32", "batch": 1, "seq": 131, "heads": 32, "kvHeads": 8, "headDim": 128, "qkPairs": 276672, "attendedKeys": 131, "dtypeBytes": 4 }, "attrs": { "num_heads": 32, "kv_num_heads": 8, "sparse_block_size": 64 }, "inputs": { "queryT": { "shape": [1, 131, 4096], "dtype": "float32", "dist": "normal", "seed": 9300, "scale": 1 }, "keyT": { "shape": [1, 131, 1024], "dtype": "float32", "dist": "normal", "seed": 9301, "scale": 1 }, "valueT": { "shape": [1, 131, 1024], "dtype": "float32", "dist": "normal", "seed": 9302, "scale": 1 }, "pastKeyT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9303, "scale": 1 }, "pastValueT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9304, "scale": 1 }, "blockRowIndicesT": { "shape": [4, 17], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT" } } }, "blockColIndicesT": { "shape": [4, 112], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/block_column_indices_t_pattern" } } }, "totalSequenceLengthT": { "shape": [1], "dtype": "int32", "data": { "kind": "values", "values": [131] } }, "keyTotalSequenceLengthsT": { "shape": [1], "dtype": "int32", "dist": "constant", "value": 131 } }, "outputs": { "outputT": { "shape": [1, 131, 4096], "dtype": "float32" } }, "bench": { "metrics": [{ "type": "gflops", "value": "4 * args.qkPairs * args.headDim" }] } }, { "name": "hybrid-neighbor-s132-h32kv8-d128", "preset": "model", "vars": { "dtype": "float32", "batch": 1, "seq": 132, "heads": 32, "kvHeads": 8, "headDim": 128, "qkPairs": 280896, "attendedKeys": 132, "dtypeBytes": 4 }, "attrs": { "num_heads": 32, "kv_num_heads": 8, "sparse_block_size": 64 }, "inputs": { "queryT": { "shape": [1, 132, 4096], "dtype": "float32", "dist": "normal", "seed": 9300, "scale": 1 }, "keyT": { "shape": [1, 132, 1024], "dtype": "float32", "dist": "normal", "seed": 9301, "scale": 1 }, "valueT": { "shape": [1, 132, 1024], "dtype": "float32", "dist": "normal", "seed": 9302, "scale": 1 }, "pastKeyT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9303, "scale": 1 }, "pastValueT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9304, "scale": 1 }, "blockRowIndicesT": { "shape": [4, 17], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT" } } }, "blockColIndicesT": { "shape": [4, 112], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/block_column_indices_t_pattern" } } }, "totalSequenceLengthT": { "shape": [1], "dtype": "int32", "data": { "kind": "values", "values": [132] } }, "keyTotalSequenceLengthsT": { "shape": [1], "dtype": "int32", "dist": "constant", "value": 132 } }, "outputs": { "outputT": { "shape": [1, 132, 4096], "dtype": "float32" } }, "bench": { "metrics": [{ "type": "gflops", "value": "4 * args.qkPairs * args.headDim" }] } }, { "name": "hybrid-neighbor-s512-h32kv8-d128", "preset": "model", "vars": { "dtype": "float32", "batch": 1, "seq": 512, "heads": 32, "kvHeads": 8, "headDim": 128, "qkPairs": 4202496, "attendedKeys": 512, "dtypeBytes": 4 }, "attrs": { "num_heads": 32, "kv_num_heads": 8, "sparse_block_size": 64 }, "inputs": { "queryT": { "shape": [1, 512, 4096], "dtype": "float32", "dist": "normal", "seed": 9300, "scale": 1 }, "keyT": { "shape": [1, 512, 1024], "dtype": "float32", "dist": "normal", "seed": 9301, "scale": 1 }, "valueT": { "shape": [1, 512, 1024], "dtype": "float32", "dist": "normal", "seed": 9302, "scale": 1 }, "pastKeyT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9303, "scale": 1 }, "pastValueT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9304, "scale": 1 }, "blockRowIndicesT": { "shape": [4, 17], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT" } } }, "blockColIndicesT": { "shape": [4, 112], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/block_column_indices_t_pattern" } } }, "totalSequenceLengthT": { "shape": [1], "dtype": "int32", "data": { "kind": "values", "values": [512] } }, "keyTotalSequenceLengthsT": { "shape": [1], "dtype": "int32", "dist": "constant", "value": 512 } }, "outputs": { "outputT": { "shape": [1, 512, 4096], "dtype": "float32" } }, "bench": { "metrics": [{ "type": "gflops", "value": "4 * args.qkPairs * args.headDim" }] } }, { "name": "hybrid-neighbor-s514-h32kv8-d128", "preset": "model", "vars": { "dtype": "float32", "batch": 1, "seq": 514, "heads": 32, "kvHeads": 8, "headDim": 128, "qkPairs": 4232288, "attendedKeys": 514, "dtypeBytes": 4 }, "attrs": { "num_heads": 32, "kv_num_heads": 8, "sparse_block_size": 64 }, "inputs": { "queryT": { "shape": [1, 514, 4096], "dtype": "float32", "dist": "normal", "seed": 9300, "scale": 1 }, "keyT": { "shape": [1, 514, 1024], "dtype": "float32", "dist": "normal", "seed": 9301, "scale": 1 }, "valueT": { "shape": [1, 514, 1024], "dtype": "float32", "dist": "normal", "seed": 9302, "scale": 1 }, "pastKeyT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9303, "scale": 1 }, "pastValueT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9304, "scale": 1 }, "blockRowIndicesT": { "shape": [4, 17], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT" } } }, "blockColIndicesT": { "shape": [4, 112], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/block_column_indices_t_pattern" } } }, "totalSequenceLengthT": { "shape": [1], "dtype": "int32", "data": { "kind": "values", "values": [514] } }, "keyTotalSequenceLengthsT": { "shape": [1], "dtype": "int32", "dist": "constant", "value": 514 } }, "outputs": { "outputT": { "shape": [1, 514, 4096], "dtype": "float32" } }, "bench": { "metrics": [{ "type": "gflops", "value": "4 * args.qkPairs * args.headDim" }] } }, { "name": "hybrid-neighbor-s515-h32kv8-d128", "preset": "model", "vars": { "dtype": "float32", "batch": 1, "seq": 515, "heads": 32, "kvHeads": 8, "headDim": 128, "qkPairs": 4247232, "attendedKeys": 515, "dtypeBytes": 4 }, "attrs": { "num_heads": 32, "kv_num_heads": 8, "sparse_block_size": 64 }, "inputs": { "queryT": { "shape": [1, 515, 4096], "dtype": "float32", "dist": "normal", "seed": 9300, "scale": 1 }, "keyT": { "shape": [1, 515, 1024], "dtype": "float32", "dist": "normal", "seed": 9301, "scale": 1 }, "valueT": { "shape": [1, 515, 1024], "dtype": "float32", "dist": "normal", "seed": 9302, "scale": 1 }, "pastKeyT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9303, "scale": 1 }, "pastValueT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9304, "scale": 1 }, "blockRowIndicesT": { "shape": [4, 17], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT" } } }, "blockColIndicesT": { "shape": [4, 112], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/block_column_indices_t_pattern" } } }, "totalSequenceLengthT": { "shape": [1], "dtype": "int32", "data": { "kind": "values", "values": [515] } }, "keyTotalSequenceLengthsT": { "shape": [1], "dtype": "int32", "dist": "constant", "value": 515 } }, "outputs": { "outputT": { "shape": [1, 515, 4096], "dtype": "float32" } }, "bench": { "metrics": [{ "type": "gflops", "value": "4 * args.qkPairs * args.headDim" }] } }, { "name": "hybrid-neighbor-s516-h32kv8-d128", "preset": "model", "vars": { "dtype": "float32", "batch": 1, "seq": 516, "heads": 32, "kvHeads": 8, "headDim": 128, "qkPairs": 4262208, "attendedKeys": 516, "dtypeBytes": 4 }, "attrs": { "num_heads": 32, "kv_num_heads": 8, "sparse_block_size": 64 }, "inputs": { "queryT": { "shape": [1, 516, 4096], "dtype": "float32", "dist": "normal", "seed": 9300, "scale": 1 }, "keyT": { "shape": [1, 516, 1024], "dtype": "float32", "dist": "normal", "seed": 9301, "scale": 1 }, "valueT": { "shape": [1, 516, 1024], "dtype": "float32", "dist": "normal", "seed": 9302, "scale": 1 }, "pastKeyT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9303, "scale": 1 }, "pastValueT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9304, "scale": 1 }, "blockRowIndicesT": { "shape": [4, 17], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT" } } }, "blockColIndicesT": { "shape": [4, 112], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/block_column_indices_t_pattern" } } }, "totalSequenceLengthT": { "shape": [1], "dtype": "int32", "data": { "kind": "values", "values": [516] } }, "keyTotalSequenceLengthsT": { "shape": [1], "dtype": "int32", "dist": "constant", "value": 516 } }, "outputs": { "outputT": { "shape": [1, 516, 4096], "dtype": "float32" } }, "bench": { "metrics": [{ "type": "gflops", "value": "4 * args.qkPairs * args.headDim" }] } }, { "name": "hybrid-neighbor-s517-h32kv8-d128", "preset": "model", "vars": { "dtype": "float32", "batch": 1, "seq": 517, "heads": 32, "kvHeads": 8, "headDim": 128, "qkPairs": 4277216, "attendedKeys": 517, "dtypeBytes": 4 }, "attrs": { "num_heads": 32, "kv_num_heads": 8, "sparse_block_size": 64 }, "inputs": { "queryT": { "shape": [1, 517, 4096], "dtype": "float32", "dist": "normal", "seed": 9300, "scale": 1 }, "keyT": { "shape": [1, 517, 1024], "dtype": "float32", "dist": "normal", "seed": 9301, "scale": 1 }, "valueT": { "shape": [1, 517, 1024], "dtype": "float32", "dist": "normal", "seed": 9302, "scale": 1 }, "pastKeyT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9303, "scale": 1 }, "pastValueT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9304, "scale": 1 }, "blockRowIndicesT": { "shape": [4, 17], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT" } } }, "blockColIndicesT": { "shape": [4, 112], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/block_column_indices_t_pattern" } } }, "totalSequenceLengthT": { "shape": [1], "dtype": "int32", "data": { "kind": "values", "values": [517] } }, "keyTotalSequenceLengthsT": { "shape": [1], "dtype": "int32", "dist": "constant", "value": 517 } }, "outputs": { "outputT": { "shape": [1, 517, 4096], "dtype": "float32" } }, "bench": { "metrics": [{ "type": "gflops", "value": "4 * args.qkPairs * args.headDim" }] } }, { "name": "hybrid-neighbor-s577-h32kv8-d128", "preset": "model", "vars": { "dtype": "float32", "batch": 1, "seq": 577, "heads": 32, "kvHeads": 8, "headDim": 128, "qkPairs": 5234720, "attendedKeys": 577, "dtypeBytes": 4 }, "attrs": { "num_heads": 32, "kv_num_heads": 8, "sparse_block_size": 64 }, "inputs": { "queryT": { "shape": [1, 577, 4096], "dtype": "float32", "dist": "normal", "seed": 9300, "scale": 1 }, "keyT": { "shape": [1, 577, 1024], "dtype": "float32", "dist": "normal", "seed": 9301, "scale": 1 }, "valueT": { "shape": [1, 577, 1024], "dtype": "float32", "dist": "normal", "seed": 9302, "scale": 1 }, "pastKeyT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9303, "scale": 1 }, "pastValueT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9304, "scale": 1 }, "blockRowIndicesT": { "shape": [4, 17], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT" } } }, "blockColIndicesT": { "shape": [4, 112], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/block_column_indices_t_pattern" } } }, "totalSequenceLengthT": { "shape": [1], "dtype": "int32", "data": { "kind": "values", "values": [577] } }, "keyTotalSequenceLengthsT": { "shape": [1], "dtype": "int32", "dist": "constant", "value": 577 } }, "outputs": { "outputT": { "shape": [1, 577, 4096], "dtype": "float32" } }, "bench": { "metrics": [{ "type": "gflops", "value": "4 * args.qkPairs * args.headDim" }] } }, { "name": "isolate-packed-rotary-prompt-s65", "preset": "model", "vars": { "dtype": "float32", "batch": 1, "seq": 65, "heads": 32, "kvHeads": 8, "headDim": 128, "qkPairs": 68640, "attendedKeys": 65, "dtypeBytes": 4 }, "attrs": { "num_heads": 32, "kv_num_heads": 8, "sparse_block_size": 64, "do_rotary": 1 }, "inputs": { "queryT": { "shape": [1, 65, 6144], "dtype": "float32", "dist": "normal", "seed": 9330, "scale": 1 }, "pastKeyT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9333, "scale": 1 }, "pastValueT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9334, "scale": 1 }, "blockRowIndicesT": { "shape": [4, 17], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT" } } }, "blockColIndicesT": { "shape": [4, 112], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/block_column_indices_t_pattern" } } }, "totalSequenceLengthT": { "shape": [1], "dtype": "int32", "data": { "kind": "values", "values": [65] } }, "keyTotalSequenceLengthsT": { "shape": [1], "dtype": "int32", "dist": "constant", "value": 65 }, "cosCacheT": { "shape": [1024, 64], "dtype": "float32", "dist": "normal", "seed": 9335, "scale": 1 }, "sinCacheT": { "shape": [1024, 64], "dtype": "float32", "dist": "normal", "seed": 9336, "scale": 1 } }, "outputs": { "outputT": { "shape": [1, 65, 4096], "dtype": "float32" } }, "bench": { "metrics": [{ "type": "gflops", "value": "4 * args.qkPairs * args.headDim" }] } }, { "name": "isolate-separate-history-s65", "preset": "model", "vars": { "dtype": "float32", "batch": 1, "seq": 65, "heads": 32, "kvHeads": 8, "headDim": 128, "qkPairs": 1264000, "attendedKeys": 1020, "dtypeBytes": 4 }, "attrs": { "num_heads": 32, "kv_num_heads": 8, "sparse_block_size": 64 }, "inputs": { "queryT": { "shape": [1, 65, 4096], "dtype": "float32", "dist": "normal", "seed": 9300, "scale": 1 }, "keyT": { "shape": [1, 65, 1024], "dtype": "float32", "dist": "normal", "seed": 9301, "scale": 1 }, "valueT": { "shape": [1, 65, 1024], "dtype": "float32", "dist": "normal", "seed": 9302, "scale": 1 }, "pastKeyT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9303, "scale": 1 }, "pastValueT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9304, "scale": 1 }, "blockRowIndicesT": { "shape": [4, 17], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT" } } }, "blockColIndicesT": { "shape": [4, 112], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/block_column_indices_t_pattern" } } }, "totalSequenceLengthT": { "shape": [1], "dtype": "int32", "data": { "kind": "values", "values": [1020] } }, "keyTotalSequenceLengthsT": { "shape": [1], "dtype": "int32", "dist": "constant", "value": 1020 } }, "outputs": { "outputT": { "shape": [1, 65, 4096], "dtype": "float32" } }, "bench": { "metrics": [{ "type": "gflops", "value": "4 * args.qkPairs * args.headDim" }] } }, { "name": "isolate-separate-history-s129", "preset": "model", "vars": { "dtype": "float32", "batch": 1, "seq": 129, "heads": 32, "kvHeads": 8, "headDim": 128, "qkPairs": 2474880, "attendedKeys": 1020, "dtypeBytes": 4 }, "attrs": { "num_heads": 32, "kv_num_heads": 8, "sparse_block_size": 64 }, "inputs": { "queryT": { "shape": [1, 129, 4096], "dtype": "float32", "dist": "normal", "seed": 9300, "scale": 1 }, "keyT": { "shape": [1, 129, 1024], "dtype": "float32", "dist": "normal", "seed": 9301, "scale": 1 }, "valueT": { "shape": [1, 129, 1024], "dtype": "float32", "dist": "normal", "seed": 9302, "scale": 1 }, "pastKeyT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9303, "scale": 1 }, "pastValueT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9304, "scale": 1 }, "blockRowIndicesT": { "shape": [4, 17], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT" } } }, "blockColIndicesT": { "shape": [4, 112], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/block_column_indices_t_pattern" } } }, "totalSequenceLengthT": { "shape": [1], "dtype": "int32", "data": { "kind": "values", "values": [1020] } }, "keyTotalSequenceLengthsT": { "shape": [1], "dtype": "int32", "dist": "constant", "value": 1020 } }, "outputs": { "outputT": { "shape": [1, 129, 4096], "dtype": "float32" } }, "bench": { "metrics": [{ "type": "gflops", "value": "4 * args.qkPairs * args.headDim" }] } }, { "name": "isolate-separate-history-s513", "preset": "model", "vars": { "dtype": "float32", "batch": 1, "seq": 513, "heads": 32, "kvHeads": 8, "headDim": 128, "qkPairs": 9052032, "attendedKeys": 1020, "dtypeBytes": 4 }, "attrs": { "num_heads": 32, "kv_num_heads": 8, "sparse_block_size": 64 }, "inputs": { "queryT": { "shape": [1, 513, 4096], "dtype": "float32", "dist": "normal", "seed": 9300, "scale": 1 }, "keyT": { "shape": [1, 513, 1024], "dtype": "float32", "dist": "normal", "seed": 9301, "scale": 1 }, "valueT": { "shape": [1, 513, 1024], "dtype": "float32", "dist": "normal", "seed": 9302, "scale": 1 }, "pastKeyT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9303, "scale": 1 }, "pastValueT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9304, "scale": 1 }, "blockRowIndicesT": { "shape": [4, 17], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT" } } }, "blockColIndicesT": { "shape": [4, 112], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/block_column_indices_t_pattern" } } }, "totalSequenceLengthT": { "shape": [1], "dtype": "int32", "data": { "kind": "values", "values": [1020] } }, "keyTotalSequenceLengthsT": { "shape": [1], "dtype": "int32", "dist": "constant", "value": 1020 } }, "outputs": { "outputT": { "shape": [1, 513, 4096], "dtype": "float32" } }, "bench": { "metrics": [{ "type": "gflops", "value": "4 * args.qkPairs * args.headDim" }] } }, { "name": "hybrid-packed-rotary-s65-past955", "preset": "model", "vars": { "dtype": "float32", "batch": 1, "seq": 65, "heads": 32, "kvHeads": 8, "headDim": 128, "qkPairs": 1264000, "attendedKeys": 1020, "dtypeBytes": 4 }, "attrs": { "num_heads": 32, "kv_num_heads": 8, "sparse_block_size": 64, "do_rotary": 1 }, "inputs": { "queryT": { "shape": [1, 65, 6144], "dtype": "float32", "dist": "normal", "seed": 9330, "scale": 1 }, "pastKeyT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9333, "scale": 1 }, "pastValueT": { "shape": [1, 8, 1024, 128], "dtype": "float32", "dist": "normal", "seed": 9334, "scale": 1 }, "blockRowIndicesT": { "shape": [4, 17], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT" } } }, "blockColIndicesT": { "shape": [4, 112], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/block_column_indices_t_pattern" } } }, "totalSequenceLengthT": { "shape": [1], "dtype": "int32", "data": { "kind": "values", "values": [1020] } }, "keyTotalSequenceLengthsT": { "shape": [1], "dtype": "int32", "dist": "constant", "value": 1020 }, "cosCacheT": { "shape": [1024, 64], "dtype": "float32", "dist": "normal", "seed": 9335, "scale": 1 }, "sinCacheT": { "shape": [1024, 64], "dtype": "float32", "dist": "normal", "seed": 9336, "scale": 1 } }, "outputs": { "outputT": { "shape": [1, 65, 4096], "dtype": "float32" } }, "bench": { "metrics": [{ "type": "gflops", "value": "4 * args.qkPairs * args.headDim" }] } }, { "name": "hybrid-d96-s131", "preset": "model", "vars": { "dtype": "float32", "batch": 1, "seq": 131, "heads": 32, "kvHeads": 8, "headDim": 96, "qkPairs": 276672, "attendedKeys": 131, "dtypeBytes": 4 }, "attrs": { "num_heads": 32, "kv_num_heads": 8, "sparse_block_size": 64 }, "inputs": { "queryT": { "shape": [1, 131, 3072], "dtype": "float32", "dist": "normal", "seed": 9300, "scale": 1 }, "keyT": { "shape": [1, 131, 768], "dtype": "float32", "dist": "normal", "seed": 9301, "scale": 1 }, "valueT": { "shape": [1, 131, 768], "dtype": "float32", "dist": "normal", "seed": 9302, "scale": 1 }, "pastKeyT": { "shape": [1, 8, 1024, 96], "dtype": "float32", "dist": "normal", "seed": 9303, "scale": 1 }, "pastValueT": { "shape": [1, 8, 1024, 96], "dtype": "float32", "dist": "normal", "seed": 9304, "scale": 1 }, "blockRowIndicesT": { "shape": [4, 17], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT" } } }, "blockColIndicesT": { "shape": [4, 112], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/block_column_indices_t_pattern" } } }, "totalSequenceLengthT": { "shape": [1], "dtype": "int32", "data": { "kind": "values", "values": [131] } }, "keyTotalSequenceLengthsT": { "shape": [1], "dtype": "int32", "dist": "constant", "value": 131 } }, "outputs": { "outputT": { "shape": [1, 131, 3072], "dtype": "float32" } }, "bench": { "metrics": [{ "type": "gflops", "value": "4 * args.qkPairs * args.headDim" }] } }, { "name": "hybrid-d32-s131", "preset": "model", "vars": { "dtype": "float32", "batch": 1, "seq": 131, "heads": 32, "kvHeads": 8, "headDim": 32, "qkPairs": 276672, "attendedKeys": 131, "dtypeBytes": 4 }, "attrs": { "num_heads": 32, "kv_num_heads": 8, "sparse_block_size": 64 }, "inputs": { "queryT": { "shape": [1, 131, 1024], "dtype": "float32", "dist": "normal", "seed": 9300, "scale": 1 }, "keyT": { "shape": [1, 131, 256], "dtype": "float32", "dist": "normal", "seed": 9301, "scale": 1 }, "valueT": { "shape": [1, 131, 256], "dtype": "float32", "dist": "normal", "seed": 9302, "scale": 1 }, "pastKeyT": { "shape": [1, 8, 1024, 32], "dtype": "float32", "dist": "normal", "seed": 9303, "scale": 1 }, "pastValueT": { "shape": [1, 8, 1024, 32], "dtype": "float32", "dist": "normal", "seed": 9304, "scale": 1 }, "blockRowIndicesT": { "shape": [4, 17], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT" } } }, "blockColIndicesT": { "shape": [4, 112], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/block_column_indices_t_pattern" } } }, "totalSequenceLengthT": { "shape": [1], "dtype": "int32", "data": { "kind": "values", "values": [131] } }, "keyTotalSequenceLengthsT": { "shape": [1], "dtype": "int32", "dist": "constant", "value": 131 } }, "outputs": { "outputT": { "shape": [1, 131, 1024], "dtype": "float32" } }, "bench": { "metrics": [{ "type": "gflops", "value": "4 * args.qkPairs * args.headDim" }] } }, { "name": "hybrid-d64-s131", "preset": "model", "vars": { "dtype": "float32", "batch": 1, "seq": 131, "heads": 32, "kvHeads": 8, "headDim": 64, "qkPairs": 276672, "attendedKeys": 131, "dtypeBytes": 4 }, "attrs": { "num_heads": 32, "kv_num_heads": 8, "sparse_block_size": 64 }, "inputs": { "queryT": { "shape": [1, 131, 2048], "dtype": "float32", "dist": "normal", "seed": 9300, "scale": 1 }, "keyT": { "shape": [1, 131, 512], "dtype": "float32", "dist": "normal", "seed": 9301, "scale": 1 }, "valueT": { "shape": [1, 131, 512], "dtype": "float32", "dist": "normal", "seed": 9302, "scale": 1 }, "pastKeyT": { "shape": [1, 8, 1024, 64], "dtype": "float32", "dist": "normal", "seed": 9303, "scale": 1 }, "pastValueT": { "shape": [1, 8, 1024, 64], "dtype": "float32", "dist": "normal", "seed": 9304, "scale": 1 }, "blockRowIndicesT": { "shape": [4, 17], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT" } } }, "blockColIndicesT": { "shape": [4, 112], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/block_column_indices_t_pattern" } } }, "totalSequenceLengthT": { "shape": [1], "dtype": "int32", "data": { "kind": "values", "values": [131] } }, "keyTotalSequenceLengthsT": { "shape": [1], "dtype": "int32", "dist": "constant", "value": 131 } }, "outputs": { "outputT": { "shape": [1, 131, 2048], "dtype": "float32" } }, "bench": { "metrics": [{ "type": "gflops", "value": "4 * args.qkPairs * args.headDim" }] } }, { "name": "hybrid-d32-s131-narrow", "preset": "model", "vars": { "dtype": "float32", "batch": 1, "seq": 131, "heads": 32, "kvHeads": 8, "headDim": 32, "qkPairs": 276672, "attendedKeys": 131, "dtypeBytes": 4 }, "attrs": { "num_heads": 32, "kv_num_heads": 8, "sparse_block_size": 64 }, "inputs": { "queryT": { "shape": [1, 131, 1024], "dtype": "float32", "dist": "normal", "seed": 9300, "scale": 1 }, "keyT": { "shape": [1, 131, 256], "dtype": "float32", "dist": "normal", "seed": 9301, "scale": 1 }, "valueT": { "shape": [1, 131, 256], "dtype": "float32", "dist": "normal", "seed": 9302, "scale": 1 }, "pastKeyT": { "shape": [1, 8, 1024, 32], "dtype": "float32", "dist": "normal", "seed": 9303, "scale": 1 }, "pastValueT": { "shape": [1, 8, 1024, 32], "dtype": "float32", "dist": "normal", "seed": 9304, "scale": 1 }, "blockRowIndicesT": { "shape": [4, 17], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/sparse-prompt-b1-s1024-h32kv8-d128-blk64_input_blockRowIndicesT" } } }, "blockColIndicesT": { "shape": [4, 112], "dtype": "int32", "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/block_column_indices_t_pattern" } } }, "totalSequenceLengthT": { "shape": [1], "dtype": "int32", "data": { "kind": "values", "values": [131] } }, "keyTotalSequenceLengthsT": { "shape": [1], "dtype": "int32", "dist": "constant", "value": 131 } }, "outputs": { "outputT": { "shape": [1, 131, 1024], "dtype": "float32" } }, "bench": { "metrics": [{ "type": "gflops", "value": "4 * args.qkPairs * args.headDim" }] }, "tunables": { "NARROW_MIN_WORKGROUPS": 1 } } ] }