diff --git "a/build/webgpu/test.json" "b/build/webgpu/test.json" new file mode 100644--- /dev/null +++ "b/build/webgpu/test.json" @@ -0,0 +1,18534 @@ +{ + "fixtureArrays": { + "ort_packed_qkv_batch2_headsize32_output_outputT": [-0.89160156, 0.02633667, -0.84863281, 1.4345703, -2.9199219, -0.20654297, 1.859375, 0.45678711, 0.076782227, 0.044921875, 0.67333984, 0.97119141, 1.5351562, -1.1025391, 0.86572266, -0.34765625, -1.2431641, -1.8027344, -1.0126953, -0.71533203, 0.77832031, -0.59228516, -2.0039062, 3.1328125, -1.2080078, 2.015625, 0.82666016, 2.0605469, -0.46264648, 1.5634766, 4.3164062, -1.03125, 0.21158576, -1.9827913, 0.45935997, -0.4045772, 0.04772174, 0.22094353, 0.71238005, -1.20860445, -0.56270063, -1.10830855, 0.14455934, 2.87315512, -1.14114404, 0.66515017, 0.16263856, -1.75517499, 0.77650774, -0.44058144, 0.31942442, 1.51513433, 0.41078627, 0.41566271, 1.74393702, 0.51457298, 0.78953874, -3.1088841, -0.47052401, -0.75475156, 0.90861011, -1.82471263, 2.04898596, -0.67790967, 1.17194295, -2.17825341, -0.02712549, -1.53178656, -0.48020893, -0.00040733, 0.770356, -2.0638032, -0.53738528, -1.89084208, 0.19988713, 2.60725045, -1.06034219, 0.82412785, 0.37603328, -2.78852081, 1.08301103, -1.26178384, 0.1630473, 3.16210985, 1.36142719, 1.32916999, 2.69524455, 1.45106804, 0.6715064, -4.3170352, -1.34025633, -1.59496248, 0.37821823, -2.85797405, 3.11096096, -0.17414713], + "packed_bias_attn_bias_long_b2_h32_input_tokenOffsetT": [0, 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, 40, 41, 42, 43, 44, 45, 46, 33, 34, 35, 36, 37, 38, 39, 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], + "pmha_tile_reuse_float32_tokenOffsetT": [0, 129, 258, 387, 516, 517, 518, 519, 520, 521, 522, 523, 524, 525, 526, 527, 528, 529, 530, 531, 532, 533, 534, 535, 536, 537, 538, 539, 540, 541, 542, 543, 544, 545, 546, 547, 548, 549, 550, 551, 552, 553, 554, 555, 556, 557, 558, 559, 560, 561, 562, 563, 564, 565, 566, 567, 568, 569, 570, 571, 572, 573, 574, 575, 576, 577, 578, 579, 580, 581, 582, 583, 584, 585, 586, 587, 588, 589, 590, 591, 592, 593, 594, 595, 596, 597, 598, 599, 600, 601, 602, 603, 604, 605, 606, 607, 608, 609, 610, 611, 612, 613, 614, 615, 616, 617, 618, 619, 620, 621, 622, 623, 624, 625, 626, 627, 628, 629, 630, 631, 632, 633, 634, 635, 636, 637, 638, 639, 640, 641, 642, 643, 644, 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, 126, 127, 128, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 324, 325, 326, 327, 328, 329, 330, 331, 332, 333, 334, 335, 336, 337, 338, 339, 340, 341, 342, 343, 344, 345, 346, 347, 348, 349, 350, 351, 352, 353, 354, 355, 356, 357, 358, 359, 360, 361, 362, 363, 364, 365, 366, 367, 368, 369, 370, 371, 372, 373, 374, 375, 376, 377, 378, 379, 380, 381, 382, 383, 384, 385, 386, 388, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 405, 406, 407, 408, 409, 410, 411, 412, 413, 414, 415, 416, 417, 418, 419, 420, 421, 422, 423, 424, 425, 426, 427, 428, 429, 430, 431, 432, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 470, 471, 472, 473, 474, 475, 476, 477, 478, 479, 480, 481, 482, 483, 484, 485, 486, 487, 488, 489, 490, 491, 492, 493, 494, 495, 496, 497, 498, 499, 500, 501, 502, 503, 504, 505, 506, 507, 508, 509, 510, 511, 512, 513, 514, 515], + "packed_schedule_3_tokenOffsetT": [0, 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, 65, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 63, 64, 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, 126, 127, 128, 129], + "packed_schedule_4_tokenOffsetT": [0, 15, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29], + "packed_schedule_5_tokenOffsetT": [0, 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, 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, 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], + "packed_schedule_6_tokenOffsetT": [31, 32, 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, 0, 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, 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, 124, 125, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154], + "packed_schedule_7_tokenOffsetT": [0, 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], + "packed_schedule_8_cumulativeSequenceLengthT": [0, 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], + "packed_schedule_9_tokenOffsetT": [0, 129, 258, 387, 516, 645, 774, 903, 1032, 1161, 1290, 1419, 1548, 1677, 1806, 1935, 2064, 2193, 2322, 2451, 2580, 2709, 2838, 2967, 3096, 3225, 3354, 3483, 3612, 3741, 3870, 3999, 4000, 4001, 4002, 4003, 4004, 4005, 4006, 4007, 4008, 4009, 4010, 4011, 4012, 4013, 4014, 4015, 4016, 4017, 4018, 4019, 4020, 4021, 4022, 4023, 4024, 4025, 4026, 4027, 4028, 4029, 4030, 4031, 4032, 4033, 4034, 4035, 4036, 4037, 4038, 4039, 4040, 4041, 4042, 4043, 4044, 4045, 4046, 4047, 4048, 4049, 4050, 4051, 4052, 4053, 4054, 4055, 4056, 4057, 4058, 4059, 4060, 4061, 4062, 4063, 4064, 4065, 4066, 4067, 4068, 4069, 4070, 4071, 4072, 4073, 4074, 4075, 4076, 4077, 4078, 4079, 4080, 4081, 4082, 4083, 4084, 4085, 4086, 4087, 4088, 4089, 4090, 4091, 4092, 4093, 4094, 4095, 4096, 4097, 4098, 4099, 4100, 4101, 4102, 4103, 4104, 4105, 4106, 4107, 4108, 4109, 4110, 4111, 4112, 4113, 4114, 4115, 4116, 4117, 4118, 4119, 4120, 4121, 4122, 4123, 4124, 4125, 4126, 4127, 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, 126, 127, 128, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 324, 325, 326, 327, 328, 329, 330, 331, 332, 333, 334, 335, 336, 337, 338, 339, 340, 341, 342, 343, 344, 345, 346, 347, 348, 349, 350, 351, 352, 353, 354, 355, 356, 357, 358, 359, 360, 361, 362, 363, 364, 365, 366, 367, 368, 369, 370, 371, 372, 373, 374, 375, 376, 377, 378, 379, 380, 381, 382, 383, 384, 385, 386, 388, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 405, 406, 407, 408, 409, 410, 411, 412, 413, 414, 415, 416, 417, 418, 419, 420, 421, 422, 423, 424, 425, 426, 427, 428, 429, 430, 431, 432, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 470, 471, 472, 473, 474, 475, 476, 477, 478, 479, 480, 481, 482, 483, 484, 485, 486, 487, 488, 489, 490, 491, 492, 493, 494, 495, 496, 497, 498, 499, 500, 501, 502, 503, 504, 505, 506, 507, 508, 509, 510, 511, 512, 513, 514, 515, 517, 518, 519, 520, 521, 522, 523, 524, 525, 526, 527, 528, 529, 530, 531, 532, 533, 534, 535, 536, 537, 538, 539, 540, 541, 542, 543, 544, 545, 546, 547, 548, 549, 550, 551, 552, 553, 554, 555, 556, 557, 558, 559, 560, 561, 562, 563, 564, 565, 566, 567, 568, 569, 570, 571, 572, 573, 574, 575, 576, 577, 578, 579, 580, 581, 582, 583, 584, 585, 586, 587, 588, 589, 590, 591, 592, 593, 594, 595, 596, 597, 598, 599, 600, 601, 602, 603, 604, 605, 606, 607, 608, 609, 610, 611, 612, 613, 614, 615, 616, 617, 618, 619, 620, 621, 622, 623, 624, 625, 626, 627, 628, 629, 630, 631, 632, 633, 634, 635, 636, 637, 638, 639, 640, 641, 642, 643, 644, 646, 647, 648, 649, 650, 651, 652, 653, 654, 655, 656, 657, 658, 659, 660, 661, 662, 663, 664, 665, 666, 667, 668, 669, 670, 671, 672, 673, 674, 675, 676, 677, 678, 679, 680, 681, 682, 683, 684, 685, 686, 687, 688, 689, 690, 691, 692, 693, 694, 695, 696, 697, 698, 699, 700, 701, 702, 703, 704, 705, 706, 707, 708, 709, 710, 711, 712, 713, 714, 715, 716, 717, 718, 719, 720, 721, 722, 723, 724, 725, 726, 727, 728, 729, 730, 731, 732, 733, 734, 735, 736, 737, 738, 739, 740, 741, 742, 743, 744, 745, 746, 747, 748, 749, 750, 751, 752, 753, 754, 755, 756, 757, 758, 759, 760, 761, 762, 763, 764, 765, 766, 767, 768, 769, 770, 771, 772, 773, 775, 776, 777, 778, 779, 780, 781, 782, 783, 784, 785, 786, 787, 788, 789, 790, 791, 792, 793, 794, 795, 796, 797, 798, 799, 800, 801, 802, 803, 804, 805, 806, 807, 808, 809, 810, 811, 812, 813, 814, 815, 816, 817, 818, 819, 820, 821, 822, 823, 824, 825, 826, 827, 828, 829, 830, 831, 832, 833, 834, 835, 836, 837, 838, 839, 840, 841, 842, 843, 844, 845, 846, 847, 848, 849, 850, 851, 852, 853, 854, 855, 856, 857, 858, 859, 860, 861, 862, 863, 864, 865, 866, 867, 868, 869, 870, 871, 872, 873, 874, 875, 876, 877, 878, 879, 880, 881, 882, 883, 884, 885, 886, 887, 888, 889, 890, 891, 892, 893, 894, 895, 896, 897, 898, 899, 900, 901, 902, 904, 905, 906, 907, 908, 909, 910, 911, 912, 913, 914, 915, 916, 917, 918, 919, 920, 921, 922, 923, 924, 925, 926, 927, 928, 929, 930, 931, 932, 933, 934, 935, 936, 937, 938, 939, 940, 941, 942, 943, 944, 945, 946, 947, 948, 949, 950, 951, 952, 953, 954, 955, 956, 957, 958, 959, 960, 961, 962, 963, 964, 965, 966, 967, 968, 969, 970, 971, 972, 973, 974, 975, 976, 977, 978, 979, 980, 981, 982, 983, 984, 985, 986, 987, 988, 989, 990, 991, 992, 993, 994, 995, 996, 997, 998, 999, 1000, 1001, 1002, 1003, 1004, 1005, 1006, 1007, 1008, 1009, 1010, 1011, 1012, 1013, 1014, 1015, 1016, 1017, 1018, 1019, 1020, 1021, 1022, 1023, 1024, 1025, 1026, 1027, 1028, 1029, 1030, 1031, 1033, 1034, 1035, 1036, 1037, 1038, 1039, 1040, 1041, 1042, 1043, 1044, 1045, 1046, 1047, 1048, 1049, 1050, 1051, 1052, 1053, 1054, 1055, 1056, 1057, 1058, 1059, 1060, 1061, 1062, 1063, 1064, 1065, 1066, 1067, 1068, 1069, 1070, 1071, 1072, 1073, 1074, 1075, 1076, 1077, 1078, 1079, 1080, 1081, 1082, 1083, 1084, 1085, 1086, 1087, 1088, 1089, 1090, 1091, 1092, 1093, 1094, 1095, 1096, 1097, 1098, 1099, 1100, 1101, 1102, 1103, 1104, 1105, 1106, 1107, 1108, 1109, 1110, 1111, 1112, 1113, 1114, 1115, 1116, 1117, 1118, 1119, 1120, 1121, 1122, 1123, 1124, 1125, 1126, 1127, 1128, 1129, 1130, 1131, 1132, 1133, 1134, 1135, 1136, 1137, 1138, 1139, 1140, 1141, 1142, 1143, 1144, 1145, 1146, 1147, 1148, 1149, 1150, 1151, 1152, 1153, 1154, 1155, 1156, 1157, 1158, 1159, 1160, 1162, 1163, 1164, 1165, 1166, 1167, 1168, 1169, 1170, 1171, 1172, 1173, 1174, 1175, 1176, 1177, 1178, 1179, 1180, 1181, 1182, 1183, 1184, 1185, 1186, 1187, 1188, 1189, 1190, 1191, 1192, 1193, 1194, 1195, 1196, 1197, 1198, 1199, 1200, 1201, 1202, 1203, 1204, 1205, 1206, 1207, 1208, 1209, 1210, 1211, 1212, 1213, 1214, 1215, 1216, 1217, 1218, 1219, 1220, 1221, 1222, 1223, 1224, 1225, 1226, 1227, 1228, 1229, 1230, 1231, 1232, 1233, 1234, 1235, 1236, 1237, 1238, 1239, 1240, 1241, 1242, 1243, 1244, 1245, 1246, 1247, 1248, 1249, 1250, 1251, 1252, 1253, 1254, 1255, 1256, 1257, 1258, 1259, 1260, 1261, 1262, 1263, 1264, 1265, 1266, 1267, 1268, 1269, 1270, 1271, 1272, 1273, 1274, 1275, 1276, 1277, 1278, 1279, 1280, 1281, 1282, 1283, 1284, 1285, 1286, 1287, 1288, 1289, 1291, 1292, 1293, 1294, 1295, 1296, 1297, 1298, 1299, 1300, 1301, 1302, 1303, 1304, 1305, 1306, 1307, 1308, 1309, 1310, 1311, 1312, 1313, 1314, 1315, 1316, 1317, 1318, 1319, 1320, 1321, 1322, 1323, 1324, 1325, 1326, 1327, 1328, 1329, 1330, 1331, 1332, 1333, 1334, 1335, 1336, 1337, 1338, 1339, 1340, 1341, 1342, 1343, 1344, 1345, 1346, 1347, 1348, 1349, 1350, 1351, 1352, 1353, 1354, 1355, 1356, 1357, 1358, 1359, 1360, 1361, 1362, 1363, 1364, 1365, 1366, 1367, 1368, 1369, 1370, 1371, 1372, 1373, 1374, 1375, 1376, 1377, 1378, 1379, 1380, 1381, 1382, 1383, 1384, 1385, 1386, 1387, 1388, 1389, 1390, 1391, 1392, 1393, 1394, 1395, 1396, 1397, 1398, 1399, 1400, 1401, 1402, 1403, 1404, 1405, 1406, 1407, 1408, 1409, 1410, 1411, 1412, 1413, 1414, 1415, 1416, 1417, 1418, 1420, 1421, 1422, 1423, 1424, 1425, 1426, 1427, 1428, 1429, 1430, 1431, 1432, 1433, 1434, 1435, 1436, 1437, 1438, 1439, 1440, 1441, 1442, 1443, 1444, 1445, 1446, 1447, 1448, 1449, 1450, 1451, 1452, 1453, 1454, 1455, 1456, 1457, 1458, 1459, 1460, 1461, 1462, 1463, 1464, 1465, 1466, 1467, 1468, 1469, 1470, 1471, 1472, 1473, 1474, 1475, 1476, 1477, 1478, 1479, 1480, 1481, 1482, 1483, 1484, 1485, 1486, 1487, 1488, 1489, 1490, 1491, 1492, 1493, 1494, 1495, 1496, 1497, 1498, 1499, 1500, 1501, 1502, 1503, 1504, 1505, 1506, 1507, 1508, 1509, 1510, 1511, 1512, 1513, 1514, 1515, 1516, 1517, 1518, 1519, 1520, 1521, 1522, 1523, 1524, 1525, 1526, 1527, 1528, 1529, 1530, 1531, 1532, 1533, 1534, 1535, 1536, 1537, 1538, 1539, 1540, 1541, 1542, 1543, 1544, 1545, 1546, 1547, 1549, 1550, 1551, 1552, 1553, 1554, 1555, 1556, 1557, 1558, 1559, 1560, 1561, 1562, 1563, 1564, 1565, 1566, 1567, 1568, 1569, 1570, 1571, 1572, 1573, 1574, 1575, 1576, 1577, 1578, 1579, 1580, 1581, 1582, 1583, 1584, 1585, 1586, 1587, 1588, 1589, 1590, 1591, 1592, 1593, 1594, 1595, 1596, 1597, 1598, 1599, 1600, 1601, 1602, 1603, 1604, 1605, 1606, 1607, 1608, 1609, 1610, 1611, 1612, 1613, 1614, 1615, 1616, 1617, 1618, 1619, 1620, 1621, 1622, 1623, 1624, 1625, 1626, 1627, 1628, 1629, 1630, 1631, 1632, 1633, 1634, 1635, 1636, 1637, 1638, 1639, 1640, 1641, 1642, 1643, 1644, 1645, 1646, 1647, 1648, 1649, 1650, 1651, 1652, 1653, 1654, 1655, 1656, 1657, 1658, 1659, 1660, 1661, 1662, 1663, 1664, 1665, 1666, 1667, 1668, 1669, 1670, 1671, 1672, 1673, 1674, 1675, 1676, 1678, 1679, 1680, 1681, 1682, 1683, 1684, 1685, 1686, 1687, 1688, 1689, 1690, 1691, 1692, 1693, 1694, 1695, 1696, 1697, 1698, 1699, 1700, 1701, 1702, 1703, 1704, 1705, 1706, 1707, 1708, 1709, 1710, 1711, 1712, 1713, 1714, 1715, 1716, 1717, 1718, 1719, 1720, 1721, 1722, 1723, 1724, 1725, 1726, 1727, 1728, 1729, 1730, 1731, 1732, 1733, 1734, 1735, 1736, 1737, 1738, 1739, 1740, 1741, 1742, 1743, 1744, 1745, 1746, 1747, 1748, 1749, 1750, 1751, 1752, 1753, 1754, 1755, 1756, 1757, 1758, 1759, 1760, 1761, 1762, 1763, 1764, 1765, 1766, 1767, 1768, 1769, 1770, 1771, 1772, 1773, 1774, 1775, 1776, 1777, 1778, 1779, 1780, 1781, 1782, 1783, 1784, 1785, 1786, 1787, 1788, 1789, 1790, 1791, 1792, 1793, 1794, 1795, 1796, 1797, 1798, 1799, 1800, 1801, 1802, 1803, 1804, 1805, 1807, 1808, 1809, 1810, 1811, 1812, 1813, 1814, 1815, 1816, 1817, 1818, 1819, 1820, 1821, 1822, 1823, 1824, 1825, 1826, 1827, 1828, 1829, 1830, 1831, 1832, 1833, 1834, 1835, 1836, 1837, 1838, 1839, 1840, 1841, 1842, 1843, 1844, 1845, 1846, 1847, 1848, 1849, 1850, 1851, 1852, 1853, 1854, 1855, 1856, 1857, 1858, 1859, 1860, 1861, 1862, 1863, 1864, 1865, 1866, 1867, 1868, 1869, 1870, 1871, 1872, 1873, 1874, 1875, 1876, 1877, 1878, 1879, 1880, 1881, 1882, 1883, 1884, 1885, 1886, 1887, 1888, 1889, 1890, 1891, 1892, 1893, 1894, 1895, 1896, 1897, 1898, 1899, 1900, 1901, 1902, 1903, 1904, 1905, 1906, 1907, 1908, 1909, 1910, 1911, 1912, 1913, 1914, 1915, 1916, 1917, 1918, 1919, 1920, 1921, 1922, 1923, 1924, 1925, 1926, 1927, 1928, 1929, 1930, 1931, 1932, 1933, 1934, 1936, 1937, 1938, 1939, 1940, 1941, 1942, 1943, 1944, 1945, 1946, 1947, 1948, 1949, 1950, 1951, 1952, 1953, 1954, 1955, 1956, 1957, 1958, 1959, 1960, 1961, 1962, 1963, 1964, 1965, 1966, 1967, 1968, 1969, 1970, 1971, 1972, 1973, 1974, 1975, 1976, 1977, 1978, 1979, 1980, 1981, 1982, 1983, 1984, 1985, 1986, 1987, 1988, 1989, 1990, 1991, 1992, 1993, 1994, 1995, 1996, 1997, 1998, 1999, 2000, 2001, 2002, 2003, 2004, 2005, 2006, 2007, 2008, 2009, 2010, 2011, 2012, 2013, 2014, 2015, 2016, 2017, 2018, 2019, 2020, 2021, 2022, 2023, 2024, 2025, 2026, 2027, 2028, 2029, 2030, 2031, 2032, 2033, 2034, 2035, 2036, 2037, 2038, 2039, 2040, 2041, 2042, 2043, 2044, 2045, 2046, 2047, 2048, 2049, 2050, 2051, 2052, 2053, 2054, 2055, 2056, 2057, 2058, 2059, 2060, 2061, 2062, 2063, 2065, 2066, 2067, 2068, 2069, 2070, 2071, 2072, 2073, 2074, 2075, 2076, 2077, 2078, 2079, 2080, 2081, 2082, 2083, 2084, 2085, 2086, 2087, 2088, 2089, 2090, 2091, 2092, 2093, 2094, 2095, 2096, 2097, 2098, 2099, 2100, 2101, 2102, 2103, 2104, 2105, 2106, 2107, 2108, 2109, 2110, 2111, 2112, 2113, 2114, 2115, 2116, 2117, 2118, 2119, 2120, 2121, 2122, 2123, 2124, 2125, 2126, 2127, 2128, 2129, 2130, 2131, 2132, 2133, 2134, 2135, 2136, 2137, 2138, 2139, 2140, 2141, 2142, 2143, 2144, 2145, 2146, 2147, 2148, 2149, 2150, 2151, 2152, 2153, 2154, 2155, 2156, 2157, 2158, 2159, 2160, 2161, 2162, 2163, 2164, 2165, 2166, 2167, 2168, 2169, 2170, 2171, 2172, 2173, 2174, 2175, 2176, 2177, 2178, 2179, 2180, 2181, 2182, 2183, 2184, 2185, 2186, 2187, 2188, 2189, 2190, 2191, 2192, 2194, 2195, 2196, 2197, 2198, 2199, 2200, 2201, 2202, 2203, 2204, 2205, 2206, 2207, 2208, 2209, 2210, 2211, 2212, 2213, 2214, 2215, 2216, 2217, 2218, 2219, 2220, 2221, 2222, 2223, 2224, 2225, 2226, 2227, 2228, 2229, 2230, 2231, 2232, 2233, 2234, 2235, 2236, 2237, 2238, 2239, 2240, 2241, 2242, 2243, 2244, 2245, 2246, 2247, 2248, 2249, 2250, 2251, 2252, 2253, 2254, 2255, 2256, 2257, 2258, 2259, 2260, 2261, 2262, 2263, 2264, 2265, 2266, 2267, 2268, 2269, 2270, 2271, 2272, 2273, 2274, 2275, 2276, 2277, 2278, 2279, 2280, 2281, 2282, 2283, 2284, 2285, 2286, 2287, 2288, 2289, 2290, 2291, 2292, 2293, 2294, 2295, 2296, 2297, 2298, 2299, 2300, 2301, 2302, 2303, 2304, 2305, 2306, 2307, 2308, 2309, 2310, 2311, 2312, 2313, 2314, 2315, 2316, 2317, 2318, 2319, 2320, 2321, 2323, 2324, 2325, 2326, 2327, 2328, 2329, 2330, 2331, 2332, 2333, 2334, 2335, 2336, 2337, 2338, 2339, 2340, 2341, 2342, 2343, 2344, 2345, 2346, 2347, 2348, 2349, 2350, 2351, 2352, 2353, 2354, 2355, 2356, 2357, 2358, 2359, 2360, 2361, 2362, 2363, 2364, 2365, 2366, 2367, 2368, 2369, 2370, 2371, 2372, 2373, 2374, 2375, 2376, 2377, 2378, 2379, 2380, 2381, 2382, 2383, 2384, 2385, 2386, 2387, 2388, 2389, 2390, 2391, 2392, 2393, 2394, 2395, 2396, 2397, 2398, 2399, 2400, 2401, 2402, 2403, 2404, 2405, 2406, 2407, 2408, 2409, 2410, 2411, 2412, 2413, 2414, 2415, 2416, 2417, 2418, 2419, 2420, 2421, 2422, 2423, 2424, 2425, 2426, 2427, 2428, 2429, 2430, 2431, 2432, 2433, 2434, 2435, 2436, 2437, 2438, 2439, 2440, 2441, 2442, 2443, 2444, 2445, 2446, 2447, 2448, 2449, 2450, 2452, 2453, 2454, 2455, 2456, 2457, 2458, 2459, 2460, 2461, 2462, 2463, 2464, 2465, 2466, 2467, 2468, 2469, 2470, 2471, 2472, 2473, 2474, 2475, 2476, 2477, 2478, 2479, 2480, 2481, 2482, 2483, 2484, 2485, 2486, 2487, 2488, 2489, 2490, 2491, 2492, 2493, 2494, 2495, 2496, 2497, 2498, 2499, 2500, 2501, 2502, 2503, 2504, 2505, 2506, 2507, 2508, 2509, 2510, 2511, 2512, 2513, 2514, 2515, 2516, 2517, 2518, 2519, 2520, 2521, 2522, 2523, 2524, 2525, 2526, 2527, 2528, 2529, 2530, 2531, 2532, 2533, 2534, 2535, 2536, 2537, 2538, 2539, 2540, 2541, 2542, 2543, 2544, 2545, 2546, 2547, 2548, 2549, 2550, 2551, 2552, 2553, 2554, 2555, 2556, 2557, 2558, 2559, 2560, 2561, 2562, 2563, 2564, 2565, 2566, 2567, 2568, 2569, 2570, 2571, 2572, 2573, 2574, 2575, 2576, 2577, 2578, 2579, 2581, 2582, 2583, 2584, 2585, 2586, 2587, 2588, 2589, 2590, 2591, 2592, 2593, 2594, 2595, 2596, 2597, 2598, 2599, 2600, 2601, 2602, 2603, 2604, 2605, 2606, 2607, 2608, 2609, 2610, 2611, 2612, 2613, 2614, 2615, 2616, 2617, 2618, 2619, 2620, 2621, 2622, 2623, 2624, 2625, 2626, 2627, 2628, 2629, 2630, 2631, 2632, 2633, 2634, 2635, 2636, 2637, 2638, 2639, 2640, 2641, 2642, 2643, 2644, 2645, 2646, 2647, 2648, 2649, 2650, 2651, 2652, 2653, 2654, 2655, 2656, 2657, 2658, 2659, 2660, 2661, 2662, 2663, 2664, 2665, 2666, 2667, 2668, 2669, 2670, 2671, 2672, 2673, 2674, 2675, 2676, 2677, 2678, 2679, 2680, 2681, 2682, 2683, 2684, 2685, 2686, 2687, 2688, 2689, 2690, 2691, 2692, 2693, 2694, 2695, 2696, 2697, 2698, 2699, 2700, 2701, 2702, 2703, 2704, 2705, 2706, 2707, 2708, 2710, 2711, 2712, 2713, 2714, 2715, 2716, 2717, 2718, 2719, 2720, 2721, 2722, 2723, 2724, 2725, 2726, 2727, 2728, 2729, 2730, 2731, 2732, 2733, 2734, 2735, 2736, 2737, 2738, 2739, 2740, 2741, 2742, 2743, 2744, 2745, 2746, 2747, 2748, 2749, 2750, 2751, 2752, 2753, 2754, 2755, 2756, 2757, 2758, 2759, 2760, 2761, 2762, 2763, 2764, 2765, 2766, 2767, 2768, 2769, 2770, 2771, 2772, 2773, 2774, 2775, 2776, 2777, 2778, 2779, 2780, 2781, 2782, 2783, 2784, 2785, 2786, 2787, 2788, 2789, 2790, 2791, 2792, 2793, 2794, 2795, 2796, 2797, 2798, 2799, 2800, 2801, 2802, 2803, 2804, 2805, 2806, 2807, 2808, 2809, 2810, 2811, 2812, 2813, 2814, 2815, 2816, 2817, 2818, 2819, 2820, 2821, 2822, 2823, 2824, 2825, 2826, 2827, 2828, 2829, 2830, 2831, 2832, 2833, 2834, 2835, 2836, 2837, 2839, 2840, 2841, 2842, 2843, 2844, 2845, 2846, 2847, 2848, 2849, 2850, 2851, 2852, 2853, 2854, 2855, 2856, 2857, 2858, 2859, 2860, 2861, 2862, 2863, 2864, 2865, 2866, 2867, 2868, 2869, 2870, 2871, 2872, 2873, 2874, 2875, 2876, 2877, 2878, 2879, 2880, 2881, 2882, 2883, 2884, 2885, 2886, 2887, 2888, 2889, 2890, 2891, 2892, 2893, 2894, 2895, 2896, 2897, 2898, 2899, 2900, 2901, 2902, 2903, 2904, 2905, 2906, 2907, 2908, 2909, 2910, 2911, 2912, 2913, 2914, 2915, 2916, 2917, 2918, 2919, 2920, 2921, 2922, 2923, 2924, 2925, 2926, 2927, 2928, 2929, 2930, 2931, 2932, 2933, 2934, 2935, 2936, 2937, 2938, 2939, 2940, 2941, 2942, 2943, 2944, 2945, 2946, 2947, 2948, 2949, 2950, 2951, 2952, 2953, 2954, 2955, 2956, 2957, 2958, 2959, 2960, 2961, 2962, 2963, 2964, 2965, 2966, 2968, 2969, 2970, 2971, 2972, 2973, 2974, 2975, 2976, 2977, 2978, 2979, 2980, 2981, 2982, 2983, 2984, 2985, 2986, 2987, 2988, 2989, 2990, 2991, 2992, 2993, 2994, 2995, 2996, 2997, 2998, 2999, 3000, 3001, 3002, 3003, 3004, 3005, 3006, 3007, 3008, 3009, 3010, 3011, 3012, 3013, 3014, 3015, 3016, 3017, 3018, 3019, 3020, 3021, 3022, 3023, 3024, 3025, 3026, 3027, 3028, 3029, 3030, 3031, 3032, 3033, 3034, 3035, 3036, 3037, 3038, 3039, 3040, 3041, 3042, 3043, 3044, 3045, 3046, 3047, 3048, 3049, 3050, 3051, 3052, 3053, 3054, 3055, 3056, 3057, 3058, 3059, 3060, 3061, 3062, 3063, 3064, 3065, 3066, 3067, 3068, 3069, 3070, 3071, 3072, 3073, 3074, 3075, 3076, 3077, 3078, 3079, 3080, 3081, 3082, 3083, 3084, 3085, 3086, 3087, 3088, 3089, 3090, 3091, 3092, 3093, 3094, 3095, 3097, 3098, 3099, 3100, 3101, 3102, 3103, 3104, 3105, 3106, 3107, 3108, 3109, 3110, 3111, 3112, 3113, 3114, 3115, 3116, 3117, 3118, 3119, 3120, 3121, 3122, 3123, 3124, 3125, 3126, 3127, 3128, 3129, 3130, 3131, 3132, 3133, 3134, 3135, 3136, 3137, 3138, 3139, 3140, 3141, 3142, 3143, 3144, 3145, 3146, 3147, 3148, 3149, 3150, 3151, 3152, 3153, 3154, 3155, 3156, 3157, 3158, 3159, 3160, 3161, 3162, 3163, 3164, 3165, 3166, 3167, 3168, 3169, 3170, 3171, 3172, 3173, 3174, 3175, 3176, 3177, 3178, 3179, 3180, 3181, 3182, 3183, 3184, 3185, 3186, 3187, 3188, 3189, 3190, 3191, 3192, 3193, 3194, 3195, 3196, 3197, 3198, 3199, 3200, 3201, 3202, 3203, 3204, 3205, 3206, 3207, 3208, 3209, 3210, 3211, 3212, 3213, 3214, 3215, 3216, 3217, 3218, 3219, 3220, 3221, 3222, 3223, 3224, 3226, 3227, 3228, 3229, 3230, 3231, 3232, 3233, 3234, 3235, 3236, 3237, 3238, 3239, 3240, 3241, 3242, 3243, 3244, 3245, 3246, 3247, 3248, 3249, 3250, 3251, 3252, 3253, 3254, 3255, 3256, 3257, 3258, 3259, 3260, 3261, 3262, 3263, 3264, 3265, 3266, 3267, 3268, 3269, 3270, 3271, 3272, 3273, 3274, 3275, 3276, 3277, 3278, 3279, 3280, 3281, 3282, 3283, 3284, 3285, 3286, 3287, 3288, 3289, 3290, 3291, 3292, 3293, 3294, 3295, 3296, 3297, 3298, 3299, 3300, 3301, 3302, 3303, 3304, 3305, 3306, 3307, 3308, 3309, 3310, 3311, 3312, 3313, 3314, 3315, 3316, 3317, 3318, 3319, 3320, 3321, 3322, 3323, 3324, 3325, 3326, 3327, 3328, 3329, 3330, 3331, 3332, 3333, 3334, 3335, 3336, 3337, 3338, 3339, 3340, 3341, 3342, 3343, 3344, 3345, 3346, 3347, 3348, 3349, 3350, 3351, 3352, 3353, 3355, 3356, 3357, 3358, 3359, 3360, 3361, 3362, 3363, 3364, 3365, 3366, 3367, 3368, 3369, 3370, 3371, 3372, 3373, 3374, 3375, 3376, 3377, 3378, 3379, 3380, 3381, 3382, 3383, 3384, 3385, 3386, 3387, 3388, 3389, 3390, 3391, 3392, 3393, 3394, 3395, 3396, 3397, 3398, 3399, 3400, 3401, 3402, 3403, 3404, 3405, 3406, 3407, 3408, 3409, 3410, 3411, 3412, 3413, 3414, 3415, 3416, 3417, 3418, 3419, 3420, 3421, 3422, 3423, 3424, 3425, 3426, 3427, 3428, 3429, 3430, 3431, 3432, 3433, 3434, 3435, 3436, 3437, 3438, 3439, 3440, 3441, 3442, 3443, 3444, 3445, 3446, 3447, 3448, 3449, 3450, 3451, 3452, 3453, 3454, 3455, 3456, 3457, 3458, 3459, 3460, 3461, 3462, 3463, 3464, 3465, 3466, 3467, 3468, 3469, 3470, 3471, 3472, 3473, 3474, 3475, 3476, 3477, 3478, 3479, 3480, 3481, 3482, 3484, 3485, 3486, 3487, 3488, 3489, 3490, 3491, 3492, 3493, 3494, 3495, 3496, 3497, 3498, 3499, 3500, 3501, 3502, 3503, 3504, 3505, 3506, 3507, 3508, 3509, 3510, 3511, 3512, 3513, 3514, 3515, 3516, 3517, 3518, 3519, 3520, 3521, 3522, 3523, 3524, 3525, 3526, 3527, 3528, 3529, 3530, 3531, 3532, 3533, 3534, 3535, 3536, 3537, 3538, 3539, 3540, 3541, 3542, 3543, 3544, 3545, 3546, 3547, 3548, 3549, 3550, 3551, 3552, 3553, 3554, 3555, 3556, 3557, 3558, 3559, 3560, 3561, 3562, 3563, 3564, 3565, 3566, 3567, 3568, 3569, 3570, 3571, 3572, 3573, 3574, 3575, 3576, 3577, 3578, 3579, 3580, 3581, 3582, 3583, 3584, 3585, 3586, 3587, 3588, 3589, 3590, 3591, 3592, 3593, 3594, 3595, 3596, 3597, 3598, 3599, 3600, 3601, 3602, 3603, 3604, 3605, 3606, 3607, 3608, 3609, 3610, 3611, 3613, 3614, 3615, 3616, 3617, 3618, 3619, 3620, 3621, 3622, 3623, 3624, 3625, 3626, 3627, 3628, 3629, 3630, 3631, 3632, 3633, 3634, 3635, 3636, 3637, 3638, 3639, 3640, 3641, 3642, 3643, 3644, 3645, 3646, 3647, 3648, 3649, 3650, 3651, 3652, 3653, 3654, 3655, 3656, 3657, 3658, 3659, 3660, 3661, 3662, 3663, 3664, 3665, 3666, 3667, 3668, 3669, 3670, 3671, 3672, 3673, 3674, 3675, 3676, 3677, 3678, 3679, 3680, 3681, 3682, 3683, 3684, 3685, 3686, 3687, 3688, 3689, 3690, 3691, 3692, 3693, 3694, 3695, 3696, 3697, 3698, 3699, 3700, 3701, 3702, 3703, 3704, 3705, 3706, 3707, 3708, 3709, 3710, 3711, 3712, 3713, 3714, 3715, 3716, 3717, 3718, 3719, 3720, 3721, 3722, 3723, 3724, 3725, 3726, 3727, 3728, 3729, 3730, 3731, 3732, 3733, 3734, 3735, 3736, 3737, 3738, 3739, 3740, 3742, 3743, 3744, 3745, 3746, 3747, 3748, 3749, 3750, 3751, 3752, 3753, 3754, 3755, 3756, 3757, 3758, 3759, 3760, 3761, 3762, 3763, 3764, 3765, 3766, 3767, 3768, 3769, 3770, 3771, 3772, 3773, 3774, 3775, 3776, 3777, 3778, 3779, 3780, 3781, 3782, 3783, 3784, 3785, 3786, 3787, 3788, 3789, 3790, 3791, 3792, 3793, 3794, 3795, 3796, 3797, 3798, 3799, 3800, 3801, 3802, 3803, 3804, 3805, 3806, 3807, 3808, 3809, 3810, 3811, 3812, 3813, 3814, 3815, 3816, 3817, 3818, 3819, 3820, 3821, 3822, 3823, 3824, 3825, 3826, 3827, 3828, 3829, 3830, 3831, 3832, 3833, 3834, 3835, 3836, 3837, 3838, 3839, 3840, 3841, 3842, 3843, 3844, 3845, 3846, 3847, 3848, 3849, 3850, 3851, 3852, 3853, 3854, 3855, 3856, 3857, 3858, 3859, 3860, 3861, 3862, 3863, 3864, 3865, 3866, 3867, 3868, 3869, 3871, 3872, 3873, 3874, 3875, 3876, 3877, 3878, 3879, 3880, 3881, 3882, 3883, 3884, 3885, 3886, 3887, 3888, 3889, 3890, 3891, 3892, 3893, 3894, 3895, 3896, 3897, 3898, 3899, 3900, 3901, 3902, 3903, 3904, 3905, 3906, 3907, 3908, 3909, 3910, 3911, 3912, 3913, 3914, 3915, 3916, 3917, 3918, 3919, 3920, 3921, 3922, 3923, 3924, 3925, 3926, 3927, 3928, 3929, 3930, 3931, 3932, 3933, 3934, 3935, 3936, 3937, 3938, 3939, 3940, 3941, 3942, 3943, 3944, 3945, 3946, 3947, 3948, 3949, 3950, 3951, 3952, 3953, 3954, 3955, 3956, 3957, 3958, 3959, 3960, 3961, 3962, 3963, 3964, 3965, 3966, 3967, 3968, 3969, 3970, 3971, 3972, 3973, 3974, 3975, 3976, 3977, 3978, 3979, 3980, 3981, 3982, 3983, 3984, 3985, 3986, 3987, 3988, 3989, 3990, 3991, 3992, 3993, 3994, 3995, 3996, 3997, 3998], + "packed_schedule_10_cumulativeSequenceLengthT": [0, 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, 160], + "packed_schedule_11_tokenOffsetT": [0, 1, 2, 129, 130, 131, 132, 133, 258, 259, 260, 261, 262, 263, 264, 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, 126, 127, 128, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 265, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 324, 325, 326, 327, 328, 329, 330, 331, 332, 333, 334, 335, 336, 337, 338, 339, 340, 341, 342, 343, 344, 345, 346, 347, 348, 349, 350, 351, 352, 353, 354, 355, 356, 357, 358, 359, 360, 361, 362, 363, 364, 365, 366, 367, 368, 369, 370, 371, 372, 373, 374, 375, 376, 377, 378, 379, 380, 381, 382, 383, 384, 385, 386], + "packed_schedule_12_tokenOffsetT": [0, 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, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256], + "packed_schedule_13_tokenOffsetT": [0, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 324, 325, 326, 327, 328, 329, 330, 331, 332, 333, 334, 335, 336, 337, 338, 339, 340, 341, 342, 343, 344, 345, 346, 347, 348, 349, 350, 351, 352, 353, 354, 355, 356, 357, 358, 359, 360, 361, 362, 363, 364, 365, 366, 367, 368, 369, 370, 371, 372, 373, 374, 375, 376, 377, 378, 379, 380, 381, 382, 383, 384, 385, 386, 387, 388, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 405, 406, 407, 408, 409, 410, 411, 412, 413, 414, 415, 416, 417, 418, 419, 420, 421, 422, 423, 424, 425, 426, 427, 428, 429, 430, 431, 432, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 470, 471, 472, 473, 474, 475, 476, 477, 478, 479, 480, 481, 482, 483, 484, 485, 486, 487, 488, 489, 490, 491, 492, 493, 494, 495, 496, 497, 498, 499, 500, 501, 502, 503, 504, 505, 506, 507, 508, 509, 510, 511, 512, 513, 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, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256], + "packed_schedule_14_tokenOffsetT": [0, 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, 126, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 127, 128], + "packed_schedule_15_tokenOffsetT": [0, 1, 2, 3, 4, 5, 6, 9, 10, 11, 12, 13, 14, 15, 16, 17, 7, 8], + "packed_schedule_16_tokenOffsetT": [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 15, 16], + "packed_schedule_17_tokenOffsetT": [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 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, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30], + "packed_schedule_18_tokenOffsetT": [0, 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, 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, 31, 32], + "portable_schedule_19_tokenOffsetT": [0, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15], + "portable_schedule_20_tokenOffsetT": [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33], + "portable_schedule_21_tokenOffsetT": [0, 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], + "portable_schedule_22_tokenOffsetT": [0, 63, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 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, 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], + "portable_schedule_23_tokenOffsetT": [0, 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], + "portable_schedule_24_tokenOffsetT": [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 16], + "portable_schedule_25_tokenOffsetT": [0, 1, 2, 3, 4, 5, 6, 7, 9, 10, 11, 12, 13, 14, 15, 16, 17, 8], + "portable_schedule_26_tokenOffsetT": [0, 127, 254, 255, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 324, 325, 326, 327, 328, 329, 330, 331, 332, 333, 334, 335, 336, 337, 338, 339, 340, 341, 342, 343, 344, 345, 346, 347, 348, 349, 350, 351, 352, 353, 354, 355, 356, 357, 358, 359, 360, 361, 362, 363, 364, 365, 366, 367, 368, 369, 370, 371, 372, 373, 374, 375, 376, 377, 378, 379, 380, 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, 126, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253], + "unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT": [0.25, -0.5, 0.75, 0.125, -0.375, 0.625, -0.125, 0.5, -0.25, 0.875, -0.75, 0.375, 0.0625, -0.625, 0.1875, 0.9375, -0.1875], + "unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT": [0, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 32, 33], + "unequal_float32_d32v4_b3t34h2_bias0attn0_input_tokenOffsetT": [0, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 33], + "unequal_float32_d32v20_b4t34h2_bias0attn0_input_tokenOffsetT": [17, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 50], + "unequal_float32_d32v28_b3t48h3_bias0attn0_input_tokenOffsetT": [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 15, 16, 33], + "unequal_float32_d64v48_b3t65h2_bias0attn0_input_tokenOffsetT": [0, 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, 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, 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, 64, 65], + "unequal_f16_bias_key_sum_exceeds_f16_finite_range_input_valueT": [0.5, 0.625, 0.75, 0.875, 1, 1.125, 1.25, 1.375, 1.5, 1.625, 1.75, 1.875, 2, 2.125, 2.25, 2.375, 2.5], + "wide_schedule_34": [0, 9, 10, 11, 12, 13, 14, 1, 2, 3, 4, 5, 6, 7, 8, 15, 16, 17], + "wide_schedule_35": [0, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 33, 34, 35], + "wide_schedule_36": [32, 33, 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, 0, 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, 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, 93, 94, 95], + "wide_schedule_37": [0, 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, 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, 67, 68, 69], + "wide_schedule_38": [0, 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, 126, 127, 128, 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, 129, 130, 131], + "wide_schedule_39": [0, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 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, 126, 127, 128, 129, 130, 259, 260, 261], + "wide_schedule_40": [0, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 324, 325, 326, 327, 328, 329, 330, 331, 332, 333, 334, 335, 336, 337, 338, 339, 340, 341, 342, 343, 344, 345, 346, 347, 348, 349, 350, 351, 352, 353, 354, 355, 356, 357, 358, 359, 360, 361, 362, 363, 364, 365, 366, 367, 368, 369, 370, 371, 372, 373, 374, 375, 376, 377, 378, 379, 380, 381, 382, 383, 384, 385, 386, 387, 388, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 405, 406, 407, 408, 409, 410, 411, 412, 413, 414, 415, 416, 417, 418, 419, 420, 421, 422, 423, 424, 425, 426, 427, 428, 429, 430, 431, 432, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 470, 471, 472, 473, 474, 475, 476, 477, 478, 479, 480, 481, 482, 483, 484, 485, 486, 487, 488, 489, 490, 491, 492, 493, 494, 495, 496, 497, 498, 499, 500, 501, 502, 503, 504, 505, 506, 507, 508, 509, 510, 511, 512, 513, 514, 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, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 515, 516, 517], + "wide_schedule_41": [0, 515, 516, 517, 518, 519, 520, 521, 522, 523, 524, 525, 526, 527, 528, 529, 530, 531, 532, 533, 534, 535, 536, 537, 538, 539, 540, 541, 542, 543, 544, 545, 546, 547, 548, 549, 550, 551, 552, 553, 554, 555, 556, 557, 558, 559, 560, 561, 562, 563, 564, 565, 566, 567, 568, 569, 570, 571, 572, 573, 574, 575, 576, 577, 578, 579, 580, 581, 582, 583, 584, 585, 586, 587, 588, 589, 590, 591, 592, 593, 594, 595, 596, 597, 598, 599, 600, 601, 602, 603, 604, 605, 606, 607, 608, 609, 610, 611, 612, 613, 614, 615, 616, 617, 618, 619, 620, 621, 622, 623, 624, 625, 626, 627, 628, 629, 630, 631, 632, 633, 634, 635, 636, 637, 638, 639, 640, 641, 642, 643, 644, 645, 646, 647, 648, 649, 650, 651, 652, 653, 654, 655, 656, 657, 658, 659, 660, 661, 662, 663, 664, 665, 666, 667, 668, 669, 670, 671, 672, 673, 674, 675, 676, 677, 678, 679, 680, 681, 682, 683, 684, 685, 686, 687, 688, 689, 690, 691, 692, 693, 694, 695, 696, 697, 698, 699, 700, 701, 702, 703, 704, 705, 706, 707, 708, 709, 710, 711, 712, 713, 714, 715, 716, 717, 718, 719, 720, 721, 722, 723, 724, 725, 726, 727, 728, 729, 730, 731, 732, 733, 734, 735, 736, 737, 738, 739, 740, 741, 742, 743, 744, 745, 746, 747, 748, 749, 750, 751, 752, 753, 754, 755, 756, 757, 758, 759, 760, 761, 762, 763, 764, 765, 766, 767, 768, 769, 770, 771, 772, 773, 774, 775, 776, 777, 778, 779, 780, 781, 782, 783, 784, 785, 786, 787, 788, 789, 790, 791, 792, 793, 794, 795, 796, 797, 798, 799, 800, 801, 802, 803, 804, 805, 806, 807, 808, 809, 810, 811, 812, 813, 814, 815, 816, 817, 818, 819, 820, 821, 822, 823, 824, 825, 826, 827, 828, 829, 830, 831, 832, 833, 834, 835, 836, 837, 838, 839, 840, 841, 842, 843, 844, 845, 846, 847, 848, 849, 850, 851, 852, 853, 854, 855, 856, 857, 858, 859, 860, 861, 862, 863, 864, 865, 866, 867, 868, 869, 870, 871, 872, 873, 874, 875, 876, 877, 878, 879, 880, 881, 882, 883, 884, 885, 886, 887, 888, 889, 890, 891, 892, 893, 894, 895, 896, 897, 898, 899, 900, 901, 902, 903, 904, 905, 906, 907, 908, 909, 910, 911, 912, 913, 914, 915, 916, 917, 918, 919, 920, 921, 922, 923, 924, 925, 926, 927, 928, 929, 930, 931, 932, 933, 934, 935, 936, 937, 938, 939, 940, 941, 942, 943, 944, 945, 946, 947, 948, 949, 950, 951, 952, 953, 954, 955, 956, 957, 958, 959, 960, 961, 962, 963, 964, 965, 966, 967, 968, 969, 970, 971, 972, 973, 974, 975, 976, 977, 978, 979, 980, 981, 982, 983, 984, 985, 986, 987, 988, 989, 990, 991, 992, 993, 994, 995, 996, 997, 998, 999, 1000, 1001, 1002, 1003, 1004, 1005, 1006, 1007, 1008, 1009, 1010, 1011, 1012, 1013, 1014, 1015, 1016, 1017, 1018, 1019, 1020, 1021, 1022, 1023, 1024, 1025, 1026, 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, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 324, 325, 326, 327, 328, 329, 330, 331, 332, 333, 334, 335, 336, 337, 338, 339, 340, 341, 342, 343, 344, 345, 346, 347, 348, 349, 350, 351, 352, 353, 354, 355, 356, 357, 358, 359, 360, 361, 362, 363, 364, 365, 366, 367, 368, 369, 370, 371, 372, 373, 374, 375, 376, 377, 378, 379, 380, 381, 382, 383, 384, 385, 386, 387, 388, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 405, 406, 407, 408, 409, 410, 411, 412, 413, 414, 415, 416, 417, 418, 419, 420, 421, 422, 423, 424, 425, 426, 427, 428, 429, 430, 431, 432, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 470, 471, 472, 473, 474, 475, 476, 477, 478, 479, 480, 481, 482, 483, 484, 485, 486, 487, 488, 489, 490, 491, 492, 493, 494, 495, 496, 497, 498, 499, 500, 501, 502, 503, 504, 505, 506, 507, 508, 509, 510, 511, 512, 513, 514, 1027, 1028, 1029], + "fragment_schedule_634ea297a37d": [15, 16, 17, 18, 19, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 42, 43, 44], + "fragment_schedule_326ff2747fbb": [0, 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, 126, 127, 128, 129, 130, 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, 131, 132, 133], + "fragment_schedule_eb0c4a513a9e": [345, 346, 347, 348, 349, 350, 351, 352, 353, 354, 355, 356, 357, 358, 359, 360, 361, 362, 363, 364, 365, 366, 367, 368, 369, 370, 371, 372, 373, 374, 375, 376, 377, 378, 379, 380, 381, 382, 383, 384, 385, 386, 387, 388, 389, 390, 391, 392, 393, 394, 395, 396, 397, 398, 399, 400, 401, 402, 403, 404, 405, 406, 407, 408, 409, 410, 411, 412, 413, 414, 415, 416, 417, 418, 419, 420, 421, 422, 423, 424, 425, 426, 427, 428, 429, 430, 431, 432, 433, 434, 435, 436, 437, 438, 439, 440, 441, 442, 443, 444, 445, 446, 447, 448, 449, 450, 451, 452, 453, 454, 455, 456, 457, 458, 459, 460, 461, 462, 463, 464, 465, 466, 467, 468, 469, 470, 471, 472, 473, 474, 475, 476, 477, 478, 479, 480, 481, 482, 483, 484, 485, 486, 487, 488, 489, 490, 491, 492, 493, 494, 495, 496, 497, 498, 499, 500, 501, 502, 503, 504, 505, 506, 507, 508, 509, 510, 511, 512, 513, 514, 515, 690, 691, 692, 693, 694, 695, 696, 697, 698, 699, 700, 701, 702, 703, 704, 705, 706, 707, 708, 709, 710, 711, 712, 713, 714, 715, 716, 717, 718, 719, 720, 721, 722, 723, 724, 725, 726, 727, 728, 729, 730, 731, 732, 733, 734, 735, 736, 737, 738, 739, 740, 741, 742, 743, 744, 745, 746, 747, 748, 749, 750, 751, 752, 753, 754, 755, 756, 757, 758, 759, 760, 761, 762, 763, 764, 765, 766, 767, 768, 769, 770, 771, 772, 773, 774, 775, 776, 777, 778, 779, 780, 781, 782, 783, 784, 785, 786, 787, 788, 789, 790, 791, 792, 793, 794, 795, 796, 797, 798, 799, 800, 801, 802, 803, 804, 805, 806, 807, 808, 809, 810, 811, 812, 813, 814, 815, 816, 817, 818, 819, 820, 821, 822, 823, 824, 825, 826, 827, 828, 829, 830, 831, 832, 833, 834, 835, 836, 837, 838, 839, 840, 841, 842, 843, 844, 845, 846, 847, 848, 849, 850, 851, 852, 853, 854, 855, 856, 857, 858, 859, 860, 861, 862, 863, 864, 865, 866, 867, 868, 869, 870, 871, 872, 873, 874, 875, 876, 877, 878, 879, 880, 881, 882, 883, 884, 885, 886, 887, 888, 889, 890, 891, 892, 893, 894, 895, 896, 897, 898, 899, 900, 901, 902, 903, 904, 905, 906, 907, 908, 909, 910, 911, 912, 913, 914, 915, 916, 917, 918, 919, 920, 921, 922, 923, 924, 925, 926, 927, 928, 929, 930, 931, 932, 933, 934, 935, 936, 937, 938, 939, 940, 941, 942, 943, 944, 945, 946, 947, 948, 949, 950, 951, 952, 953, 954, 955, 956, 957, 958, 959, 960, 961, 962, 963, 964, 965, 966, 967, 968, 969, 970, 971, 972, 973, 974, 975, 976, 977, 978, 979, 980, 981, 982, 983, 984, 985, 986, 987, 988, 989, 990, 991, 992, 993, 994, 995, 996, 997, 998, 999, 1000, 1001, 1002, 1003, 1004, 1005, 1006, 1007, 1008, 1009, 1010, 1011, 1012, 1013, 1014, 1015, 1016, 1017, 1018, 1019, 1020, 1021, 1022, 1023, 1024, 1025, 1026, 1027, 1028, 1029, 1030, 1031, 0, 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, 126, 127, 128, 129, 130, 131, 132, 133, 134, 135, 136, 137, 138, 139, 140, 141, 142, 143, 144, 145, 146, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163, 164, 165, 166, 167, 168, 169, 170, 171, 172, 173, 174, 175, 176, 177, 178, 179, 180, 181, 182, 183, 184, 185, 186, 187, 188, 189, 190, 191, 192, 193, 194, 195, 196, 197, 198, 199, 200, 201, 202, 203, 204, 205, 206, 207, 208, 209, 210, 211, 212, 213, 214, 215, 216, 217, 218, 219, 220, 221, 222, 223, 224, 225, 226, 227, 228, 229, 230, 231, 232, 233, 234, 235, 236, 237, 238, 239, 240, 241, 242, 243, 244, 245, 246, 247, 248, 249, 250, 251, 252, 253, 254, 255, 256, 257, 258, 259, 260, 261, 262, 263, 264, 265, 266, 267, 268, 269, 270, 271, 272, 273, 274, 275, 276, 277, 278, 279, 280, 281, 282, 283, 284, 285, 286, 287, 288, 289, 290, 291, 292, 293, 294, 295, 296, 297, 298, 299, 300, 301, 302, 303, 304, 305, 306, 307, 308, 309, 310, 311, 312, 313, 314, 315, 316, 317, 318, 319, 320, 321, 322, 323, 324, 325, 326, 327, 328, 329, 330, 331, 332, 333, 334, 335, 336, 337, 338, 339, 340, 341, 342, 343, 344, 516, 517, 518, 519, 520, 521, 522, 523, 524, 525, 526, 527, 528, 529, 530, 531, 532, 533, 534, 535, 536, 537, 538, 539, 540, 541, 542, 543, 544, 545, 546, 547, 548, 549, 550, 551, 552, 553, 554, 555, 556, 557, 558, 559, 560, 561, 562, 563, 564, 565, 566, 567, 568, 569, 570, 571, 572, 573, 574, 575, 576, 577, 578, 579, 580, 581, 582, 583, 584, 585, 586, 587, 588, 589, 590, 591, 592, 593, 594, 595, 596, 597, 598, 599, 600, 601, 602, 603, 604, 605, 606, 607, 608, 609, 610, 611, 612, 613, 614, 615, 616, 617, 618, 619, 620, 621, 622, 623, 624, 625, 626, 627, 628, 629, 630, 631, 632, 633, 634, 635, 636, 637, 638, 639, 640, 641, 642, 643, 644, 645, 646, 647, 648, 649, 650, 651, 652, 653, 654, 655, 656, 657, 658, 659, 660, 661, 662, 663, 664, 665, 666, 667, 668, 669, 670, 671, 672, 673, 674, 675, 676, 677, 678, 679, 680, 681, 682, 683, 684, 685, 686, 687, 688, 689, 1032, 1033, 1034] + }, + "cases": [ + { + "name": "ort_packed_qkv_batch2_headsize32", + "provenance": { + "source": "onnxruntime/test/testdata/attention/packed_multihead_attention_test_data.txt", + "test": "PackedMultiHeadAttentionTest.PackedQKV_Padding_NoBias_unfused" + }, + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [3, 1, 3, 32], + "data": { + "kind": "values", + "values": [-0.35420692, 1.31206024, -2.80201197, 2.42258096, -0.86031514, -1.44535458, -0.10832444, -2.00132895, 1.62475216, 0.10978927, 1.84596729, 0.4890855, 1.44369888, 0.87542874, -1.16434252, 0.52133209, 1.54848897, -2.21174526, -0.28574878, 0.70815033, 1.18327498, 3.14097571, -0.25795099, 1.89341247, -0.11603792, 0.38110194, 0.40873206, -1.14149106, 0.79770875, -0.98069525, -1.53588808, 0.50821728, -0.04407793, 1.29459429, 1.05810797, 1.92067695, -0.65047157, 0.99029726, -1.69796586, 1.1532042, -1.66444266, 1.78305888, 1.20582056, 1.69975281, 0.34572244, -0.60833001, 2.59864879, -1.05330181, -1.16554165, -0.03781542, -1.13475525, 0.7159515, -0.9116956, 1.2668606, 1.60492957, -0.53510487, -1.4018085, 1.83253956, 2.70238972, -1.48750985, 0.47105616, -0.79477602, -1.93152475, 1.04042351, -0.89167893, 0.02633595, -0.84866279, 1.4348911, -2.91941142, -0.20650116, 1.85965109, 0.45669034, 0.07678832, 0.04492294, 0.67326981, 0.97103029, 1.53470886, -1.10242307, 0.86584085, -0.34770033, -1.24311507, -1.80293822, -1.01317739, -0.71518499, 0.77814674, -0.59236068, -2.00310278, 3.13277125, -1.20754123, 2.01506066, 0.8265081, 2.0608449, -0.46267471, 1.56365979, 4.31514502, -1.03099275, 3.50846076, -2.50027657, -2.59866142, 1.58495271, 2.21110034, -2.74877763, -1.00267041, 0.62646407, 2.5022738, -0.27291518, -0.33037442, 0.75840306, 0.45437157, -0.79876304, 0.83509272, 2.53716302, 0.01348384, -2.16307616, 2.01661849, 2.10746121, -1.70485222, 1.35548759, 1.3940165, -0.99451691, -4.13484812, 0.56262714, -0.92725742, -0.16389316, -1.31260049, 2.32357836, -3.05251694, -1.12570131, -2.65206385, 1.26134932, -1.01682174, 0.64366758, 0.95474619, 2.06720352, 0.5175072, -0.07041813, 0.53124994, -3.26612782, 1.3701334, 0.13939659, -0.57418114, 0.80680281, -3.40751696, -0.15847699, 0.97837782, -0.09121911, 1.1845212, 0.52711177, -1.8613584, -0.11258313, 0.85863215, -2.6026113, 0.72695309, 1.44092011, 0.4378598, -1.63415265, -1.05772328, 0.12997569, 0.07356137, -0.62493324, -2.11639142, -1.50897706, 1.63863683, 2.32786226, 1.32746494, 0.75751448, 0.57184196, 0.86446053, -0.62406683, 0.78861046, 0.01044065, 3.51772785, -1.33701336, 0.27977663, -0.35464612, 0.74973166, 0.033521, 1.55007398, 0.6984942, -2.47725606, -1.89363778, -1.79874682, -0.56210291, -1.7555604, 1.07565808, -0.18023658, 1.63777173, 1.28198206, 2.19431949, 0.6799897, -0.52531999, -1.8990674, 1.87849474, -1.80381167, 0.52235699, 2.38887334, -1.58878529, 0.6957109, 1.65044296, -0.2702429, 3.5958097, -1.97888982, 1.17034674, 0.26716161, -1.16770899, 0.74609619, 0.78886843, 0.1571752, -0.93303132, -0.84753871, -4.32799959, -1.94716609, -1.16980326, 1.62631667, 2.41053247, 3.78186774, 0.26432252, -0.40396988, 2.04414082, 0.65150046, 0.47777444, -2.57569051, 0.99004912, 2.47947693, -0.43267637, -1.80009198, 0.92961007, 2.05127883, -2.85521173, -0.21652693, -0.89153922, 0.1552467, -2.16850328, 1.46751809, 2.51663852, -0.49499366, 0.19886012, 0.77093124, -1.14819765, 1.47111738, 2.42824388, 1.5636996, 1.6993413, -0.42460468, -2.25951004, -1.18074155, 3.51091242, -0.30183151, -1.83517075, -0.56233191, 2.35561657, -3.63751698, -3.20001125, -1.6612078, 3.23455381, -1.86251283, 1.3515805, -2.2148149, -0.11812399, -1.7426343, -0.57895988, -0.04181165, 0.78120053, -2.22377038, -0.53264999, -2.03721714, 0.21023634, 2.55751204, -1.045228, 0.85386503, 0.41594937, -2.98181081, 1.14034331, -1.41539204, 0.13379651, 3.47018123, 1.53924727, 1.50004411, 2.87318921, 1.62624204, 0.64942807, -4.54302311, -1.5029422, -1.75212634, 0.2790069, -3.05124855, 3.30960631, -0.07991691] + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [2, 2], "data": { "kind": "values", "values": [0, 2, 3, 1] } }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [3], + "data": { "kind": "values", "values": [0, 1, 3] } + } + }, + "outputs": { + "outputT": { + "dtype": "float16", + "shape": [3, 32], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/ort_packed_qkv_batch2_headsize32_output_outputT" } + } + } + }, + "tolerance": 0.005, + "relTolerance": 0.005 + }, + { + "name": "ort_q_k_v_batch2_headsize32", + "provenance": { + "source": "onnxruntime/test/testdata/attention/packed_multihead_attention_test_data.txt", + "test": "PackedMultiHeadAttentionTest.PackedQKV_Padding_NoBias_unfused" + }, + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [3, 32], + "data": { + "kind": "values", + "values": [-0.35420692, 1.31206024, -2.80201197, 2.42258096, -0.86031514, -1.44535458, -0.10832444, -2.00132895, 1.62475216, 0.10978927, 1.84596729, 0.4890855, 1.44369888, 0.87542874, -1.16434252, 0.52133209, 1.54848897, -2.21174526, -0.28574878, 0.70815033, 1.18327498, 3.14097571, -0.25795099, 1.89341247, -0.11603792, 0.38110194, 0.40873206, -1.14149106, 0.79770875, -0.98069525, -1.53588808, 0.50821728, 3.50846076, -2.50027657, -2.59866142, 1.58495271, 2.21110034, -2.74877763, -1.00267041, 0.62646407, 2.5022738, -0.27291518, -0.33037442, 0.75840306, 0.45437157, -0.79876304, 0.83509272, 2.53716302, 0.01348384, -2.16307616, 2.01661849, 2.10746121, -1.70485222, 1.35548759, 1.3940165, -0.99451691, -4.13484812, 0.56262714, -0.92725742, -0.16389316, -1.31260049, 2.32357836, -3.05251694, -1.12570131, 1.87849474, -1.80381167, 0.52235699, 2.38887334, -1.58878529, 0.6957109, 1.65044296, -0.2702429, 3.5958097, -1.97888982, 1.17034674, 0.26716161, -1.16770899, 0.74609619, 0.78886843, 0.1571752, -0.93303132, -0.84753871, -4.32799959, -1.94716609, -1.16980326, 1.62631667, 2.41053247, 3.78186774, 0.26432252, -0.40396988, 2.04414082, 0.65150046, 0.47777444, -2.57569051, 0.99004912, 2.47947693] + } + }, + "keyT": { + "dtype": "float16", + "shape": [3, 32], + "data": { + "kind": "values", + "values": [-0.04407793, 1.29459429, 1.05810797, 1.92067695, -0.65047157, 0.99029726, -1.69796586, 1.1532042, -1.66444266, 1.78305888, 1.20582056, 1.69975281, 0.34572244, -0.60833001, 2.59864879, -1.05330181, -1.16554165, -0.03781542, -1.13475525, 0.7159515, -0.9116956, 1.2668606, 1.60492957, -0.53510487, -1.4018085, 1.83253956, 2.70238972, -1.48750985, 0.47105616, -0.79477602, -1.93152475, 1.04042351, -2.65206385, 1.26134932, -1.01682174, 0.64366758, 0.95474619, 2.06720352, 0.5175072, -0.07041813, 0.53124994, -3.26612782, 1.3701334, 0.13939659, -0.57418114, 0.80680281, -3.40751696, -0.15847699, 0.97837782, -0.09121911, 1.1845212, 0.52711177, -1.8613584, -0.11258313, 0.85863215, -2.6026113, 0.72695309, 1.44092011, 0.4378598, -1.63415265, -1.05772328, 0.12997569, 0.07356137, -0.62493324, -0.43267637, -1.80009198, 0.92961007, 2.05127883, -2.85521173, -0.21652693, -0.89153922, 0.1552467, -2.16850328, 1.46751809, 2.51663852, -0.49499366, 0.19886012, 0.77093124, -1.14819765, 1.47111738, 2.42824388, 1.5636996, 1.6993413, -0.42460468, -2.25951004, -1.18074155, 3.51091242, -0.30183151, -1.83517075, -0.56233191, 2.35561657, -3.63751698, -3.20001125, -1.6612078, 3.23455381, -1.86251283] + } + }, + "valueT": { + "dtype": "float16", + "shape": [3, 32], + "data": { + "kind": "values", + "values": [-0.89167893, 0.02633595, -0.84866279, 1.4348911, -2.91941142, -0.20650116, 1.85965109, 0.45669034, 0.07678832, 0.04492294, 0.67326981, 0.97103029, 1.53470886, -1.10242307, 0.86584085, -0.34770033, -1.24311507, -1.80293822, -1.01317739, -0.71518499, 0.77814674, -0.59236068, -2.00310278, 3.13277125, -1.20754123, 2.01506066, 0.8265081, 2.0608449, -0.46267471, 1.56365979, 4.31514502, -1.03099275, -2.11639142, -1.50897706, 1.63863683, 2.32786226, 1.32746494, 0.75751448, 0.57184196, 0.86446053, -0.62406683, 0.78861046, 0.01044065, 3.51772785, -1.33701336, 0.27977663, -0.35464612, 0.74973166, 0.033521, 1.55007398, 0.6984942, -2.47725606, -1.89363778, -1.79874682, -0.56210291, -1.7555604, 1.07565808, -0.18023658, 1.63777173, 1.28198206, 2.19431949, 0.6799897, -0.52531999, -1.8990674, 1.3515805, -2.2148149, -0.11812399, -1.7426343, -0.57895988, -0.04181165, 0.78120053, -2.22377038, -0.53264999, -2.03721714, 0.21023634, 2.55751204, -1.045228, 0.85386503, 0.41594937, -2.98181081, 1.14034331, -1.41539204, 0.13379651, 3.47018123, 1.53924727, 1.50004411, 2.87318921, 1.62624204, 0.64942807, -4.54302311, -1.5029422, -1.75212634, 0.2790069, -3.05124855, 3.30960631, -0.07991691] + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [2, 2], "data": { "kind": "values", "values": [0, 2, 3, 1] } }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [3], + "data": { "kind": "values", "values": [0, 1, 3] } + } + }, + "outputs": { + "outputT": { + "dtype": "float16", + "shape": [3, 32], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/ort_packed_qkv_batch2_headsize32_output_outputT" } + } + } + }, + "tolerance": 0.005, + "relTolerance": 0.005 + }, + { + "name": "ort_packed_qkv_headsize8_attn_bias", + "provenance": { + "source": "onnxruntime/test/testdata/attention/packed_multihead_attention_test_data.txt", + "test": "PackedMultiHeadAttentionTest.PackedQKV_Padding_NoBias_AttnBias" + }, + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [3, 2, 3, 8], + "data": { + "kind": "values", + "values": [-1.83615911, 0.08698978, 0.05601556, -1.1451025, -2.30377889, -0.39893439, 0.73342341, -0.09851928, 0.86949563, -0.10868365, -0.37917313, -1.23103046, 0.25640076, -1.50652349, 0.71594471, 0.49057019, -1.19203627, 0.38844836, 0.68121153, 0.21624038, -1.77549291, 0.18574584, 0.90408206, -0.22868094, -0.45148617, -0.16055907, -1.48271382, -0.07961921, -0.65701288, -0.25778309, -0.72851723, 0.86755788, -1.4129281, -0.19686662, 1.25451696, -1.59823179, -1.16262913, 0.84965342, 0.61178929, -1.26162946, -0.95558548, 1.38712502, 0.81038797, 0.1435982, 0.1535247, 0.00469783, 0.03943123, 0.53865469, -0.69794613, 1.2722131, 0.01081287, 0.61428916, -0.00753688, 0.33604777, -0.65822184, -0.83206612, 0.56394553, -0.14227687, 1.25733399, -0.1135006, 0.02068172, -0.21710102, 1.00936651, 0.04366954, -1.60517013, -0.88120216, 1.91317511, -0.54577649, 0.62921578, 0.57512373, -0.31156173, 0.57847321, 0.14741525, 1.17016482, -0.68234873, -1.34005868, -0.22676668, -0.29179415, -0.15037467, 0.1572255, 0.72092402, -1.80008531, 1.11335325, -0.5861423, 0.15335107, -0.43153843, -0.81098813, 1.15529966, 0.04376111, -0.09540533, -0.38106504, 1.00102639, -0.5632515, 0.90788037, -0.33356044, 0.74308902, -0.20033565, -1.51847255, 0.95205748, 0.54009491, 1.1931591, 0.81655478, 0.87503016, 0.0973243, 0.47295785, 0.65468878, -1.44158995, -0.05122741, -0.347552, 0.66963655, 0.7266466, 1.59155345, -0.15860432, -0.24945745, 0.67483073, 0.18782829, -0.56960964, 1.16764832, -0.72244978, 0.55027384, -0.53218621, -0.11167067, 0.67364228, -0.59705222, -0.24946509, 0.20462716, -0.56092483, -0.65660709, -1.13806772, 0.70947856, -0.65793264, -0.50718778, -1.20698619, 0.32613355, 0.61786091, -0.34040576, -0.37327161, 1.19222152, -0.23447749, 0.0614714, 0.32951999, 1.06427121, 2.26385999, 0.23828916] + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [2, 2], "data": { "kind": "values", "values": [0, 2, 3, 1] } }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [3], + "data": { "kind": "values", "values": [0, 1, 3] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [2, 2, 2, 2], + "data": { + "kind": "values", + "values": [0.4781123, 0.82420444, 0.654424, 0.3995186, 0.5482078, 0.55570245, 0.4216576, 0.46001542, 0.4781123, 0.82420444, 0.654424, 0.3995186, 0.5482078, 0.55570245, 0.4216576, 0.46001542] + } + } + }, + "outputs": { + "outputT": { + "dtype": "float16", + "shape": [3, 16], + "data": { + "kind": "values", + "values": [-1.1923828, 0.38842773, 0.68115234, 0.21618652, -1.7753906, 0.18579102, 0.90429688, -0.2286377, -0.95556641, 1.3867188, 0.81054688, 0.14355469, 0.15356445, 0.004699707, 0.039428711, 0.53857422, -0.72265625, -0.49584961, 1.1572266, -0.098266602, -0.10211182, 0.93652344, -0.56201172, 0.56103516, -0.27758789, 0.89697266, -0.26831055, 0.27734375, 0.12451172, 1.0283203, 1.6679688, 0.35424805, -1.3427734, -0.76660156, 1.6884766, -0.41259766, 0.41162109, 0.68261719, -0.38598633, 0.57324219, -0.17407227, 0.57763672, -0.3046875, 0.51025391, -0.097045898, 0.98974609, 1.0234375, 0.47949219] + } + } + }, + "tolerance": 0.005, + "relTolerance": 0.005 + }, + { + "name": "ort_packed_qkv_headsize8_broadcast_attn_bias", + "provenance": { + "source": "onnxruntime/test/testdata/attention/packed_multihead_attention_test_data.txt", + "test": "PackedMultiHeadAttentionTest.PackedQKV_Padding_NoBias_BroadcastAttnBias_unfused" + }, + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [9, 2, 3, 8], + "data": { + "kind": "values", + "values": [-1.83615911, 0.08698978, 0.05601556, -1.1451025, -2.30377889, -0.39893439, 0.73342341, -0.09851928, 0.86949563, -0.10868365, -0.37917313, -1.23103046, 0.25640076, -1.50652349, 0.71594471, 0.49057019, -1.19203627, 0.38844836, 0.68121153, 0.21624038, -1.77549291, 0.18574584, 0.90408206, -0.22868094, -0.45148617, -0.16055907, -1.48271382, -0.07961921, -0.65701288, -0.25778309, -0.72851723, 0.86755788, -1.4129281, -0.19686662, 1.25451696, -1.59823179, -1.16262913, 0.84965342, 0.61178929, -1.26162946, -0.95558548, 1.38712502, 0.81038797, 0.1435982, 0.1535247, 0.00469783, 0.03943123, 0.53865469, -1.61145306, 1.15819526, 0.92267495, 0.05697182, -1.15679109, 0.30815905, -0.16966841, 1.59468949, -0.39776036, -0.32152289, 0.83949518, -0.68910766, 2.10605407, -1.13187802, -0.02778089, -0.46233135, 0.50737989, 1.88879848, -0.6633718, -1.2957499, -1.86634719, -1.01549423, -0.28819263, -0.0001958, 1.06674862, -0.94525659, -1.26431036, 0.54500824, 2.33555818, -1.55515289, -1.15762365, 0.94301772, -2.59193254, -0.02682346, 0.07309747, -1.76444602, -1.02990091, -0.9244284, 0.80414194, -1.34978604, -2.00984406, -0.95374167, 0.84587288, -1.78314638, -0.26245379, -1.30501473, -1.76663613, 0.81748939, -0.11946517, -0.52010381, 1.17592037, 1.18040121, 1.71579909, 0.74054289, -0.28932455, 0.43039253, 0.18744308, 0.95530021, 1.00306034, -0.06650767, 0.14434582, -0.07039906, -0.42095628, 1.47149456, 0.80457681, -0.33367968, -0.10504961, 0.27371234, -0.07084391, -0.40500754, -0.04883647, -0.94577336, -0.59887785, -1.49919224, 1.02610373, 0.28144905, 0.54606986, 0.8928315, -1.55426455, 0.81868863, -0.32872301, 0.28885615, -0.12510297, -0.35719556, 0.36152089, -0.06192274, -0.28507957, 0.75713438, 1.02707481, -0.13490768, -0.18182218, -0.9390018, 1.18116772, 1.98105478, -0.36385471, 1.03757906, 0.52380687, -0.98392582, -1.63426089, -0.48980984, 0.39993629, -0.29305443, 0.89878768, -0.56480503, -0.59390724, -0.20499256, 0.45794842, 0.49178654, -0.25790799, -0.32643789, 0.38682714, -0.2696268, 0.14796616, -0.8352797, 0.16718683, 1.49241018, 0.43081248, -1.43449056, 0.39671475, -0.70072436, 0.36021376, 0.2386084, -0.31765643, -0.26952648, -1.24825561, 1.21374047, 0.2382634, 0.63743669, 0.32060453, 0.06873609, 1.35627246, -0.55090475, 0.75872916, -0.7215752, -0.75826746, -0.4332965, 0.17739578, 0.32338822, -0.67136252, 0.01797877, -0.59793222, 0.78523242, 0.25322497, 1.50049233, -0.24714193, 1.01905107, -0.93621284, -0.84905517, -1.20235336, -2.049963, -0.01251672, -1.23131406, -1.35081804, -0.40014201, 0.2604157, 0.23221464, 0.90103126, -0.98436964, 0.78283572, -1.78756905, -0.67545807, 0.50556731, 0.18512395, -1.26351786, -1.42564654, -1.06456351, -1.15261269, -0.49967349, 0.75130051, 1.48657084, -0.17923722, -0.58394587, 0.90255487, -0.60496974, -0.31978402, 0.04127264, 0.56045622, 0.52480292, -0.67152452, -1.51145959, 0.3588618, -0.41121563, 0.61315238, -1.03636253, -1.6869067, -1.10258508, 0.07205302, 1.72984779, -0.86800003, -0.91829145, -0.97764057, -0.55795985, -0.74666536, 1.14945197, 0.85595274, 0.87929201, 0.52238762, -0.38138887, -0.09198012, 0.10379392, 0.32577509, -0.24800496, 0.61258811, -0.14612751, 0.82892537, -0.14666887, 0.26352236, 0.27397016, -0.85447711, 0.16153991, 1.38688803, -1.06463683, -1.42907071, 0.43985835, -0.49964419, 0.67612225, 0.03910975, 0.37068886, 0.3865315, -0.84676158, 0.88043237, -0.70293593, -1.8680588, 0.50196815, -0.54198223, -0.384848, -0.05891366, -0.79304564, -0.58239877, 0.26545495, -0.3829754, 0.79097033, -0.17095472, -0.31460449, 0.78414583, 0.25754923, 0.49601099, 1.01128983, -0.85820115, 1.31935477, 1.79820418, 0.49160564, -1.0499413, -0.65047348, 0.03673789, -1.4339056, 0.33562672, 0.16967399, 1.70193112, -0.31023103, -0.33280775, 0.70939517, -1.27211368, -0.58704966, -1.81665146, 0.03955793, 1.41204929, -1.20291567, 0.22243243, 2.67244053, 0.75709999, -0.4434697, 0.94684786, -0.61597395, -0.71324098, -0.2485882, -0.42040613, -0.47096899, -2.1841073, 1.29030848, 0.95546865, -0.0163781, -0.40429014, 0.48408982, 1.63168097, 0.71464306, 0.93015838, -0.16759235, -2.20110703, 0.78718776, -0.37711632, -0.30456182, -0.73898333, 0.39148647, -0.04324996, -1.18526721, -1.38802266, 0.82284755, 2.10404348, -0.25014257, 2.06826448, 0.6873188, -0.72305721, -3.06288505, 1.35467649, -2.27782655, 1.27983427, 1.74217772, -1.50595713, 1.5257833, 0.56935966, -0.94778556, -1.28090227, -0.02918145, 0.3356843, -3.3641305, 1.40020049, 1.05561769, -1.90628219, 1.3822602, 0.99804544, 1.02540088, -2.04115343, 0.01709199, 1.14326739, -2.85616302, -1.63115931, 1.21209049, -2.88173485, 0.29401016, 0.99273205, 2.60591888, 1.3968792, -0.34032899, 0.35641205, 0.91440523, -0.69939649, 2.6461401, -1.05158651, -3.61572313, -2.07312417, 0.55376863, -0.07229304, 0.0295074, -1.17457461, 0.59849799, 0.96896434, 1.49846828, -1.48152626, -0.40123838, 0.66036439, -2.08949637, 0.93013203, 1.78407586, 0.65847164, -1.24047744, -0.11745319, 0.80385536, -0.55450106, -0.64750642, -0.84041762, 0.10586573, 0.56384206, 0.21395147, 1.26861954, 2.0200429, -0.83739442, -0.11827546, -0.42885846, -1.5538944, -0.16418101, 0.3018229, 0.76461935, 0.89762378, -0.7026118, 1.31333566, 0.8644017, -0.55341989, -1.74471772, 0.38858974, 0.77225429, -0.47355813, 0.59074765, -0.50501788, -1.72981727, -1.25862873, -1.04708695, 1.049909, 0.61408597, 0.48327276, 0.61544299, -0.57864964, -0.80768973, 0.39645281] + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [2, 8], + "data": { "kind": "values", "values": [0, 8, 9, 10, 11, 12, 13, 14, 15, 1, 2, 3, 4, 5, 6, 7] } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [3], + "data": { "kind": "values", "values": [0, 1, 9] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 2, 8, 8], + "data": { + "kind": "values", + "values": [0.09734076, -0.01747033, 0.008497253, -0.03361112, -0.028750911, -0.017142132, -0.11563814, 0.10432467, 0.057628587, 0.030893803, -0.096876964, 0.11924802, -0.009177148, 0.05799888, -0.030559167, 0.034150958, 0.07427484, 0.028848544, -0.031371966, 0.07186346, -0.093020484, -0.066411436, 0.06858949, 0.07350862, 0.008785307, -0.07727124, 0.09184219, 0.081774, 0.008513406, -0.04256621, -0.078802094, -0.09632437, -0.0073524266, 0.035430342, -0.01940021, 0.05564849, -0.03556633, -0.11830491, -0.05839123, -0.05537042, 0.06146671, 0.0026960075, -0.07718696, -0.06268193, 0.039188996, 0.07048777, -0.0009800941, -0.04509197, 0.04516627, -0.09005861, 0.06470743, 0.12054373, 0.03366275, 0.009849519, 0.015899211, 0.11534238, -0.002090156, -0.0851126, -0.05993013, 0.030745044, -0.02415888, 0.06568785, -0.012787953, 0.021764219, -0.07153495, 0.072637364, 0.09073421, 0.041491434, -0.11252555, -0.059031114, 0.11179088, 0.0655573, -0.11293891, 0.05239816, -0.110543296, -0.017368436, -0.050893307, 0.04980643, -0.12137364, -0.08896659, 0.04159136, -0.021769747, 0.018932432, 0.087893024, -0.014453113, 0.012075469, 0.0026388168, -0.07094294, 0.035509557, 0.065090835, 0.115079954, -0.11314371, 0.121840075, 0.030929685, -0.051646978, -0.10540052, -0.08004643, -0.089356855, -0.112021506, 0.07381505, -0.06653316, -0.11657183, 0.08264731, 0.10898076, -0.006200552, -0.0960899, -0.015346348, -0.047637567, 0.051793993, -0.09059112, -0.034093216, 0.108242184, -0.049242377, -0.12165685, -0.09124553, 0.05479528, -0.048966378, -0.027208641, 0.10195236, 0.047513425, 0.013226762, -0.07403794, 0.06855075, -0.06551643, -0.084110215, 0.11237715, 0.07026932, -0.014076158] + } + } + }, + "outputs": { + "outputT": { + "dtype": "float16", + "shape": [9, 16], + "data": { + "kind": "values", + "values": [-1.1923828, 0.38842773, 0.68115234, 0.21618652, -1.7753906, 0.18579102, 0.90429688, -0.2286377, -0.95556641, 1.3867188, 0.81054688, 0.14355469, 0.15356445, 0.004699707, 0.039428711, 0.53857422, 0.46826172, -0.38012695, 0.23950195, 0.50976562, -0.35839844, -0.4597168, 0.0065727234, -0.62060547, -0.28198242, -0.4128418, -0.44287109, 0.22363281, 0.062408447, 0.058746338, -0.75292969, 0.82275391, 0.18103027, 0.17089844, 0.31738281, -0.098022461, -1.0439453, -0.43823242, -0.19689941, -0.25537109, -0.38745117, -0.32739258, -0.40332031, 0.23425293, 0.13305664, -0.3503418, -1.0107422, 0.859375, 0.14135742, -0.1895752, 0.50634766, 0.25634766, -1.0400391, -0.37524414, -0.16821289, -0.30371094, -0.68164062, -1.4511719, -0.52636719, 0.19763184, 0.082946777, 0.1776123, -0.97021484, 0.73828125, 0.23144531, -0.75341797, 0.52148438, 0.52148438, -0.92041016, -0.27001953, 0.054473877, -0.19238281, -0.68212891, -1.3339844, -0.75, 0.37451172, -0.060058594, 0.0034179688, -0.8671875, 0.65429688, 0.20629883, -0.37646484, 0.37646484, 0.13183594, -0.94873047, -0.34375, -0.059875488, -0.22937012, -0.58105469, -0.63671875, -0.50732422, 0.30419922, 0.086791992, -0.23132324, -0.91210938, 0.74902344, 0.49804688, -1.3681641, 0.75146484, 1.0007825, -0.93847656, 0.23278809, 0.44604492, 0.047668457, -1.1943359, -0.75732422, 0.48681641, -0.67480469, -0.0028400421, -0.37817383, -1.2705078, 0.81201172, 0.13586426, 0.0039863586, 0.36181641, 0.22192383, -0.89648438, -0.609375, -0.26049805, -0.43920898, -1.046875, -0.19360352, 0.30297852, -0.20007324, 0.20483398, -0.50097656, -1.1142578, 0.71923828, 0.37207031, -1.8447266, 0.88818359, 0.71191406, -1.3359375, 0.56396484, 0.52001953, 0.36254883, -0.73681641, -0.90722656, -0.29516602, 0.021835327, 0.030609131, -0.098083496, -0.96777344, 0.75927734] + } + } + }, + "tolerance": 0.005, + "relTolerance": 0.005 + }, + { + "name": "ort_q_k_v_bias_attn_bias_no_padding", + "provenance": { + "source": "onnxruntime/test/testdata/attention/attention_test_data.txt", + "test": "PackedMultiHeadAttentionTest.Q_K_V_NoPadding_Bias_AttnBias_unfused" + }, + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [8, 8], + "data": { + "kind": "values", + "values": [-0.029273793, 0.079709493, 0.064531095, 0.24270254, -0.28326464, 0.20984903, -0.10173888, 0.18373983, 0.089472905, -0.0063416883, -0.049477674, 0.36512995, -0.23620239, 0.1464397, 0.068258412, 0.31627196, 0.12436871, -0.0075563118, -0.11576633, 0.41008925, -0.19456652, 0.20145792, 0.11790096, 0.39789933, 0.002485469, 0.029660821, -0.043821491, 0.3892332, -0.26994205, 0.14530671, 0.12950704, 0.36185294, -0.029273793, 0.079709493, 0.064531095, 0.24270254, -0.28326464, 0.20984903, -0.10173888, 0.18373983, 0.089472905, -0.0063416883, -0.049477674, 0.36512995, -0.23620239, 0.1464397, 0.068258412, 0.31627196, 0.12436871, -0.0075563118, -0.11576633, 0.41008925, -0.19456652, 0.20145792, 0.11790096, 0.39789933, 0.002485469, 0.029660821, -0.043821491, 0.3892332, -0.26994205, 0.14530671, 0.12950704, 0.36185294] + } + }, + "keyT": { + "dtype": "float16", + "shape": [8, 8], + "data": { + "kind": "values", + "values": [-0.32538497, 0.34121913, -0.18170178, -0.015152611, 0.20429322, 0.25979176, 0.21269324, 0.0025638193, -0.24246037, 0.21112341, -0.36959589, -0.16091451, 0.24183474, 0.18856162, 0.094487116, -0.3053959, -0.35736683, 0.29276621, -0.4217523, -0.20031664, 0.33148992, 0.26928401, 0.19360018, -0.39494509, -0.28043351, 0.24279942, -0.29154932, -0.13657911, 0.31932494, 0.3500579, 0.027172565, -0.19327414, -0.32538497, 0.34121913, -0.18170178, -0.015152611, 0.20429322, 0.25979176, 0.21269324, 0.0025638193, -0.24246037, 0.21112341, -0.36959589, -0.16091451, 0.24183474, 0.18856162, 0.094487116, -0.3053959, -0.35736683, 0.29276621, -0.4217523, -0.20031664, 0.33148992, 0.26928401, 0.19360018, -0.39494509, -0.28043351, 0.24279942, -0.29154932, -0.13657911, 0.31932494, 0.3500579, 0.027172565, -0.19327414] + } + }, + "valueT": { + "dtype": "float16", + "shape": [8, 8], + "data": { + "kind": "values", + "values": [0.56916672, -0.2443777, 0.47111356, -0.52134115, 0.010381341, 0.0696759, -0.071910433, -0.35201436, 0.70809275, -0.24479815, 0.41633749, -0.34744334, -0.0044222325, 0.25929695, -0.087832771, -0.281232, 0.90039468, -0.28931504, 0.56394172, -0.43948689, -0.05856207, 0.33713666, -0.10320446, -0.38833332, 0.76054728, -0.29080144, 0.50414616, -0.42371163, -0.047198489, 0.31959397, -0.22683662, -0.30321664, 0.56916672, -0.2443777, 0.47111356, -0.52134115, 0.010381341, 0.0696759, -0.071910433, -0.35201436, 0.70809275, -0.24479815, 0.41633749, -0.34744334, -0.0044222325, 0.25929695, -0.087832771, -0.281232, 0.90039468, -0.28931504, 0.56394172, -0.43948689, -0.05856207, 0.33713666, -0.10320446, -0.38833332, 0.76054728, -0.29080144, 0.50414616, -0.42371163, -0.047198489, 0.31959397, -0.22683662, -0.30321664] + } + }, + "biasT": { + "dtype": "float16", + "shape": [24], + "data": { + "kind": "values", + "values": [-0.38124341, 0.02696526, -0.11914945, -0.43795273, 0.04772711, -0.03419551, -0.30606642, 0.42656231, -0.25891554, 0.13431972, 0.22861153, 0.06360734, -0.10595283, -0.42839217, 0.28931111, -0.13180739, 0.27079183, 0.42074734, -0.40314156, -0.43726659, -0.40546918, 0.06927037, 0.16979086, 0.41458064] + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [1, 8], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5, 6, 7] } + }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 8] } }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 2, 8, 8], + "data": { + "kind": "values", + "values": [-10.808288, -10.887209, 7.8799553, -4.6565766, -1.6700006, -0.033962168, 7.4929152, 10.944146, 8.640254, -18.862164, -3.1202927, -6.3049207, 3.4508536, 11.722519, 3.3550568, -5.4888172, -2.0828252, -13.241742, 2.9868939, 1.4455698, -15.262972, -10.457437, -8.4519463, -4.4281874, 10.212368, -0.28622282, 12.087646, 6.5218501, 8.1785011, 13.985523, -8.2068987, 5.4260745, -10.808288, -10.887209, 7.8799553, -4.6565766, -1.6700006, -0.033962168, 7.4929152, 10.944146, 8.640254, -18.862164, -3.1202927, -6.3049207, 3.4508536, 11.722519, 3.3550568, -5.4888172, -2.0828252, -13.241742, 2.9868939, 1.4455698, -15.262972, -10.457437, -8.4519463, -4.4281874, 10.212368, -0.28622282, 12.087646, 6.5218501, 8.1785011, 13.985523, -8.2068987, 5.4260745, -10.808288, -10.887209, 7.8799553, -4.6565766, -1.6700006, -0.033962168, 7.4929152, 10.944146, 8.640254, -18.862164, -3.1202927, -6.3049207, 3.4508536, 11.722519, 3.3550568, -5.4888172, -2.0828252, -13.241742, 2.9868939, 1.4455698, -15.262972, -10.457437, -8.4519463, -4.4281874, 10.212368, -0.28622282, 12.087646, 6.5218501, 8.1785011, 13.985523, -8.2068987, 5.4260745, -10.808288, -10.887209, 7.8799553, -4.6565766, -1.6700006, -0.033962168, 7.4929152, 10.944146, 8.640254, -18.862164, -3.1202927, -6.3049207, 3.4508536, 11.722519, 3.3550568, -5.4888172, -2.0828252, -13.241742, 2.9868939, 1.4455698, -15.262972, -10.457437, -8.4519463, -4.4281874, 10.212368, -0.28622282, 12.087646, 6.5218501, 8.1785011, 13.985523, -8.2068987, 5.4260745] + } + } + }, + "outputs": { + "outputT": { + "dtype": "float16", + "shape": [8, 8], + "data": { + "kind": "values", + "values": [1.0419922, 0.13000488, 0.10528564, -0.86230469, -0.45336914, 0.39013672, -0.048858643, 0.10571289, 0.97265625, 0.17590332, 0.015625, -0.79248047, -0.40917969, 0.31933594, 0.082763672, 0.12976074, 1.1455078, 0.13134766, 0.15014648, -0.87451172, -0.46142578, 0.40161133, 0.04309082, 0.042663574, 1.0009766, 0.17004395, 0.033752441, -0.80078125, -0.41625977, 0.33349609, 0.080383301, 0.11846924, 1.0419922, 0.13000488, 0.10528564, -0.86230469, -0.45336914, 0.39013672, -0.048858643, 0.10571289, 0.97265625, 0.17590332, 0.015625, -0.79248047, -0.40917969, 0.31933594, 0.082763672, 0.12976074, 1.1455078, 0.13134766, 0.15014648, -0.87451172, -0.46142578, 0.40161133, 0.04309082, 0.042663574, 1.0009766, 0.17004395, 0.033752441, -0.80078125, -0.41625977, 0.33349609, 0.080383301, 0.11846924] + } + } + }, + "tolerance": 0.005, + "relTolerance": 0.005 + }, + { + "name": "packed_b3_h32", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [7, 2, 3, 32], + "data": { "kind": "fillFloat32", "sinStep": 0.13, "cosStep": 0.264, "scale": 0.6 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 4], + "data": { "kind": "values", "values": [0, 4, 5, 8, 9, 10, 11, 1, 2, 3, 6, 7] } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 3, 7] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [7, 64] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "separate_b3_h32", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [7, 64], + "data": { "kind": "fillFloat32", "sinStep": 0.14, "cosStep": 0.281, "scale": 0.6 } + }, + "keyT": { + "dtype": "float32", + "shape": [7, 64], + "data": { "kind": "fillFloat32", "sinStep": 0.229, "cosStep": 0.34299999999999997, "scale": 0.6 } + }, + "valueT": { + "dtype": "float32", + "shape": [7, 64], + "data": { "kind": "fillFloat32", "sinStep": 0.127, "cosStep": 0.359, "scale": 0.6 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 4], + "data": { "kind": "values", "values": [0, 4, 5, 8, 9, 10, 11, 1, 2, 3, 6, 7] } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 3, 7] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [7, 64] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "packed_attn_bias_b3_h32", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [7, 2, 3, 32], + "data": { "kind": "fillFloat32", "sinStep": 0.15, "cosStep": 0.29800000000000004, "scale": 0.6 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 4], + "data": { "kind": "values", "values": [0, 4, 5, 8, 9, 10, 11, 1, 2, 3, 6, 7] } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 3, 7] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 2, 4, 4], + "data": { "kind": "fillFloat32", "sinStep": 0.514, "cosStep": 0.222, "scale": 1.5 } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [7, 64] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "separate_attn_bias_b3_h32", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [7, 64], + "data": { "kind": "fillFloat32", "sinStep": 0.16, "cosStep": 0.315, "scale": 0.6 } + }, + "keyT": { + "dtype": "float32", + "shape": [7, 64], + "data": { "kind": "fillFloat32", "sinStep": 0.255, "cosStep": 0.365, "scale": 0.6 } + }, + "valueT": { + "dtype": "float32", + "shape": [7, 64], + "data": { "kind": "fillFloat32", "sinStep": 0.165, "cosStep": 0.40499999999999997, "scale": 0.6 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 4], + "data": { "kind": "values", "values": [0, 4, 5, 8, 9, 10, 11, 1, 2, 3, 6, 7] } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 3, 7] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 2, 4, 4], + "data": { "kind": "fillFloat32", "sinStep": 0.535, "cosStep": 0.23500000000000001, "scale": 1.5 } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [7, 64] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "packed_bias_b3_h32", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [7, 2, 3, 32], + "data": { "kind": "fillFloat32", "sinStep": 0.16999999999999998, "cosStep": 0.332, "scale": 0.6 } + }, + "biasT": { + "dtype": "float32", + "shape": [192], + "data": { "kind": "fillFloat32", "sinStep": 0.412, "cosStep": 0.30400000000000005, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 4], + "data": { "kind": "values", "values": [0, 4, 5, 8, 9, 10, 11, 1, 2, 3, 6, 7] } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 3, 7] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [7, 64] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "separate_bias_b3_h32", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [7, 64], + "data": { "kind": "fillFloat32", "sinStep": 0.18, "cosStep": 0.34900000000000003, "scale": 0.6 } + }, + "keyT": { + "dtype": "float32", + "shape": [7, 64], + "data": { "kind": "fillFloat32", "sinStep": 0.281, "cosStep": 0.387, "scale": 0.6 } + }, + "valueT": { + "dtype": "float32", + "shape": [7, 64], + "data": { "kind": "fillFloat32", "sinStep": 0.203, "cosStep": 0.45099999999999996, "scale": 0.6 } + }, + "biasT": { + "dtype": "float32", + "shape": [192], + "data": { "kind": "fillFloat32", "sinStep": 0.419, "cosStep": 0.333, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 4], + "data": { "kind": "values", "values": [0, 4, 5, 8, 9, 10, 11, 1, 2, 3, 6, 7] } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 3, 7] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [7, 64] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "packed_bias_attn_bias_b3_h32", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [7, 2, 3, 32], + "data": { "kind": "fillFloat32", "sinStep": 0.19, "cosStep": 0.366, "scale": 0.6 } + }, + "biasT": { + "dtype": "float32", + "shape": [192], + "data": { "kind": "fillFloat32", "sinStep": 0.426, "cosStep": 0.362, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 4], + "data": { "kind": "values", "values": [0, 4, 5, 8, 9, 10, 11, 1, 2, 3, 6, 7] } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 3, 7] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 2, 4, 4], + "data": { "kind": "fillFloat32", "sinStep": 0.598, "cosStep": 0.274, "scale": 1.5 } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [7, 64] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "separate_bias_attn_bias_b3_h32", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [7, 64], + "data": { "kind": "fillFloat32", "sinStep": 0.2, "cosStep": 0.383, "scale": 0.6 } + }, + "keyT": { + "dtype": "float32", + "shape": [7, 64], + "data": { "kind": "fillFloat32", "sinStep": 0.307, "cosStep": 0.409, "scale": 0.6 } + }, + "valueT": { + "dtype": "float32", + "shape": [7, 64], + "data": { "kind": "fillFloat32", "sinStep": 0.241, "cosStep": 0.497, "scale": 0.6 } + }, + "biasT": { + "dtype": "float32", + "shape": [192], + "data": { "kind": "fillFloat32", "sinStep": 0.433, "cosStep": 0.391, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 4], + "data": { "kind": "values", "values": [0, 4, 5, 8, 9, 10, 11, 1, 2, 3, 6, 7] } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 3, 7] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 2, 4, 4], + "data": { "kind": "fillFloat32", "sinStep": 0.619, "cosStep": 0.28700000000000003, "scale": 1.5 } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [7, 64] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "packed_bias_attn_bias_b3_h16", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [7, 2, 3, 16], + "data": { "kind": "fillFloat32", "sinStep": 0.21000000000000002, "cosStep": 0.4, "scale": 0.6 } + }, + "biasT": { + "dtype": "float32", + "shape": [96], + "data": { "kind": "fillFloat32", "sinStep": 0.44, "cosStep": 0.42000000000000004, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 4], + "data": { "kind": "values", "values": [0, 4, 5, 8, 9, 10, 11, 1, 2, 3, 6, 7] } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 3, 7] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 2, 4, 4], + "data": { "kind": "fillFloat32", "sinStep": 0.64, "cosStep": 0.30000000000000004, "scale": 1.5 } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [7, 32] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "separate_bias_attn_bias_b3_h16", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [7, 32], + "data": { "kind": "fillFloat32", "sinStep": 0.22, "cosStep": 0.41700000000000004, "scale": 0.6 } + }, + "keyT": { + "dtype": "float32", + "shape": [7, 32], + "data": { "kind": "fillFloat32", "sinStep": 0.33299999999999996, "cosStep": 0.431, "scale": 0.6 } + }, + "valueT": { + "dtype": "float32", + "shape": [7, 32], + "data": { "kind": "fillFloat32", "sinStep": 0.279, "cosStep": 0.5429999999999999, "scale": 0.6 } + }, + "biasT": { + "dtype": "float32", + "shape": [96], + "data": { "kind": "fillFloat32", "sinStep": 0.447, "cosStep": 0.449, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 4], + "data": { "kind": "values", "values": [0, 4, 5, 8, 9, 10, 11, 1, 2, 3, 6, 7] } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 3, 7] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 2, 4, 4], + "data": { "kind": "fillFloat32", "sinStep": 0.661, "cosStep": 0.313, "scale": 1.5 } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [7, 32] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "packed_bias_attn_bias_b3_h32_f16", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [7, 2, 3, 32], + "data": { + "kind": "fillFloat32", + "sinStep": 0.22999999999999998, + "cosStep": 0.43400000000000005, + "scale": 0.6 + } + }, + "biasT": { + "dtype": "float16", + "shape": [192], + "data": { "kind": "fillFloat32", "sinStep": 0.454, "cosStep": 0.47800000000000004, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 4], + "data": { "kind": "values", "values": [0, 4, 5, 8, 9, 10, 11, 1, 2, 3, 6, 7] } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 3, 7] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 2, 4, 4], + "data": { "kind": "fillFloat32", "sinStep": 0.6819999999999999, "cosStep": 0.326, "scale": 1.5 } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [7, 64] } }, + "tolerance": 0.004, + "relTolerance": 0.004 + }, + { + "name": "separate_bias_attn_bias_b3_h32_f16", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [7, 64], + "data": { "kind": "fillFloat32", "sinStep": 0.24, "cosStep": 0.45100000000000007, "scale": 0.6 } + }, + "keyT": { + "dtype": "float16", + "shape": [7, 64], + "data": { "kind": "fillFloat32", "sinStep": 0.359, "cosStep": 0.45299999999999996, "scale": 0.6 } + }, + "valueT": { + "dtype": "float16", + "shape": [7, 64], + "data": { "kind": "fillFloat32", "sinStep": 0.317, "cosStep": 0.589, "scale": 0.6 } + }, + "biasT": { + "dtype": "float16", + "shape": [192], + "data": { "kind": "fillFloat32", "sinStep": 0.46099999999999997, "cosStep": 0.507, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 4], + "data": { "kind": "values", "values": [0, 4, 5, 8, 9, 10, 11, 1, 2, 3, 6, 7] } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 3, 7] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 2, 4, 4], + "data": { "kind": "fillFloat32", "sinStep": 0.7030000000000001, "cosStep": 0.33899999999999997, "scale": 1.5 } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [7, 64] } }, + "tolerance": 0.004, + "relTolerance": 0.004 + }, + { + "name": "packed_span_tiles_b5_h32", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [24, 2, 3, 32], + "data": { "kind": "fillFloat32", "sinStep": 0.25, "cosStep": 0.468, "scale": 0.6 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [5, 8], + "data": { + "kind": "values", + "values": [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 16, 17, 18, 19, 20, 24, 32, 33, 34, 35, 36, 37, 38, 11, 12, 13, 14, 15, 21, 22, 23, 25, 26, 27, 28, 29, 30, 31, 39] + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [6], + "data": { "kind": "values", "values": [0, 8, 11, 16, 17, 24] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [24, 64] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "packed_bias_attn_bias_long_b2_h32", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [40, 2, 3, 32], + "data": { "kind": "fillFloat32", "sinStep": 0.26, "cosStep": 0.485, "scale": 0.6 } + }, + "biasT": { + "dtype": "float32", + "shape": [192], + "data": { "kind": "fillFloat32", "sinStep": 0.475, "cosStep": 0.565, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [2, 40], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/packed_bias_attn_bias_long_b2_h32_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [3], + "data": { "kind": "values", "values": [0, 33, 40] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [2, 2, 40, 40], + "data": { "kind": "fillFloat32", "sinStep": 0.745, "cosStep": 0.365, "scale": 1.5 } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [40, 64] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "separate_long_b2_h32", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [40, 64], + "data": { "kind": "fillFloat32", "sinStep": 0.27, "cosStep": 0.502, "scale": 0.6 } + }, + "keyT": { + "dtype": "float32", + "shape": [40, 64], + "data": { "kind": "fillFloat32", "sinStep": 0.398, "cosStep": 0.486, "scale": 0.6 } + }, + "valueT": { + "dtype": "float32", + "shape": [40, 64], + "data": { "kind": "fillFloat32", "sinStep": 0.374, "cosStep": 0.6579999999999999, "scale": 0.6 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [2, 40], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/packed_bias_attn_bias_long_b2_h32_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [3], + "data": { "kind": "values", "values": [0, 33, 40] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [40, 64] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "equal_lengths_b2_h32", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [6, 2, 3, 32], + "data": { "kind": "fillFloat32", "sinStep": 0.28, "cosStep": 0.519, "scale": 0.6 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [2, 3], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5] } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [3], + "data": { "kind": "values", "values": [0, 3, 6] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [6, 64] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "separate_v_head_dim_b3_h32", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [7, 64], + "data": { "kind": "fillFloat32", "sinStep": 0.29, "cosStep": 0.536, "scale": 0.6 } + }, + "keyT": { + "dtype": "float32", + "shape": [7, 64], + "data": { "kind": "fillFloat32", "sinStep": 0.424, "cosStep": 0.508, "scale": 0.6 } + }, + "valueT": { + "dtype": "float32", + "shape": [7, 32], + "data": { "kind": "fillFloat32", "sinStep": 0.412, "cosStep": 0.704, "scale": 0.6 } + }, + "biasT": { + "dtype": "float32", + "shape": [160], + "data": { "kind": "fillFloat32", "sinStep": 0.496, "cosStep": 0.652, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 4], + "data": { "kind": "values", "values": [0, 4, 5, 8, 9, 10, 11, 1, 2, 3, 6, 7] } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 3, 7] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [7, 32] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "separate_odd_head_dim_b3", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [7, 6], + "data": { "kind": "fillFloat32", "sinStep": 0.3, "cosStep": 0.553, "scale": 0.6 } + }, + "keyT": { + "dtype": "float32", + "shape": [7, 6], + "data": { "kind": "fillFloat32", "sinStep": 0.437, "cosStep": 0.519, "scale": 0.6 } + }, + "valueT": { + "dtype": "float32", + "shape": [7, 4], + "data": { "kind": "fillFloat32", "sinStep": 0.431, "cosStep": 0.727, "scale": 0.6 } + }, + "biasT": { + "dtype": "float32", + "shape": [16], + "data": { "kind": "fillFloat32", "sinStep": 0.503, "cosStep": 0.681, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 4], + "data": { "kind": "values", "values": [0, 4, 5, 8, 9, 10, 11, 1, 2, 3, 6, 7] } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 3, 7] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 2, 4, 4], + "data": { "kind": "fillFloat32", "sinStep": 0.829, "cosStep": 0.41700000000000004, "scale": 1.5 } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [7, 4] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "packed_head_dim_8_b3", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [7, 3, 3, 8], + "data": { "kind": "fillFloat32", "sinStep": 0.31, "cosStep": 0.5700000000000001, "scale": 0.6 } + }, + "biasT": { + "dtype": "float32", + "shape": [72], + "data": { "kind": "fillFloat32", "sinStep": 0.51, "cosStep": 0.7100000000000001, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 4], + "data": { "kind": "values", "values": [0, 4, 5, 8, 9, 10, 11, 1, 2, 3, 6, 7] } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 3, 7] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 3, 4, 4], + "data": { "kind": "fillFloat32", "sinStep": 0.8500000000000001, "cosStep": 0.43000000000000005, "scale": 1.5 } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [7, 24] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "packed_broadcast_attn_bias_b3_h32", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [7, 2, 3, 32], + "data": { "kind": "fillFloat32", "sinStep": 0.32, "cosStep": 0.5870000000000001, "scale": 0.6 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 4], + "data": { "kind": "values", "values": [0, 4, 5, 8, 9, 10, 11, 1, 2, 3, 6, 7] } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 3, 7] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 1, 4, 4], + "data": { "kind": "fillFloat32", "sinStep": 0.871, "cosStep": 0.44299999999999995, "scale": 1.5 } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [7, 64] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "separate_broadcast_head_attn_bias_b3_h32", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [7, 64], + "data": { "kind": "fillFloat32", "sinStep": 0.33, "cosStep": 0.604, "scale": 0.6 } + }, + "keyT": { + "dtype": "float32", + "shape": [7, 64], + "data": { "kind": "fillFloat32", "sinStep": 0.476, "cosStep": 0.552, "scale": 0.6 } + }, + "valueT": { + "dtype": "float32", + "shape": [7, 64], + "data": { "kind": "fillFloat32", "sinStep": 0.488, "cosStep": 0.796, "scale": 0.6 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 4], + "data": { "kind": "values", "values": [0, 4, 5, 8, 9, 10, 11, 1, 2, 3, 6, 7] } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 3, 7] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 4, 4], + "data": { "kind": "fillFloat32", "sinStep": 0.892, "cosStep": 0.45599999999999996, "scale": 1.5 } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [7, 64] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "single_token_h32", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [1, 2, 3, 32], + "data": { "kind": "fillFloat32", "sinStep": 0.34, "cosStep": 0.621, "scale": 0.6 } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [1, 64] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "single_sequence_b1_h32", + "attrs": { "num_heads": 4 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [5, 128], + "data": { "kind": "fillFloat32", "sinStep": 0.35, "cosStep": 0.638, "scale": 0.6 } + }, + "keyT": { + "dtype": "float32", + "shape": [5, 128], + "data": { "kind": "fillFloat32", "sinStep": 0.502, "cosStep": 0.5740000000000001, "scale": 0.6 } + }, + "valueT": { + "dtype": "float32", + "shape": [5, 128], + "data": { "kind": "fillFloat32", "sinStep": 0.526, "cosStep": 0.8420000000000001, "scale": 0.6 } + }, + "biasT": { + "dtype": "float32", + "shape": [384], + "data": { "kind": "fillFloat32", "sinStep": 0.538, "cosStep": 0.8260000000000001, "scale": 0.4 } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 5], "data": { "kind": "values", "values": [0, 1, 2, 3, 4] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 5] } }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 4, 5, 5], + "data": { "kind": "fillFloat32", "sinStep": 0.9339999999999999, "cosStep": 0.482, "scale": 1.5 } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [5, 128] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "explicit_scale_b3_h32", + "attrs": { "num_heads": 2, "scale": 0.125 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [7, 2, 3, 32], + "data": { "kind": "fillFloat32", "sinStep": 0.36, "cosStep": 0.655, "scale": 0.6 } + }, + "biasT": { + "dtype": "float32", + "shape": [192], + "data": { "kind": "fillFloat32", "sinStep": 0.545, "cosStep": 0.8550000000000001, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 4], + "data": { "kind": "values", "values": [0, 4, 5, 8, 9, 10, 11, 1, 2, 3, 6, 7] } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 3, 7] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [7, 64] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-tile-reuse-float32", + "attrs": { "num_heads": 4 }, + "inputs": { + "queryT": { + "shape": [133, 4, 3, 128], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 1.513, "cosStep": 0.882, "scale": 0.5 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [5, 129], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/pmha_tile_reuse_float32_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [6], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 133] } + } + }, + "outputs": { "outputT": { "shape": [133, 512], "dtype": "float32" } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-tile-reuse-float32-bias", + "attrs": { "num_heads": 4 }, + "inputs": { + "queryT": { + "shape": [133, 4, 3, 128], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 1.513, "cosStep": 0.882, "scale": 0.5 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [5, 129], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/pmha_tile_reuse_float32_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [6], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 133] } + }, + "attentionBiasT": { + "shape": [1, 1, 129, 129], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 1.173, "cosStep": 0.819, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [133, 512], "dtype": "float32" } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-tile-reuse-float16", + "attrs": { "num_heads": 4 }, + "inputs": { + "queryT": { + "shape": [133, 4, 3, 128], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 1.513, "cosStep": 0.882, "scale": 0.5 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [5, 129], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/pmha_tile_reuse_float32_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [6], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 133] } + } + }, + "outputs": { "outputT": { "shape": [133, 512], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-tile-reuse-float16-bias", + "attrs": { "num_heads": 4 }, + "inputs": { + "queryT": { + "shape": [133, 4, 3, 128], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 1.513, "cosStep": 0.882, "scale": 0.5 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [5, 129], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/pmha_tile_reuse_float32_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [6], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 133] } + }, + "attentionBiasT": { + "shape": [1, 1, 129, 129], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 1.173, "cosStep": 0.819, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [133, 512], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-general-float32-0-t129-h3-d16-v16", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [129, 3, 3, 16], + "data": { "kind": "fillFloat32", "sinStep": 0.85, "cosStep": 1.449, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 65], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_3_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 63, 64, 129] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [129, 48] } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-general-float32-1-t129-h3-d32-v32", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [129, 96], + "data": { "kind": "fillFloat32", "sinStep": 1.173, "cosStep": 0.357, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 65], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_3_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 63, 64, 129] } + }, + "keyT": { + "dtype": "float32", + "shape": [129, 96], + "data": { "kind": "fillFloat32", "sinStep": 1.19, "cosStep": 0.378, "scale": 0.4 } + }, + "valueT": { + "dtype": "float32", + "shape": [129, 96], + "data": { "kind": "fillFloat32", "sinStep": 1.207, "cosStep": 0.399, "scale": 0.4 } + }, + "biasT": { + "dtype": "float32", + "shape": [288], + "data": { "kind": "fillFloat32", "sinStep": 1.224, "cosStep": 0.42, "scale": 0.2 } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 1, 65, 65], + "data": { "kind": "fillFloat32", "sinStep": 1.241, "cosStep": 0.441, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [129, 96] } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-general-float32-2-t129-h3-d48-v48", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [129, 3, 3, 48], + "data": { "kind": "fillFloat32", "sinStep": 1.496, "cosStep": 0.756, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 65], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_3_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 63, 64, 129] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 3, 65, 65], + "data": { "kind": "fillFloat32", "sinStep": 0.051, "cosStep": 0.84, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [129, 144] } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-general-float32-3-t129-h3-d64-v64", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [129, 192], + "data": { "kind": "fillFloat32", "sinStep": 0.306, "cosStep": 1.155, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 65], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_3_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 63, 64, 129] } + }, + "keyT": { + "dtype": "float32", + "shape": [129, 192], + "data": { "kind": "fillFloat32", "sinStep": 0.323, "cosStep": 1.176, "scale": 0.4 } + }, + "valueT": { + "dtype": "float32", + "shape": [129, 192], + "data": { "kind": "fillFloat32", "sinStep": 0.34, "cosStep": 1.197, "scale": 0.4 } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 65, 65], + "data": { "kind": "fillFloat32", "sinStep": 0.374, "cosStep": 1.239, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [129, 192] } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-general-float32-4-t129-h3-d96-v96", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [129, 3, 3, 96], + "data": { "kind": "fillFloat32", "sinStep": 0.629, "cosStep": 0.063, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 65], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_3_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 63, 64, 129] } + }, + "biasT": { + "dtype": "float32", + "shape": [864], + "data": { "kind": "fillFloat32", "sinStep": 0.68, "cosStep": 0.126, "scale": 0.2 } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 3, 65, 65], + "data": { "kind": "fillFloat32", "sinStep": 0.697, "cosStep": 0.147, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [129, 288] } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-general-float32-5-t129-h3-d128-v128", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [129, 384], + "data": { "kind": "fillFloat32", "sinStep": 0.952, "cosStep": 0.462, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 65], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_3_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 63, 64, 129] } + }, + "keyT": { + "dtype": "float32", + "shape": [129, 384], + "data": { "kind": "fillFloat32", "sinStep": 0.969, "cosStep": 0.483, "scale": 0.4 } + }, + "valueT": { + "dtype": "float32", + "shape": [129, 384], + "data": { "kind": "fillFloat32", "sinStep": 0.986, "cosStep": 0.504, "scale": 0.4 } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [129, 384] } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-general-float32-6-t129-h3-d192-v192", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [129, 3, 3, 192], + "data": { "kind": "fillFloat32", "sinStep": 1.275, "cosStep": 0.861, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 65], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_3_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 63, 64, 129] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 1, 65, 65], + "data": { "kind": "fillFloat32", "sinStep": 1.343, "cosStep": 0.945, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [129, 576] } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-general-float32-7-t129-h3-d256-v256", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [129, 768], + "data": { "kind": "fillFloat32", "sinStep": 0.085, "cosStep": 1.26, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 65], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_3_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 63, 64, 129] } + }, + "keyT": { + "dtype": "float32", + "shape": [129, 768], + "data": { "kind": "fillFloat32", "sinStep": 0.102, "cosStep": 1.281, "scale": 0.4 } + }, + "valueT": { + "dtype": "float32", + "shape": [129, 768], + "data": { "kind": "fillFloat32", "sinStep": 0.119, "cosStep": 1.302, "scale": 0.4 } + }, + "biasT": { + "dtype": "float32", + "shape": [2304], + "data": { "kind": "fillFloat32", "sinStep": 0.136, "cosStep": 1.323, "scale": 0.2 } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 3, 65, 65], + "data": { "kind": "fillFloat32", "sinStep": 0.153, "cosStep": 1.344, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [129, 768] } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-general-float32-8-t1-h1-d32-v32", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [1, 1, 3, 32], + "data": { "kind": "fillFloat32", "sinStep": 0.408, "cosStep": 0.168, "scale": 0.4 } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [1, 32] } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-general-float32-9-t7-h2-d64-v64", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [7, 128], + "data": { "kind": "fillFloat32", "sinStep": 0.731, "cosStep": 0.567, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [2, 4], + "data": { "kind": "values", "values": [0, 1, 2, 4, 5, 6, 7, 3] } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [3], + "data": { "kind": "values", "values": [0, 3, 7] } + }, + "keyT": { + "dtype": "float32", + "shape": [7, 128], + "data": { "kind": "fillFloat32", "sinStep": 0.748, "cosStep": 0.588, "scale": 0.4 } + }, + "valueT": { + "dtype": "float32", + "shape": [7, 128], + "data": { "kind": "fillFloat32", "sinStep": 0.765, "cosStep": 0.609, "scale": 0.4 } + }, + "biasT": { + "dtype": "float32", + "shape": [384], + "data": { "kind": "fillFloat32", "sinStep": 0.782, "cosStep": 0.63, "scale": 0.2 } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 1, 4, 4], + "data": { "kind": "fillFloat32", "sinStep": 0.799, "cosStep": 0.651, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [7, 128] } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-general-float32-10-t17-h7-d32-v32", + "attrs": { "num_heads": 7 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [17, 7, 3, 32], + "data": { "kind": "fillFloat32", "sinStep": 1.054, "cosStep": 0.966, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 15], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_4_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 2, 17] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 7, 15, 15], + "data": { "kind": "fillFloat32", "sinStep": 1.122, "cosStep": 1.05, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [17, 224] } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-general-float32-11-t33-h3-d64-v64", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 3, 3, 64], + "data": { "kind": "fillFloat32", "sinStep": 1.377, "cosStep": 1.365, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 31], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_5_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 32, 33] } + }, + "biasT": { + "dtype": "float32", + "shape": [576], + "data": { "kind": "fillFloat32", "sinStep": 1.428, "cosStep": 1.428, "scale": 0.2 } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 31, 31], + "data": { "kind": "fillFloat32", "sinStep": 1.445, "cosStep": 1.449, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 192] } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-general-float32-12-t33-h2-d128-v128", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 256], + "data": { "kind": "fillFloat32", "sinStep": 0.187, "cosStep": 0.273, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [5, 31], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_6_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [6], + "data": { "kind": "values", "values": [0, 0, 2, 2, 33, 33] } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 256], + "data": { "kind": "fillFloat32", "sinStep": 0.204, "cosStep": 0.294, "scale": 0.4 } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 256], + "data": { "kind": "fillFloat32", "sinStep": 0.221, "cosStep": 0.315, "scale": 0.4 } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 2, 31, 31], + "data": { "kind": "fillFloat32", "sinStep": 0.255, "cosStep": 0.357, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 256] } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-general-float32-13-t65-h1-d32-v32", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [65, 1, 3, 32], + "data": { "kind": "fillFloat32", "sinStep": 0.51, "cosStep": 0.672, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [65, 1], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_7_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [66], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/packed_schedule_8_cumulativeSequenceLengthT" } + } + }, + "biasT": { + "dtype": "float32", + "shape": [96], + "data": { "kind": "fillFloat32", "sinStep": 0.561, "cosStep": 0.735, "scale": 0.2 } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 1, 1, 1], + "data": { "kind": "fillFloat32", "sinStep": 0.578, "cosStep": 0.756, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [65, 32] } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-general-float32-14-t160-h2-d64-v64", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [160, 128], + "data": { "kind": "fillFloat32", "sinStep": 0.833, "cosStep": 1.071, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [32, 129], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_9_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [33], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/packed_schedule_10_cumulativeSequenceLengthT" } + } + }, + "keyT": { + "dtype": "float32", + "shape": [160, 128], + "data": { "kind": "fillFloat32", "sinStep": 0.85, "cosStep": 1.092, "scale": 0.4 } + }, + "valueT": { + "dtype": "float32", + "shape": [160, 128], + "data": { "kind": "fillFloat32", "sinStep": 0.867, "cosStep": 1.113, "scale": 0.4 } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [32, 1, 129, 129], + "data": { "kind": "fillFloat32", "sinStep": 0.901, "cosStep": 1.155, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [160, 128] } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-general-float32-15-t15-h3-d32-v32", + "attrs": { "num_heads": 3, "scale": 0.125 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [15, 3, 3, 32], + "data": { "kind": "fillFloat32", "sinStep": 1.156, "cosStep": 1.47, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 129], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_11_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 3, 8, 15] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 3, 129, 129], + "data": { "kind": "fillFloat32", "sinStep": 1.224, "cosStep": 0.063, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [15, 96] } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-general-float32-16-t257-h2-d64-v64", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [257, 128], + "data": { "kind": "fillFloat32", "sinStep": 1.479, "cosStep": 0.378, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [1, 257], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_12_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [2], + "data": { "kind": "values", "values": [0, 257] } + }, + "keyT": { + "dtype": "float32", + "shape": [257, 128], + "data": { "kind": "fillFloat32", "sinStep": 1.496, "cosStep": 0.399, "scale": 0.4 } + }, + "valueT": { + "dtype": "float32", + "shape": [257, 128], + "data": { "kind": "fillFloat32", "sinStep": 1.513, "cosStep": 0.42, "scale": 0.4 } + }, + "biasT": { + "dtype": "float32", + "shape": [384], + "data": { "kind": "fillFloat32", "sinStep": 0.017, "cosStep": 0.441, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [257, 128] } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-general-float32-17-t258-h1-d128-v128", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [258, 1, 3, 128], + "data": { "kind": "fillFloat32", "sinStep": 0.289, "cosStep": 0.777, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [2, 257], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_13_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [3], + "data": { "kind": "values", "values": [0, 1, 258] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 1, 257, 257], + "data": { "kind": "fillFloat32", "sinStep": 0.357, "cosStep": 0.861, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [258, 128] } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-general-float32-18-t256-h3-d48-v48", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [256, 3, 3, 48], + "data": { "kind": "fillFloat32", "sinStep": 0.612, "cosStep": 1.176, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [2, 129], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_14_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [3], + "data": { "kind": "values", "values": [0, 127, 256] } + }, + "biasT": { + "dtype": "float32", + "shape": [432], + "data": { "kind": "fillFloat32", "sinStep": 0.663, "cosStep": 1.239, "scale": 0.2 } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 3, 129, 129], + "data": { "kind": "fillFloat32", "sinStep": 0.68, "cosStep": 1.26, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [256, 144] } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-general-float32-19-t16-h3-d8-v8", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [16, 24], + "data": { "kind": "fillFloat32", "sinStep": 0.935, "cosStep": 0.084, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [2, 9], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_15_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [3], + "data": { "kind": "values", "values": [0, 7, 16] } + }, + "keyT": { + "dtype": "float32", + "shape": [16, 24], + "data": { "kind": "fillFloat32", "sinStep": 0.952, "cosStep": 0.105, "scale": 0.4 } + }, + "valueT": { + "dtype": "float32", + "shape": [16, 24], + "data": { "kind": "fillFloat32", "sinStep": 0.969, "cosStep": 0.126, "scale": 0.4 } + }, + "biasT": { + "dtype": "float32", + "shape": [72], + "data": { "kind": "fillFloat32", "sinStep": 0.986, "cosStep": 0.147, "scale": 0.2 } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [2, 3, 9, 9], + "data": { "kind": "fillFloat32", "sinStep": 1.003, "cosStep": 0.168, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [16, 24] } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-general-float32-20-t32-h3-d24-v24", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [32, 3, 3, 24], + "data": { "kind": "fillFloat32", "sinStep": 1.258, "cosStep": 0.483, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [2, 17], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_16_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [3], + "data": { "kind": "values", "values": [0, 15, 32] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 1, 17, 17], + "data": { "kind": "fillFloat32", "sinStep": 1.326, "cosStep": 0.567, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [32, 72] } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-general-float32-21-t48-h2-d32-v20", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [48, 64], + "data": { "kind": "fillFloat32", "sinStep": 0.068, "cosStep": 0.882, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [2, 31], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_17_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [3], + "data": { "kind": "values", "values": [0, 17, 48] } + }, + "keyT": { + "dtype": "float32", + "shape": [48, 64], + "data": { "kind": "fillFloat32", "sinStep": 0.085, "cosStep": 0.903, "scale": 0.4 } + }, + "valueT": { + "dtype": "float32", + "shape": [48, 40], + "data": { "kind": "fillFloat32", "sinStep": 0.102, "cosStep": 0.924, "scale": 0.4 } + }, + "biasT": { + "dtype": "float32", + "shape": [168], + "data": { "kind": "fillFloat32", "sinStep": 0.119, "cosStep": 0.945, "scale": 0.2 } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [2, 1, 31, 31], + "data": { "kind": "fillFloat32", "sinStep": 0.136, "cosStep": 0.966, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [48, 40] } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-general-float32-22-t64-h2-d64-v32", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [64, 128], + "data": { "kind": "fillFloat32", "sinStep": 0.391, "cosStep": 1.281, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [2, 33], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_18_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [3], + "data": { "kind": "values", "values": [0, 31, 64] } + }, + "keyT": { + "dtype": "float32", + "shape": [64, 128], + "data": { "kind": "fillFloat32", "sinStep": 0.408, "cosStep": 1.302, "scale": 0.4 } + }, + "valueT": { + "dtype": "float32", + "shape": [64, 64], + "data": { "kind": "fillFloat32", "sinStep": 0.425, "cosStep": 1.323, "scale": 0.4 } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 2, 33, 33], + "data": { "kind": "fillFloat32", "sinStep": 0.459, "cosStep": 1.365, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [64, 64] } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-general-float16-0-t129-h3-d16-v16", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [129, 3, 3, 16], + "data": { "kind": "fillFloat32", "sinStep": 0.85, "cosStep": 1.449, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 65], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_3_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 63, 64, 129] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [129, 48] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-general-float16-1-t129-h3-d32-v32", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [129, 96], + "data": { "kind": "fillFloat32", "sinStep": 1.173, "cosStep": 0.357, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 65], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_3_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 63, 64, 129] } + }, + "keyT": { + "dtype": "float16", + "shape": [129, 96], + "data": { "kind": "fillFloat32", "sinStep": 1.19, "cosStep": 0.378, "scale": 0.4 } + }, + "valueT": { + "dtype": "float16", + "shape": [129, 96], + "data": { "kind": "fillFloat32", "sinStep": 1.207, "cosStep": 0.399, "scale": 0.4 } + }, + "biasT": { + "dtype": "float16", + "shape": [288], + "data": { "kind": "fillFloat32", "sinStep": 1.224, "cosStep": 0.42, "scale": 0.2 } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 1, 65, 65], + "data": { "kind": "fillFloat32", "sinStep": 1.241, "cosStep": 0.441, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [129, 96] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-general-float16-2-t129-h3-d48-v48", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [129, 3, 3, 48], + "data": { "kind": "fillFloat32", "sinStep": 1.496, "cosStep": 0.756, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 65], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_3_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 63, 64, 129] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 3, 65, 65], + "data": { "kind": "fillFloat32", "sinStep": 0.051, "cosStep": 0.84, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [129, 144] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-general-float16-3-t129-h3-d64-v64", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [129, 192], + "data": { "kind": "fillFloat32", "sinStep": 0.306, "cosStep": 1.155, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 65], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_3_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 63, 64, 129] } + }, + "keyT": { + "dtype": "float16", + "shape": [129, 192], + "data": { "kind": "fillFloat32", "sinStep": 0.323, "cosStep": 1.176, "scale": 0.4 } + }, + "valueT": { + "dtype": "float16", + "shape": [129, 192], + "data": { "kind": "fillFloat32", "sinStep": 0.34, "cosStep": 1.197, "scale": 0.4 } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 1, 65, 65], + "data": { "kind": "fillFloat32", "sinStep": 0.374, "cosStep": 1.239, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [129, 192] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-general-float16-4-t129-h3-d96-v96", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [129, 3, 3, 96], + "data": { "kind": "fillFloat32", "sinStep": 0.629, "cosStep": 0.063, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 65], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_3_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 63, 64, 129] } + }, + "biasT": { + "dtype": "float16", + "shape": [864], + "data": { "kind": "fillFloat32", "sinStep": 0.68, "cosStep": 0.126, "scale": 0.2 } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 3, 65, 65], + "data": { "kind": "fillFloat32", "sinStep": 0.697, "cosStep": 0.147, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [129, 288] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-general-float16-5-t129-h3-d128-v128", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [129, 384], + "data": { "kind": "fillFloat32", "sinStep": 0.952, "cosStep": 0.462, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 65], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_3_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 63, 64, 129] } + }, + "keyT": { + "dtype": "float16", + "shape": [129, 384], + "data": { "kind": "fillFloat32", "sinStep": 0.969, "cosStep": 0.483, "scale": 0.4 } + }, + "valueT": { + "dtype": "float16", + "shape": [129, 384], + "data": { "kind": "fillFloat32", "sinStep": 0.986, "cosStep": 0.504, "scale": 0.4 } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [129, 384] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-general-float16-6-t129-h3-d192-v192", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [129, 3, 3, 192], + "data": { "kind": "fillFloat32", "sinStep": 1.275, "cosStep": 0.861, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 65], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_3_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 63, 64, 129] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 1, 65, 65], + "data": { "kind": "fillFloat32", "sinStep": 1.343, "cosStep": 0.945, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [129, 576] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-general-float16-7-t129-h3-d256-v256", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [129, 768], + "data": { "kind": "fillFloat32", "sinStep": 0.085, "cosStep": 1.26, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 65], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_3_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 63, 64, 129] } + }, + "keyT": { + "dtype": "float16", + "shape": [129, 768], + "data": { "kind": "fillFloat32", "sinStep": 0.102, "cosStep": 1.281, "scale": 0.4 } + }, + "valueT": { + "dtype": "float16", + "shape": [129, 768], + "data": { "kind": "fillFloat32", "sinStep": 0.119, "cosStep": 1.302, "scale": 0.4 } + }, + "biasT": { + "dtype": "float16", + "shape": [2304], + "data": { "kind": "fillFloat32", "sinStep": 0.136, "cosStep": 1.323, "scale": 0.2 } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 3, 65, 65], + "data": { "kind": "fillFloat32", "sinStep": 0.153, "cosStep": 1.344, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [129, 768] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-general-float16-8-t1-h1-d32-v32", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [1, 1, 3, 32], + "data": { "kind": "fillFloat32", "sinStep": 0.408, "cosStep": 0.168, "scale": 0.4 } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [1, 32] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-general-float16-9-t7-h2-d64-v64", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [7, 128], + "data": { "kind": "fillFloat32", "sinStep": 0.731, "cosStep": 0.567, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [2, 4], + "data": { "kind": "values", "values": [0, 1, 2, 4, 5, 6, 7, 3] } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [3], + "data": { "kind": "values", "values": [0, 3, 7] } + }, + "keyT": { + "dtype": "float16", + "shape": [7, 128], + "data": { "kind": "fillFloat32", "sinStep": 0.748, "cosStep": 0.588, "scale": 0.4 } + }, + "valueT": { + "dtype": "float16", + "shape": [7, 128], + "data": { "kind": "fillFloat32", "sinStep": 0.765, "cosStep": 0.609, "scale": 0.4 } + }, + "biasT": { + "dtype": "float16", + "shape": [384], + "data": { "kind": "fillFloat32", "sinStep": 0.782, "cosStep": 0.63, "scale": 0.2 } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 1, 4, 4], + "data": { "kind": "fillFloat32", "sinStep": 0.799, "cosStep": 0.651, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [7, 128] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-general-float16-10-t17-h7-d32-v32", + "attrs": { "num_heads": 7 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [17, 7, 3, 32], + "data": { "kind": "fillFloat32", "sinStep": 1.054, "cosStep": 0.966, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 15], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_4_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 2, 17] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 7, 15, 15], + "data": { "kind": "fillFloat32", "sinStep": 1.122, "cosStep": 1.05, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [17, 224] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-general-float16-11-t33-h3-d64-v64", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 3, 3, 64], + "data": { "kind": "fillFloat32", "sinStep": 1.377, "cosStep": 1.365, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 31], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_5_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 32, 33] } + }, + "biasT": { + "dtype": "float16", + "shape": [576], + "data": { "kind": "fillFloat32", "sinStep": 1.428, "cosStep": 1.428, "scale": 0.2 } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 1, 31, 31], + "data": { "kind": "fillFloat32", "sinStep": 1.445, "cosStep": 1.449, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [33, 192] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-general-float16-12-t33-h2-d128-v128", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 256], + "data": { "kind": "fillFloat32", "sinStep": 0.187, "cosStep": 0.273, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [5, 31], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_6_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [6], + "data": { "kind": "values", "values": [0, 0, 2, 2, 33, 33] } + }, + "keyT": { + "dtype": "float16", + "shape": [33, 256], + "data": { "kind": "fillFloat32", "sinStep": 0.204, "cosStep": 0.294, "scale": 0.4 } + }, + "valueT": { + "dtype": "float16", + "shape": [33, 256], + "data": { "kind": "fillFloat32", "sinStep": 0.221, "cosStep": 0.315, "scale": 0.4 } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 2, 31, 31], + "data": { "kind": "fillFloat32", "sinStep": 0.255, "cosStep": 0.357, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [33, 256] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-general-float16-13-t65-h1-d32-v32", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [65, 1, 3, 32], + "data": { "kind": "fillFloat32", "sinStep": 0.51, "cosStep": 0.672, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [65, 1], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_7_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [66], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/packed_schedule_8_cumulativeSequenceLengthT" } + } + }, + "biasT": { + "dtype": "float16", + "shape": [96], + "data": { "kind": "fillFloat32", "sinStep": 0.561, "cosStep": 0.735, "scale": 0.2 } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 1, 1, 1], + "data": { "kind": "fillFloat32", "sinStep": 0.578, "cosStep": 0.756, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [65, 32] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-general-float16-14-t160-h2-d64-v64", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [160, 128], + "data": { "kind": "fillFloat32", "sinStep": 0.833, "cosStep": 1.071, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [32, 129], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_9_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [33], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/packed_schedule_10_cumulativeSequenceLengthT" } + } + }, + "keyT": { + "dtype": "float16", + "shape": [160, 128], + "data": { "kind": "fillFloat32", "sinStep": 0.85, "cosStep": 1.092, "scale": 0.4 } + }, + "valueT": { + "dtype": "float16", + "shape": [160, 128], + "data": { "kind": "fillFloat32", "sinStep": 0.867, "cosStep": 1.113, "scale": 0.4 } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [32, 1, 129, 129], + "data": { "kind": "fillFloat32", "sinStep": 0.901, "cosStep": 1.155, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [160, 128] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-general-float16-15-t15-h3-d32-v32", + "attrs": { "num_heads": 3, "scale": 0.125 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [15, 3, 3, 32], + "data": { "kind": "fillFloat32", "sinStep": 1.156, "cosStep": 1.47, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 129], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_11_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 3, 8, 15] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 3, 129, 129], + "data": { "kind": "fillFloat32", "sinStep": 1.224, "cosStep": 0.063, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [15, 96] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-general-float16-16-t257-h2-d64-v64", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [257, 128], + "data": { "kind": "fillFloat32", "sinStep": 1.479, "cosStep": 0.378, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [1, 257], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_12_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [2], + "data": { "kind": "values", "values": [0, 257] } + }, + "keyT": { + "dtype": "float16", + "shape": [257, 128], + "data": { "kind": "fillFloat32", "sinStep": 1.496, "cosStep": 0.399, "scale": 0.4 } + }, + "valueT": { + "dtype": "float16", + "shape": [257, 128], + "data": { "kind": "fillFloat32", "sinStep": 1.513, "cosStep": 0.42, "scale": 0.4 } + }, + "biasT": { + "dtype": "float16", + "shape": [384], + "data": { "kind": "fillFloat32", "sinStep": 0.017, "cosStep": 0.441, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [257, 128] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-general-float16-17-t258-h1-d128-v128", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [258, 1, 3, 128], + "data": { "kind": "fillFloat32", "sinStep": 0.289, "cosStep": 0.777, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [2, 257], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_13_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [3], + "data": { "kind": "values", "values": [0, 1, 258] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 1, 257, 257], + "data": { "kind": "fillFloat32", "sinStep": 0.357, "cosStep": 0.861, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [258, 128] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-general-float16-18-t256-h3-d48-v48", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [256, 3, 3, 48], + "data": { "kind": "fillFloat32", "sinStep": 0.612, "cosStep": 1.176, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [2, 129], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_14_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [3], + "data": { "kind": "values", "values": [0, 127, 256] } + }, + "biasT": { + "dtype": "float16", + "shape": [432], + "data": { "kind": "fillFloat32", "sinStep": 0.663, "cosStep": 1.239, "scale": 0.2 } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 3, 129, 129], + "data": { "kind": "fillFloat32", "sinStep": 0.68, "cosStep": 1.26, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [256, 144] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-general-float16-19-t16-h3-d8-v8", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [16, 24], + "data": { "kind": "fillFloat32", "sinStep": 0.935, "cosStep": 0.084, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [2, 9], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_15_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [3], + "data": { "kind": "values", "values": [0, 7, 16] } + }, + "keyT": { + "dtype": "float16", + "shape": [16, 24], + "data": { "kind": "fillFloat32", "sinStep": 0.952, "cosStep": 0.105, "scale": 0.4 } + }, + "valueT": { + "dtype": "float16", + "shape": [16, 24], + "data": { "kind": "fillFloat32", "sinStep": 0.969, "cosStep": 0.126, "scale": 0.4 } + }, + "biasT": { + "dtype": "float16", + "shape": [72], + "data": { "kind": "fillFloat32", "sinStep": 0.986, "cosStep": 0.147, "scale": 0.2 } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [2, 3, 9, 9], + "data": { "kind": "fillFloat32", "sinStep": 1.003, "cosStep": 0.168, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [16, 24] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-general-float16-20-t32-h3-d24-v24", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [32, 3, 3, 24], + "data": { "kind": "fillFloat32", "sinStep": 1.258, "cosStep": 0.483, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [2, 17], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_16_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [3], + "data": { "kind": "values", "values": [0, 15, 32] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 1, 17, 17], + "data": { "kind": "fillFloat32", "sinStep": 1.326, "cosStep": 0.567, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [32, 72] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-general-float16-21-t48-h2-d32-v20", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [48, 64], + "data": { "kind": "fillFloat32", "sinStep": 0.068, "cosStep": 0.882, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [2, 31], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_17_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [3], + "data": { "kind": "values", "values": [0, 17, 48] } + }, + "keyT": { + "dtype": "float16", + "shape": [48, 64], + "data": { "kind": "fillFloat32", "sinStep": 0.085, "cosStep": 0.903, "scale": 0.4 } + }, + "valueT": { + "dtype": "float16", + "shape": [48, 40], + "data": { "kind": "fillFloat32", "sinStep": 0.102, "cosStep": 0.924, "scale": 0.4 } + }, + "biasT": { + "dtype": "float16", + "shape": [168], + "data": { "kind": "fillFloat32", "sinStep": 0.119, "cosStep": 0.945, "scale": 0.2 } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [2, 1, 31, 31], + "data": { "kind": "fillFloat32", "sinStep": 0.136, "cosStep": 0.966, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [48, 40] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-general-float16-22-t64-h2-d64-v32", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [64, 128], + "data": { "kind": "fillFloat32", "sinStep": 0.391, "cosStep": 1.281, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [2, 33], + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/packed_schedule_18_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [3], + "data": { "kind": "values", "values": [0, 31, 64] } + }, + "keyT": { + "dtype": "float16", + "shape": [64, 128], + "data": { "kind": "fillFloat32", "sinStep": 0.408, "cosStep": 1.302, "scale": 0.4 } + }, + "valueT": { + "dtype": "float16", + "shape": [64, 64], + "data": { "kind": "fillFloat32", "sinStep": 0.425, "cosStep": 1.323, "scale": 0.4 } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 2, 33, 33], + "data": { "kind": "fillFloat32", "sinStep": 0.459, "cosStep": 1.365, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [64, 64] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-distributed-key-tile-1-float32", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [7, 128], + "data": { "kind": "fillFloat32", "sinStep": 0.731, "cosStep": 0.567, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [2, 4], + "data": { "kind": "values", "values": [0, 1, 2, 4, 5, 6, 7, 3] } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [3], + "data": { "kind": "values", "values": [0, 3, 7] } + }, + "keyT": { + "dtype": "float32", + "shape": [7, 128], + "data": { "kind": "fillFloat32", "sinStep": 0.748, "cosStep": 0.588, "scale": 0.4 } + }, + "valueT": { + "dtype": "float32", + "shape": [7, 128], + "data": { "kind": "fillFloat32", "sinStep": 0.765, "cosStep": 0.609, "scale": 0.4 } + }, + "biasT": { + "dtype": "float32", + "shape": [384], + "data": { "kind": "fillFloat32", "sinStep": 0.782, "cosStep": 0.63, "scale": 0.2 } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 1, 4, 4], + "data": { "kind": "fillFloat32", "sinStep": 0.799, "cosStep": 0.651, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [7, 128] } }, + "tolerance": 0.0002, + "relTolerance": 0.0002, + "tunables": { "CLUSTER_TILE_K": 1 } + }, + { + "name": "pmha-distributed-key-tile-2-float16", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [7, 128], + "data": { "kind": "fillFloat32", "sinStep": 0.731, "cosStep": 0.567, "scale": 0.4 } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [2, 4], + "data": { "kind": "values", "values": [0, 1, 2, 4, 5, 6, 7, 3] } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [3], + "data": { "kind": "values", "values": [0, 3, 7] } + }, + "keyT": { + "dtype": "float16", + "shape": [7, 128], + "data": { "kind": "fillFloat32", "sinStep": 0.748, "cosStep": 0.588, "scale": 0.4 } + }, + "valueT": { + "dtype": "float16", + "shape": [7, 128], + "data": { "kind": "fillFloat32", "sinStep": 0.765, "cosStep": 0.609, "scale": 0.4 } + }, + "biasT": { + "dtype": "float16", + "shape": [384], + "data": { "kind": "fillFloat32", "sinStep": 0.782, "cosStep": 0.63, "scale": 0.2 } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 1, 4, 4], + "data": { "kind": "fillFloat32", "sinStep": 0.799, "cosStep": 0.651, "scale": 0.2 } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [7, 128] } }, + "tolerance": 0.001, + "relTolerance": 0.005, + "tunables": { "CLUSTER_TILE_K": 2 } + }, + { + "name": "pmha-portable-float32-0-t17-k33-h3-d128-v128", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [17, 3, 3, 128], + "data": { "kind": "fillFloat32", "sinStep": 1.037, "cosStep": 0.567, "scale": 0.4 } + }, + "biasT": { + "dtype": "float32", + "shape": [1152], + "data": { "kind": "fillFloat32", "sinStep": 1.071, "cosStep": 0.609, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 16], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/portable_schedule_19_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 17] } + } + }, + "outputs": { "outputT": { "shape": [17, 384], "dtype": "float32" } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-portable-float32-1-t33-k64-h2-d192-v192", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 2, 3, 192], + "data": { "kind": "fillFloat32", "sinStep": 1.326, "cosStep": 0.924, "scale": 0.4 } + }, + "biasT": { + "dtype": "float32", + "shape": [1152], + "data": { "kind": "fillFloat32", "sinStep": 1.36, "cosStep": 0.966, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [3, 17], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/portable_schedule_20_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "shape": [4], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 16, 16, 33] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 2, 17, 17], + "data": { "kind": "fillFloat32", "sinStep": 1.377, "cosStep": 0.987, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [33, 384], "dtype": "float32" } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-portable-float32-2-t33-k32-h2-d224-v224", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 2, 3, 224], + "data": { "kind": "fillFloat32", "sinStep": 0.102, "cosStep": 1.281, "scale": 0.4 } + }, + "biasT": { + "dtype": "float32", + "shape": [1344], + "data": { "kind": "fillFloat32", "sinStep": 0.136, "cosStep": 1.323, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 17], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/portable_schedule_21_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 17, 33] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [2, 1, 17, 17], + "data": { "kind": "fillFloat32", "sinStep": 0.153, "cosStep": 1.344, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [33, 448], "dtype": "float32" } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-portable-float32-3-t65-k32-h2-d256-v256", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [65, 2, 3, 256], + "data": { "kind": "fillFloat32", "sinStep": 0.391, "cosStep": 0.147, "scale": 0.4 } + }, + "biasT": { + "dtype": "float32", + "shape": [1536], + "data": { "kind": "fillFloat32", "sinStep": 0.425, "cosStep": 0.189, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [3, 63], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/portable_schedule_22_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "shape": [4], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 2, 65] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 2, 63, 63], + "data": { "kind": "fillFloat32", "sinStep": 0.442, "cosStep": 0.21, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [65, 512], "dtype": "float32" } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-portable-float32-4-t33-k33-h1-d384-v384", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 1, 3, 384], + "data": { "kind": "fillFloat32", "sinStep": 0.68, "cosStep": 0.504, "scale": 0.4 } + }, + "biasT": { + "dtype": "float32", + "shape": [1152], + "data": { "kind": "fillFloat32", "sinStep": 0.714, "cosStep": 0.546, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 31], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/portable_schedule_23_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 31, 33] } + } + }, + "outputs": { "outputT": { "shape": [33, 384], "dtype": "float32" } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-portable-float32-5-t17-k64-h2-d416-v416", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [17, 2, 3, 416], + "data": { "kind": "fillFloat32", "sinStep": 0.969, "cosStep": 0.861, "scale": 0.4 } + }, + "biasT": { + "dtype": "float32", + "shape": [2496], + "data": { "kind": "fillFloat32", "sinStep": 1.003, "cosStep": 0.903, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 16], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/portable_schedule_19_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 17] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [2, 1, 16, 16], + "data": { "kind": "fillFloat32", "sinStep": 1.02, "cosStep": 0.924, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [17, 832], "dtype": "float32" } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-portable-float32-6-t33-k32-h1-d512-v512", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 1, 3, 512], + "data": { "kind": "fillFloat32", "sinStep": 1.258, "cosStep": 1.218, "scale": 0.4 } + }, + "biasT": { + "dtype": "float32", + "shape": [1536], + "data": { "kind": "fillFloat32", "sinStep": 1.292, "cosStep": 1.26, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 17], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/portable_schedule_24_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 16, 33] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 1, 17, 17], + "data": { "kind": "fillFloat32", "sinStep": 1.309, "cosStep": 1.281, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [33, 512], "dtype": "float32" } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-portable-float32-7-t17-k32-h1-d1024-v1024", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [17, 1, 3, 1024], + "data": { "kind": "fillFloat32", "sinStep": 0.034, "cosStep": 0.084, "scale": 0.4 } + }, + "biasT": { + "dtype": "float32", + "shape": [3072], + "data": { "kind": "fillFloat32", "sinStep": 0.068, "cosStep": 0.126, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 16], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/portable_schedule_19_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 17] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [2, 1, 16, 16], + "data": { "kind": "fillFloat32", "sinStep": 0.085, "cosStep": 0.147, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [17, 1024], "dtype": "float32" } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-portable-float32-8-t7-k32-h1-d2048-v2048", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [7, 1, 3, 2048], + "data": { "kind": "fillFloat32", "sinStep": 0.323, "cosStep": 0.441, "scale": 0.4 } + }, + "biasT": { + "dtype": "float32", + "shape": [6144], + "data": { "kind": "fillFloat32", "sinStep": 0.357, "cosStep": 0.483, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 4], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 2, 4, 5, 6, 7, 3] } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 3, 7] } + } + }, + "outputs": { "outputT": { "shape": [7, 2048], "dtype": "float32" } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-portable-float32-9-t33-k33-h3-d48-v48", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 3, 3, 48], + "data": { "kind": "fillFloat32", "sinStep": 0.612, "cosStep": 0.798, "scale": 0.4 } + }, + "biasT": { + "dtype": "float32", + "shape": [432], + "data": { "kind": "fillFloat32", "sinStep": 0.646, "cosStep": 0.84, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 17], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/portable_schedule_24_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 16, 33] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 3, 17, 17], + "data": { "kind": "fillFloat32", "sinStep": 0.663, "cosStep": 0.861, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [33, 144], "dtype": "float32" } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-portable-float32-10-t17-k32-h1-d272-v272", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [17, 1, 3, 272], + "data": { "kind": "fillFloat32", "sinStep": 0.901, "cosStep": 1.155, "scale": 0.4 } + }, + "biasT": { + "dtype": "float32", + "shape": [816], + "data": { "kind": "fillFloat32", "sinStep": 0.935, "cosStep": 1.197, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 9], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/portable_schedule_25_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 8, 17] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [2, 1, 9, 9], + "data": { "kind": "fillFloat32", "sinStep": 0.952, "cosStep": 1.218, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [17, 272], "dtype": "float32" } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-portable-float32-11-t33-k64-h2-d64-v48", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 128], + "data": { "kind": "fillFloat32", "sinStep": 1.19, "cosStep": 1.512, "scale": 0.4 } + }, + "biasT": { + "dtype": "float32", + "shape": [352], + "data": { "kind": "fillFloat32", "sinStep": 1.224, "cosStep": 0.063, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 31], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/portable_schedule_23_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 31, 33] } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 128], + "data": { "kind": "fillFloat32", "sinStep": 1.207, "cosStep": 1.533, "scale": 0.4 } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 96], + "data": { "kind": "fillFloat32", "sinStep": 1.224, "cosStep": 0.063, "scale": 0.4 } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [2, 2, 31, 31], + "data": { "kind": "fillFloat32", "sinStep": 1.241, "cosStep": 0.084, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [33, 96], "dtype": "float32" } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-portable-float32-12-t1-k33-h1-d256-v256", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [1, 1, 3, 256], + "data": { "kind": "fillFloat32", "sinStep": 1.479, "cosStep": 0.378, "scale": 0.4 } + }, + "biasT": { + "dtype": "float32", + "shape": [768], + "data": { "kind": "fillFloat32", "sinStep": 1.513, "cosStep": 0.42, "scale": 0.1 } + }, + "tokenOffsetT": { "shape": [1, 1], "dtype": "int32", "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "shape": [2], "dtype": "int32", "data": { "kind": "values", "values": [0, 1] } } + }, + "outputs": { "outputT": { "shape": [1, 256], "dtype": "float32" } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-portable-float32-13-t129-k64-h2-d128-v128", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [129, 2, 3, 128], + "data": { "kind": "fillFloat32", "sinStep": 0.255, "cosStep": 0.735, "scale": 0.4 } + }, + "biasT": { + "dtype": "float32", + "shape": [768], + "data": { "kind": "fillFloat32", "sinStep": 0.289, "cosStep": 0.777, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [3, 127], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/portable_schedule_26_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "shape": [4], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 2, 129] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 127, 127], + "data": { "kind": "fillFloat32", "sinStep": 0.306, "cosStep": 0.798, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [129, 256], "dtype": "float32" } }, + "tolerance": 0.0002, + "relTolerance": 0.0002 + }, + { + "name": "pmha-portable-float16-0-t17-k33-h3-d128-v128", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [17, 3, 3, 128], + "data": { "kind": "fillFloat32", "sinStep": 1.037, "cosStep": 0.567, "scale": 0.4 } + }, + "biasT": { + "dtype": "float16", + "shape": [1152], + "data": { "kind": "fillFloat32", "sinStep": 1.071, "cosStep": 0.609, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 16], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/portable_schedule_19_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 17] } + } + }, + "outputs": { "outputT": { "shape": [17, 384], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-portable-float16-1-t33-k64-h2-d192-v192", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 2, 3, 192], + "data": { "kind": "fillFloat32", "sinStep": 1.326, "cosStep": 0.924, "scale": 0.4 } + }, + "biasT": { + "dtype": "float16", + "shape": [1152], + "data": { "kind": "fillFloat32", "sinStep": 1.36, "cosStep": 0.966, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [3, 17], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/portable_schedule_20_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "shape": [4], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 16, 16, 33] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 2, 17, 17], + "data": { "kind": "fillFloat32", "sinStep": 1.377, "cosStep": 0.987, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [33, 384], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-portable-float16-2-t33-k32-h2-d224-v224", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 2, 3, 224], + "data": { "kind": "fillFloat32", "sinStep": 0.102, "cosStep": 1.281, "scale": 0.4 } + }, + "biasT": { + "dtype": "float16", + "shape": [1344], + "data": { "kind": "fillFloat32", "sinStep": 0.136, "cosStep": 1.323, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 17], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/portable_schedule_21_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 17, 33] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [2, 1, 17, 17], + "data": { "kind": "fillFloat32", "sinStep": 0.153, "cosStep": 1.344, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [33, 448], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-portable-float16-3-t65-k32-h2-d256-v256", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [65, 2, 3, 256], + "data": { "kind": "fillFloat32", "sinStep": 0.391, "cosStep": 0.147, "scale": 0.4 } + }, + "biasT": { + "dtype": "float16", + "shape": [1536], + "data": { "kind": "fillFloat32", "sinStep": 0.425, "cosStep": 0.189, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [3, 63], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/portable_schedule_22_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "shape": [4], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 2, 65] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 2, 63, 63], + "data": { "kind": "fillFloat32", "sinStep": 0.442, "cosStep": 0.21, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [65, 512], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-portable-float16-4-t33-k33-h1-d384-v384", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 1, 3, 384], + "data": { "kind": "fillFloat32", "sinStep": 0.68, "cosStep": 0.504, "scale": 0.4 } + }, + "biasT": { + "dtype": "float16", + "shape": [1152], + "data": { "kind": "fillFloat32", "sinStep": 0.714, "cosStep": 0.546, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 31], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/portable_schedule_23_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 31, 33] } + } + }, + "outputs": { "outputT": { "shape": [33, 384], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-portable-float16-5-t17-k64-h2-d416-v416", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [17, 2, 3, 416], + "data": { "kind": "fillFloat32", "sinStep": 0.969, "cosStep": 0.861, "scale": 0.4 } + }, + "biasT": { + "dtype": "float16", + "shape": [2496], + "data": { "kind": "fillFloat32", "sinStep": 1.003, "cosStep": 0.903, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 16], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/portable_schedule_19_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 17] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [2, 1, 16, 16], + "data": { "kind": "fillFloat32", "sinStep": 1.02, "cosStep": 0.924, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [17, 832], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-portable-float16-6-t33-k32-h1-d512-v512", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 1, 3, 512], + "data": { "kind": "fillFloat32", "sinStep": 1.258, "cosStep": 1.218, "scale": 0.4 } + }, + "biasT": { + "dtype": "float16", + "shape": [1536], + "data": { "kind": "fillFloat32", "sinStep": 1.292, "cosStep": 1.26, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 17], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/portable_schedule_24_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 16, 33] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 1, 17, 17], + "data": { "kind": "fillFloat32", "sinStep": 1.309, "cosStep": 1.281, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [33, 512], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-portable-float16-7-t17-k32-h1-d1024-v1024", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [17, 1, 3, 1024], + "data": { "kind": "fillFloat32", "sinStep": 0.034, "cosStep": 0.084, "scale": 0.4 } + }, + "biasT": { + "dtype": "float16", + "shape": [3072], + "data": { "kind": "fillFloat32", "sinStep": 0.068, "cosStep": 0.126, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 16], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/portable_schedule_19_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 17] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [2, 1, 16, 16], + "data": { "kind": "fillFloat32", "sinStep": 0.085, "cosStep": 0.147, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [17, 1024], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-portable-float16-8-t7-k32-h1-d2048-v2048", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [7, 1, 3, 2048], + "data": { "kind": "fillFloat32", "sinStep": 0.323, "cosStep": 0.441, "scale": 0.4 } + }, + "biasT": { + "dtype": "float16", + "shape": [6144], + "data": { "kind": "fillFloat32", "sinStep": 0.357, "cosStep": 0.483, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 4], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 2, 4, 5, 6, 7, 3] } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 3, 7] } + } + }, + "outputs": { "outputT": { "shape": [7, 2048], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-portable-float16-9-t33-k33-h3-d48-v48", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 3, 3, 48], + "data": { "kind": "fillFloat32", "sinStep": 0.612, "cosStep": 0.798, "scale": 0.4 } + }, + "biasT": { + "dtype": "float16", + "shape": [432], + "data": { "kind": "fillFloat32", "sinStep": 0.646, "cosStep": 0.84, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 17], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/portable_schedule_24_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 16, 33] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 3, 17, 17], + "data": { "kind": "fillFloat32", "sinStep": 0.663, "cosStep": 0.861, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [33, 144], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-portable-float16-10-t17-k32-h1-d272-v272", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [17, 1, 3, 272], + "data": { "kind": "fillFloat32", "sinStep": 0.901, "cosStep": 1.155, "scale": 0.4 } + }, + "biasT": { + "dtype": "float16", + "shape": [816], + "data": { "kind": "fillFloat32", "sinStep": 0.935, "cosStep": 1.197, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 9], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/portable_schedule_25_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 8, 17] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [2, 1, 9, 9], + "data": { "kind": "fillFloat32", "sinStep": 0.952, "cosStep": 1.218, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [17, 272], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-portable-float16-11-t33-k64-h2-d64-v48", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 128], + "data": { "kind": "fillFloat32", "sinStep": 1.19, "cosStep": 1.512, "scale": 0.4 } + }, + "biasT": { + "dtype": "float16", + "shape": [352], + "data": { "kind": "fillFloat32", "sinStep": 1.224, "cosStep": 0.063, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 31], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/portable_schedule_23_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 31, 33] } + }, + "keyT": { + "dtype": "float16", + "shape": [33, 128], + "data": { "kind": "fillFloat32", "sinStep": 1.207, "cosStep": 1.533, "scale": 0.4 } + }, + "valueT": { + "dtype": "float16", + "shape": [33, 96], + "data": { "kind": "fillFloat32", "sinStep": 1.224, "cosStep": 0.063, "scale": 0.4 } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [2, 2, 31, 31], + "data": { "kind": "fillFloat32", "sinStep": 1.241, "cosStep": 0.084, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [33, 96], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-portable-float16-12-t1-k33-h1-d256-v256", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [1, 1, 3, 256], + "data": { "kind": "fillFloat32", "sinStep": 1.479, "cosStep": 0.378, "scale": 0.4 } + }, + "biasT": { + "dtype": "float16", + "shape": [768], + "data": { "kind": "fillFloat32", "sinStep": 1.513, "cosStep": 0.42, "scale": 0.1 } + }, + "tokenOffsetT": { "shape": [1, 1], "dtype": "int32", "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "shape": [2], "dtype": "int32", "data": { "kind": "values", "values": [0, 1] } } + }, + "outputs": { "outputT": { "shape": [1, 256], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-portable-float16-13-t129-k64-h2-d128-v128", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [129, 2, 3, 128], + "data": { "kind": "fillFloat32", "sinStep": 0.255, "cosStep": 0.735, "scale": 0.4 } + }, + "biasT": { + "dtype": "float16", + "shape": [768], + "data": { "kind": "fillFloat32", "sinStep": 0.289, "cosStep": 0.777, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [3, 127], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/portable_schedule_26_tokenOffsetT" } } + }, + "cumulativeSequenceLengthT": { + "shape": [4], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 2, 129] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 1, 127, 127], + "data": { "kind": "fillFloat32", "sinStep": 0.306, "cosStep": 0.798, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [129, 256], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float32_d16v12_b3t9h2_bias0attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [9, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [9, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [9, 24], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 5], + "data": { "kind": "values", "values": [0, 5, 6, 7, 10, 11, 12, 13, 14, 1, 2, 3, 4, 8, 9] } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 4, 9] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [9, 24] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d16v12_b3t9h2_bias1attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [9, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [9, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [9, 24], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 5], + "data": { "kind": "values", "values": [0, 5, 6, 7, 10, 11, 12, 13, 14, 1, 2, 3, 4, 8, 9] } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 4, 9] } + }, + "biasT": { + "dtype": "float32", + "shape": [88], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [9, 24] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d16v12_b3t9h2_bias0attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [9, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [9, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [9, 24], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 5], + "data": { "kind": "values", "values": [0, 5, 6, 7, 10, 11, 12, 13, 14, 1, 2, 3, 4, 8, 9] } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 4, 9] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 5, 5], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [9, 24] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d16v12_b3t9h2_bias1attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [9, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [9, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [9, 24], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 5], + "data": { "kind": "values", "values": [0, 5, 6, 7, 10, 11, 12, 13, 14, 1, 2, 3, 4, 8, 9] } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 4, 9] } + }, + "biasT": { + "dtype": "float32", + "shape": [88], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 5, 5], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [9, 24] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float16_d16v12_b3t9h2_bias0attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [9, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [9, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [9, 24], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 5], + "data": { "kind": "values", "values": [0, 5, 6, 7, 10, 11, 12, 13, 14, 1, 2, 3, 4, 8, 9] } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 4, 9] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [9, 24] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d16v12_b3t9h2_bias1attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [9, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [9, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [9, 24], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 5], + "data": { "kind": "values", "values": [0, 5, 6, 7, 10, 11, 12, 13, 14, 1, 2, 3, 4, 8, 9] } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 4, 9] } + }, + "biasT": { + "dtype": "float16", + "shape": [88], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [9, 24] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d16v12_b3t9h2_bias0attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [9, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [9, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [9, 24], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 5], + "data": { "kind": "values", "values": [0, 5, 6, 7, 10, 11, 12, 13, 14, 1, 2, 3, 4, 8, 9] } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 4, 9] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 1, 5, 5], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [9, 24] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d16v12_b3t9h2_bias1attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [9, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [9, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [9, 24], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 5], + "data": { "kind": "values", "values": [0, 5, 6, 7, 10, 11, 12, 13, 14, 1, 2, 3, 4, 8, 9] } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 4, 9] } + }, + "biasT": { + "dtype": "float16", + "shape": [88], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 1, 5, 5], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [9, 24] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float32_d16v20_b3t33h3_bias0attn0", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 48], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 48], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 60], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 60] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d16v20_b3t33h3_bias1attn0", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 48], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 48], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 60], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float32", + "shape": [156], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 60] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d16v20_b3t33h3_bias0attn1", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 48], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 48], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 60], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 60] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d16v20_b3t33h3_bias1attn1", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 48], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 48], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 60], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float32", + "shape": [156], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 60] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float16_d16v20_b3t33h3_bias0attn0", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 48], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [33, 48], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [33, 60], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [33, 60] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d16v20_b3t33h3_bias1attn0", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 48], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [33, 48], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [33, 60], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float16", + "shape": [156], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [33, 60] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d16v20_b3t33h3_bias0attn1", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 48], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [33, 48], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [33, 60], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [33, 60] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d16v20_b3t33h3_bias1attn1", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 48], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [33, 48], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [33, 60], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float16", + "shape": [156], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [33, 60] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float32_d32v4_b3t34h2_bias0attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [34, 8], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v4_b3t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 17, 34] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [34, 8] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d32v4_b3t34h2_bias1attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [34, 8], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v4_b3t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 17, 34] } + }, + "biasT": { + "dtype": "float32", + "shape": [136], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [34, 8] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d32v4_b3t34h2_bias0attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [34, 8], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v4_b3t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 17, 34] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [34, 8] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d32v4_b3t34h2_bias1attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [34, 8], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v4_b3t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 17, 34] } + }, + "biasT": { + "dtype": "float32", + "shape": [136], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [34, 8] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float16_d32v4_b3t34h2_bias0attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [34, 8], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v4_b3t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 17, 34] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [34, 8] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d32v4_b3t34h2_bias1attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [34, 8], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v4_b3t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 17, 34] } + }, + "biasT": { + "dtype": "float16", + "shape": [136], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [34, 8] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d32v4_b3t34h2_bias0attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [34, 8], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v4_b3t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 17, 34] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [34, 8] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d32v4_b3t34h2_bias1attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [34, 8], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v4_b3t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 17, 34] } + }, + "biasT": { + "dtype": "float16", + "shape": [136], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [34, 8] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float32_d32v20_b4t34h2_bias0attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [34, 40], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [4, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v20_b4t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [5], + "data": { "kind": "values", "values": [0, 0, 1, 17, 34] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [34, 40] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d32v20_b4t34h2_bias1attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [34, 40], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [4, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v20_b4t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [5], + "data": { "kind": "values", "values": [0, 0, 1, 17, 34] } + }, + "biasT": { + "dtype": "float32", + "shape": [168], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [34, 40] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d32v20_b4t34h2_bias0attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [34, 40], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [4, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v20_b4t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [5], + "data": { "kind": "values", "values": [0, 0, 1, 17, 34] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [4, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [34, 40] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d32v20_b4t34h2_bias1attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [34, 40], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [4, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v20_b4t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [5], + "data": { "kind": "values", "values": [0, 0, 1, 17, 34] } + }, + "biasT": { + "dtype": "float32", + "shape": [168], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [4, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [34, 40] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float16_d32v20_b4t34h2_bias0attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [34, 40], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [4, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v20_b4t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [5], + "data": { "kind": "values", "values": [0, 0, 1, 17, 34] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [34, 40] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d32v20_b4t34h2_bias1attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [34, 40], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [4, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v20_b4t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [5], + "data": { "kind": "values", "values": [0, 0, 1, 17, 34] } + }, + "biasT": { + "dtype": "float16", + "shape": [168], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [34, 40] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d32v20_b4t34h2_bias0attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [34, 40], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [4, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v20_b4t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [5], + "data": { "kind": "values", "values": [0, 0, 1, 17, 34] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [4, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [34, 40] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d32v20_b4t34h2_bias1attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [34, 40], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [4, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v20_b4t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [5], + "data": { "kind": "values", "values": [0, 0, 1, 17, 34] } + }, + "biasT": { + "dtype": "float16", + "shape": [168], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [4, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [34, 40] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float32_d32v28_b3t48h3_bias0attn0", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [48, 96], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [48, 96], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [48, 84], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v28_b3t48h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 15, 31, 48] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [48, 84] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d32v28_b3t48h3_bias1attn0", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [48, 96], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [48, 96], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [48, 84], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v28_b3t48h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 15, 31, 48] } + }, + "biasT": { + "dtype": "float32", + "shape": [276], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [48, 84] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d32v28_b3t48h3_bias0attn1", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [48, 96], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [48, 96], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [48, 84], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v28_b3t48h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 15, 31, 48] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [48, 84] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d32v28_b3t48h3_bias1attn1", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [48, 96], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [48, 96], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [48, 84], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v28_b3t48h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 15, 31, 48] } + }, + "biasT": { + "dtype": "float32", + "shape": [276], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [48, 84] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float16_d32v28_b3t48h3_bias0attn0", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [48, 96], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [48, 96], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [48, 84], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v28_b3t48h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 15, 31, 48] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [48, 84] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d32v28_b3t48h3_bias1attn0", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [48, 96], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [48, 96], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [48, 84], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v28_b3t48h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 15, 31, 48] } + }, + "biasT": { + "dtype": "float16", + "shape": [276], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [48, 84] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d32v28_b3t48h3_bias0attn1", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [48, 96], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [48, 96], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [48, 84], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v28_b3t48h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 15, 31, 48] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [48, 84] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d32v28_b3t48h3_bias1attn1", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [48, 96], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [48, 96], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [48, 84], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v28_b3t48h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 15, 31, 48] } + }, + "biasT": { + "dtype": "float16", + "shape": [276], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [48, 84] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float32_d32v36_b3t33h2_bias0attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 72], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 72] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d32v36_b3t33h2_bias1attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 72], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float32", + "shape": [200], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 72] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d32v36_b3t33h2_bias0attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 72], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 72] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d32v36_b3t33h2_bias1attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 72], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float32", + "shape": [200], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 72] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float16_d32v36_b3t33h2_bias0attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [33, 72], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [33, 72] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d32v36_b3t33h2_bias1attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [33, 72], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float16", + "shape": [200], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [33, 72] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d32v36_b3t33h2_bias0attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [33, 72], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [33, 72] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d32v36_b3t33h2_bias1attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [33, 72], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float16", + "shape": [200], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [33, 72] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float32_d48v44_b3t34h2_bias0attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [34, 96], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [34, 96], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [34, 88], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v4_b3t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 17, 34] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [34, 88] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d48v44_b3t34h2_bias1attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [34, 96], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [34, 96], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [34, 88], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v4_b3t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 17, 34] } + }, + "biasT": { + "dtype": "float32", + "shape": [280], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [34, 88] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d48v44_b3t34h2_bias0attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [34, 96], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [34, 96], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [34, 88], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v4_b3t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 17, 34] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [34, 88] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d48v44_b3t34h2_bias1attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [34, 96], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [34, 96], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [34, 88], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v4_b3t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 17, 34] } + }, + "biasT": { + "dtype": "float32", + "shape": [280], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [34, 88] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float16_d48v44_b3t34h2_bias0attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [34, 96], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [34, 96], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [34, 88], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v4_b3t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 17, 34] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [34, 88] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d48v44_b3t34h2_bias1attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [34, 96], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [34, 96], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [34, 88], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v4_b3t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 17, 34] } + }, + "biasT": { + "dtype": "float16", + "shape": [280], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [34, 88] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d48v44_b3t34h2_bias0attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [34, 96], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [34, 96], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [34, 88], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v4_b3t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 17, 34] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [34, 88] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d48v44_b3t34h2_bias1attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [34, 96], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [34, 96], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [34, 88], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v4_b3t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 17, 34] } + }, + "biasT": { + "dtype": "float16", + "shape": [280], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [34, 88] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float32_d64v48_b3t65h2_bias0attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [65, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [65, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [65, 96], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 33], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d64v48_b3t65h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 32, 65] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [65, 96] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d64v48_b3t65h2_bias1attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [65, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [65, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [65, 96], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 33], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d64v48_b3t65h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 32, 65] } + }, + "biasT": { + "dtype": "float32", + "shape": [352], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [65, 96] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d64v48_b3t65h2_bias0attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [65, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [65, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [65, 96], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 33], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d64v48_b3t65h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 32, 65] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 33, 33], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [65, 96] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d64v48_b3t65h2_bias1attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [65, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [65, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [65, 96], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 33], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d64v48_b3t65h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 32, 65] } + }, + "biasT": { + "dtype": "float32", + "shape": [352], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 33, 33], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [65, 96] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float16_d64v48_b3t65h2_bias0attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [65, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [65, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [65, 96], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 33], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d64v48_b3t65h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 32, 65] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [65, 96] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d64v48_b3t65h2_bias1attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [65, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [65, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [65, 96], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 33], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d64v48_b3t65h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 32, 65] } + }, + "biasT": { + "dtype": "float16", + "shape": [352], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [65, 96] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d64v48_b3t65h2_bias0attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [65, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [65, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [65, 96], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 33], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d64v48_b3t65h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 32, 65] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 1, 33, 33], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [65, 96] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d64v48_b3t65h2_bias1attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [65, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [65, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [65, 96], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 33], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d64v48_b3t65h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 32, 65] } + }, + "biasT": { + "dtype": "float16", + "shape": [352], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 1, 33, 33], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [65, 96] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float32_d64v68_b4t34h2_bias0attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [34, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [34, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [34, 136], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [4, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v20_b4t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [5], + "data": { "kind": "values", "values": [0, 0, 1, 17, 34] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [34, 136] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d64v68_b4t34h2_bias1attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [34, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [34, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [34, 136], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [4, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v20_b4t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [5], + "data": { "kind": "values", "values": [0, 0, 1, 17, 34] } + }, + "biasT": { + "dtype": "float32", + "shape": [392], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [34, 136] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d64v68_b4t34h2_bias0attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [34, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [34, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [34, 136], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [4, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v20_b4t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [5], + "data": { "kind": "values", "values": [0, 0, 1, 17, 34] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [4, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [34, 136] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d64v68_b4t34h2_bias1attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [34, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [34, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [34, 136], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [4, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v20_b4t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [5], + "data": { "kind": "values", "values": [0, 0, 1, 17, 34] } + }, + "biasT": { + "dtype": "float32", + "shape": [392], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [4, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [34, 136] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float16_d64v68_b4t34h2_bias0attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [34, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [34, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [34, 136], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [4, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v20_b4t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [5], + "data": { "kind": "values", "values": [0, 0, 1, 17, 34] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [34, 136] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d64v68_b4t34h2_bias1attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [34, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [34, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [34, 136], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [4, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v20_b4t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [5], + "data": { "kind": "values", "values": [0, 0, 1, 17, 34] } + }, + "biasT": { + "dtype": "float16", + "shape": [392], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [34, 136] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d64v68_b4t34h2_bias0attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [34, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [34, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [34, 136], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [4, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v20_b4t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [5], + "data": { "kind": "values", "values": [0, 0, 1, 17, 34] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [4, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [34, 136] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d64v68_b4t34h2_bias1attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [34, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [34, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [34, 136], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [4, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v20_b4t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [5], + "data": { "kind": "values", "values": [0, 0, 1, 17, 34] } + }, + "biasT": { + "dtype": "float16", + "shape": [392], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [4, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [34, 136] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float32_d64v128_b3t33h2_bias0attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 256] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d64v128_b3t33h2_bias1attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float32", + "shape": [512], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 256] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d64v128_b3t33h2_bias0attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 256] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d64v128_b3t33h2_bias1attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float32", + "shape": [512], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 256] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float16_d64v128_b3t33h2_bias0attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [33, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [33, 256] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d64v128_b3t33h2_bias1attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [33, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float16", + "shape": [512], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [33, 256] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d64v128_b3t33h2_bias0attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [33, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [33, 256] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d64v128_b3t33h2_bias1attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [33, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float16", + "shape": [512], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [33, 256] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float32_d128v32_b3t33h2_bias0attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 64] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d128v32_b3t33h2_bias1attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float32", + "shape": [576], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 64] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d128v32_b3t33h2_bias0attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 64] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d128v32_b3t33h2_bias1attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float32", + "shape": [576], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 64] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float16_d128v32_b3t33h2_bias0attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [33, 64] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d128v32_b3t33h2_bias1attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float16", + "shape": [576], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [33, 64] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d128v32_b3t33h2_bias0attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [33, 64] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d128v32_b3t33h2_bias1attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float16", + "shape": [576], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [33, 64] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float32_d128v132_b3t48h2_bias0attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [48, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [48, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [48, 264], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v28_b3t48h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 15, 31, 48] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [48, 264] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d128v132_b3t48h2_bias1attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [48, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [48, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [48, 264], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v28_b3t48h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 15, 31, 48] } + }, + "biasT": { + "dtype": "float32", + "shape": [776], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [48, 264] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d128v132_b3t48h2_bias0attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [48, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [48, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [48, 264], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v28_b3t48h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 15, 31, 48] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [48, 264] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d128v132_b3t48h2_bias1attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [48, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [48, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [48, 264], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v28_b3t48h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 15, 31, 48] } + }, + "biasT": { + "dtype": "float32", + "shape": [776], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [48, 264] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float16_d128v132_b3t48h2_bias0attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [48, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [48, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [48, 264], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v28_b3t48h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 15, 31, 48] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [48, 264] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d128v132_b3t48h2_bias1attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [48, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [48, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [48, 264], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v28_b3t48h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 15, 31, 48] } + }, + "biasT": { + "dtype": "float16", + "shape": [776], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [48, 264] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d128v132_b3t48h2_bias0attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [48, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [48, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [48, 264], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v28_b3t48h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 15, 31, 48] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [48, 264] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d128v132_b3t48h2_bias1attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [48, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [48, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [48, 264], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v28_b3t48h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 15, 31, 48] } + }, + "biasT": { + "dtype": "float16", + "shape": [776], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [48, 264] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float32_d256v512_b3t33h1_bias0attn0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 512] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d256v512_b3t33h1_bias1attn0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float32", + "shape": [1024], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 512] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d256v512_b3t33h1_bias0attn1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 512] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d256v512_b3t33h1_bias1attn1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float32", + "shape": [1024], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 512] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float16_d256v512_b3t33h1_bias0attn0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [33, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [33, 512] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d256v512_b3t33h1_bias1attn0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [33, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float16", + "shape": [1024], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [33, 512] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d256v512_b3t33h1_bias0attn1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [33, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [33, 512] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d256v512_b3t33h1_bias1attn1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [33, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float16", + "shape": [1024], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [33, 512] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float32_d64v21_b3t33h2_bias0attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 42], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 42] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d64v21_b3t33h2_bias1attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 42], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float32", + "shape": [298], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 42] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d64v21_b3t33h2_bias0attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 42], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 42] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d64v21_b3t33h2_bias1attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 42], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float32", + "shape": [298], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 42] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float16_d64v21_b3t33h2_bias0attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [33, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [33, 42], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [33, 42] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d64v21_b3t33h2_bias1attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [33, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [33, 42], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float16", + "shape": [298], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [33, 42] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d64v21_b3t33h2_bias0attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [33, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [33, 42], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [33, 42] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d64v21_b3t33h2_bias1attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [33, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [33, 42], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float16", + "shape": [298], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [33, 42] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float32_d32v20_b1t1h1_bias0attn0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [1, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [1, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [1, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [1, 20] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d32v20_b1t1h1_bias1attn0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [1, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [1, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [1, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } }, + "biasT": { + "dtype": "float32", + "shape": [84], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [1, 20] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d32v20_b1t1h1_bias0attn1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [1, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [1, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [1, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 1, 1, 1], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [1, 20] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d32v20_b1t1h1_bias1attn1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [1, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [1, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [1, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } }, + "biasT": { + "dtype": "float32", + "shape": [84], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 1, 1, 1], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [1, 20] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float16_d32v20_b1t1h1_bias0attn0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [1, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [1, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [1, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [1, 20] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d32v20_b1t1h1_bias1attn0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [1, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [1, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [1, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } }, + "biasT": { + "dtype": "float16", + "shape": [84], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [1, 20] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d32v20_b1t1h1_bias0attn1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [1, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [1, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [1, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 1, 1, 1], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [1, 20] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d32v20_b1t1h1_bias1attn1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [1, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [1, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [1, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } }, + "biasT": { + "dtype": "float16", + "shape": [84], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 1, 1, 1], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [1, 20] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float32_d64v128_b1t1h2_bias0attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [1, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [1, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [1, 256], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [1, 256] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d64v128_b1t1h2_bias1attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [1, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [1, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [1, 256], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } }, + "biasT": { + "dtype": "float32", + "shape": [512], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [1, 256] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d64v128_b1t1h2_bias0attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [1, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [1, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [1, 256], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 1, 1, 1], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [1, 256] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d64v128_b1t1h2_bias1attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [1, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [1, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [1, 256], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } }, + "biasT": { + "dtype": "float32", + "shape": [512], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 1, 1, 1], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [1, 256] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float16_d64v128_b1t1h2_bias0attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [1, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [1, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [1, 256], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [1, 256] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d64v128_b1t1h2_bias1attn0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [1, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [1, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [1, 256], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } }, + "biasT": { + "dtype": "float16", + "shape": [512], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [1, 256] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d64v128_b1t1h2_bias0attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [1, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [1, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [1, 256], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 1, 1, 1], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [1, 256] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d64v128_b1t1h2_bias1attn1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [1, 128], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [1, 128], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [1, 256], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } }, + "biasT": { + "dtype": "float16", + "shape": [512], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 1, 1, 1], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [1, 256] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float32_d256v512_b1t1h1_bias0attn0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [1, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [1, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [1, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [1, 512] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d256v512_b1t1h1_bias1attn0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [1, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [1, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [1, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } }, + "biasT": { + "dtype": "float32", + "shape": [1024], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [1, 512] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d256v512_b1t1h1_bias0attn1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [1, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [1, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [1, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 1, 1, 1], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [1, 512] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float32_d256v512_b1t1h1_bias1attn1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [1, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [1, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [1, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } }, + "biasT": { + "dtype": "float32", + "shape": [1024], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 1, 1, 1], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [1, 512] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_float16_d256v512_b1t1h1_bias0attn0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [1, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [1, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [1, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [1, 512] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d256v512_b1t1h1_bias1attn0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [1, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [1, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [1, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } }, + "biasT": { + "dtype": "float16", + "shape": [1024], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [1, 512] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d256v512_b1t1h1_bias0attn1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [1, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [1, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [1, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 1, 1, 1], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [1, 512] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float16_d256v512_b1t1h1_bias1attn1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [1, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [1, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [1, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } }, + "biasT": { + "dtype": "float16", + "shape": [1024], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 1, 1, 1], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [1, 512] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_float32_d16v20_b3t33h3_bias1attn1_keytile1", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 48], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 48], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 60], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float32", + "shape": [156], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 60] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002, + "tunables": { "CLUSTER_TILE_K": 1 } + }, + { + "name": "unequal_float32_d16v20_b3t33h3_bias1attn1_keytile2", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 48], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 48], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 60], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float32", + "shape": [156], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 60] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002, + "tunables": { "CLUSTER_TILE_K": 2 } + }, + { + "name": "unequal_float32_d16v20_b3t33h3_bias1attn1_keytile3", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 48], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 48], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 60], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float32", + "shape": [156], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 60] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002, + "tunables": { "CLUSTER_TILE_K": 3 } + }, + { + "name": "unequal_float32_d16v20_b3t33h3_bias1attn1_keytile7", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 48], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 48], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 60], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float32", + "shape": [156], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 60] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002, + "tunables": { "CLUSTER_TILE_K": 7 } + }, + { + "name": "unequal_float32_d16v20_b3t33h3_bias1attn1_keytile8", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 48], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 48], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 60], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float32", + "shape": [156], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 60] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002, + "tunables": { "CLUSTER_TILE_K": 8 } + }, + { + "name": "unequal_float32_d16v20_b3t33h3_bias1attn1_keytile15", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 48], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 48], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 60], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float32", + "shape": [156], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 60] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002, + "tunables": { "CLUSTER_TILE_K": 15 } + }, + { + "name": "unequal_float32_d16v20_b3t33h3_bias1attn1_keytile16", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 48], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 48], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 60], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float32", + "shape": [156], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 60] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002, + "tunables": { "CLUSTER_TILE_K": 16 } + }, + { + "name": "unequal_float32_d16v20_b3t33h3_bias1attn1_keytile17", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 48], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 48], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 60], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float32", + "shape": [156], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 60] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002, + "tunables": { "CLUSTER_TILE_K": 17 } + }, + { + "name": "unequal_float32_d16v20_b3t33h3_bias1attn1_keytile31", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 48], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 48], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 60], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float32", + "shape": [156], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 60] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002, + "tunables": { "CLUSTER_TILE_K": 31 } + }, + { + "name": "unequal_float32_d16v20_b3t33h3_bias1attn1_keytile32", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 48], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 48], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 60], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float32", + "shape": [156], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 60] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002, + "tunables": { "CLUSTER_TILE_K": 32 } + }, + { + "name": "unequal_float32_d32v20_b4t34h2_bias1attn1_keytile1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [34, 40], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [4, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v20_b4t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [5], + "data": { "kind": "values", "values": [0, 0, 1, 17, 34] } + }, + "biasT": { + "dtype": "float32", + "shape": [168], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [4, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [34, 40] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002, + "tunables": { "CLUSTER_TILE_K": 1 } + }, + { + "name": "unequal_float32_d32v20_b4t34h2_bias1attn1_keytile2", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [34, 40], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [4, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v20_b4t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [5], + "data": { "kind": "values", "values": [0, 0, 1, 17, 34] } + }, + "biasT": { + "dtype": "float32", + "shape": [168], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [4, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [34, 40] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002, + "tunables": { "CLUSTER_TILE_K": 2 } + }, + { + "name": "unequal_float32_d32v20_b4t34h2_bias1attn1_keytile3", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [34, 40], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [4, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v20_b4t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [5], + "data": { "kind": "values", "values": [0, 0, 1, 17, 34] } + }, + "biasT": { + "dtype": "float32", + "shape": [168], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [4, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [34, 40] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002, + "tunables": { "CLUSTER_TILE_K": 3 } + }, + { + "name": "unequal_float32_d32v20_b4t34h2_bias1attn1_keytile7", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [34, 40], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [4, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v20_b4t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [5], + "data": { "kind": "values", "values": [0, 0, 1, 17, 34] } + }, + "biasT": { + "dtype": "float32", + "shape": [168], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [4, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [34, 40] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002, + "tunables": { "CLUSTER_TILE_K": 7 } + }, + { + "name": "unequal_float32_d32v20_b4t34h2_bias1attn1_keytile8", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [34, 40], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [4, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v20_b4t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [5], + "data": { "kind": "values", "values": [0, 0, 1, 17, 34] } + }, + "biasT": { + "dtype": "float32", + "shape": [168], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [4, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [34, 40] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002, + "tunables": { "CLUSTER_TILE_K": 8 } + }, + { + "name": "unequal_float32_d32v20_b4t34h2_bias1attn1_keytile15", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [34, 40], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [4, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v20_b4t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [5], + "data": { "kind": "values", "values": [0, 0, 1, 17, 34] } + }, + "biasT": { + "dtype": "float32", + "shape": [168], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [4, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [34, 40] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002, + "tunables": { "CLUSTER_TILE_K": 15 } + }, + { + "name": "unequal_float32_d32v20_b4t34h2_bias1attn1_keytile16", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [34, 40], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [4, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v20_b4t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [5], + "data": { "kind": "values", "values": [0, 0, 1, 17, 34] } + }, + "biasT": { + "dtype": "float32", + "shape": [168], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [4, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [34, 40] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002, + "tunables": { "CLUSTER_TILE_K": 16 } + }, + { + "name": "unequal_float32_d32v20_b4t34h2_bias1attn1_keytile17", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [34, 40], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [4, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v20_b4t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [5], + "data": { "kind": "values", "values": [0, 0, 1, 17, 34] } + }, + "biasT": { + "dtype": "float32", + "shape": [168], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [4, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [34, 40] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002, + "tunables": { "CLUSTER_TILE_K": 17 } + }, + { + "name": "unequal_float32_d32v20_b4t34h2_bias1attn1_keytile31", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [34, 40], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [4, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v20_b4t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [5], + "data": { "kind": "values", "values": [0, 0, 1, 17, 34] } + }, + "biasT": { + "dtype": "float32", + "shape": [168], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [4, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [34, 40] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002, + "tunables": { "CLUSTER_TILE_K": 31 } + }, + { + "name": "unequal_float32_d32v20_b4t34h2_bias1attn1_keytile32", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [34, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [34, 40], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [4, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v20_b4t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [5], + "data": { "kind": "values", "values": [0, 0, 1, 17, 34] } + }, + "biasT": { + "dtype": "float32", + "shape": [168], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [4, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [34, 40] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002, + "tunables": { "CLUSTER_TILE_K": 32 } + }, + { + "name": "unequal_f16_bias_key_sum_exceeds_f16_finite_range", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { "dtype": "float16", "shape": [34, 64], "data": { "kind": "constant", "value": 0.0009765625 } }, + "keyT": { "dtype": "float16", "shape": [34, 64], "data": { "kind": "constant", "value": 65504.0 } }, + "valueT": { + "dtype": "float16", + "shape": [34, 40], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_f16_bias_key_sum_exceeds_f16_finite_range_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d32v4_b3t34h2_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 17, 34] } + }, + "biasT": { + "dtype": "float16", + "shape": [168], + "data": { + "kind": "values", + "values": [0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25] + } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [34, 40] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_useful_float32_d32v20_t2_bias0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [2, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [2, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [2, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 2], "data": { "kind": "values", "values": [0, 1] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 2] } } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [2, 20] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_useful_float32_d32v20_t2_bias1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [2, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [2, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [2, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 2], "data": { "kind": "values", "values": [0, 1] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 2] } }, + "biasT": { + "dtype": "float32", + "shape": [84], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 1, 2, 2], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [2, 20] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_useful_float16_d32v20_t2_bias0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [2, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [2, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [2, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 2], "data": { "kind": "values", "values": [0, 1] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 2] } } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [2, 20] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_useful_float16_d32v20_t2_bias1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [2, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [2, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [2, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 2], "data": { "kind": "values", "values": [0, 1] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 2] } }, + "biasT": { + "dtype": "float16", + "shape": [84], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 1, 2, 2], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [2, 20] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_useful_float32_d32v20_t3_bias0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [3, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [3, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [3, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 3], "data": { "kind": "values", "values": [0, 1, 2] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 3] } } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [3, 20] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_useful_float32_d32v20_t3_bias1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [3, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [3, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [3, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 3], "data": { "kind": "values", "values": [0, 1, 2] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 3] } }, + "biasT": { + "dtype": "float32", + "shape": [84], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 1, 3, 3], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [3, 20] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_useful_float16_d32v20_t3_bias0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [3, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [3, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [3, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 3], "data": { "kind": "values", "values": [0, 1, 2] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 3] } } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [3, 20] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_useful_float16_d32v20_t3_bias1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [3, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [3, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [3, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 3], "data": { "kind": "values", "values": [0, 1, 2] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 3] } }, + "biasT": { + "dtype": "float16", + "shape": [84], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 1, 3, 3], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [3, 20] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_useful_float32_d32v20_t8_bias0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [8, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [8, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [8, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [1, 8], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5, 6, 7] } + }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 8] } } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [8, 20] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_useful_float32_d32v20_t8_bias1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [8, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [8, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [8, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [1, 8], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5, 6, 7] } + }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 8] } }, + "biasT": { + "dtype": "float32", + "shape": [84], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 1, 8, 8], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [8, 20] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_useful_float16_d32v20_t8_bias0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [8, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [8, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [8, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [1, 8], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5, 6, 7] } + }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 8] } } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [8, 20] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_useful_float16_d32v20_t8_bias1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [8, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [8, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [8, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [1, 8], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5, 6, 7] } + }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 8] } }, + "biasT": { + "dtype": "float16", + "shape": [84], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 1, 8, 8], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [8, 20] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_useful_float32_d32v20_t15_bias0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [15, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [15, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [15, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [1, 15], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14] } + }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 15] } } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [15, 20] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_useful_float32_d32v20_t15_bias1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [15, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [15, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [15, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [1, 15], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14] } + }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 15] } }, + "biasT": { + "dtype": "float32", + "shape": [84], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 1, 15, 15], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [15, 20] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_useful_float16_d32v20_t15_bias0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [15, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [15, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [15, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [1, 15], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14] } + }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 15] } } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [15, 20] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_useful_float16_d32v20_t15_bias1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [15, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [15, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [15, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [1, 15], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14] } + }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 15] } }, + "biasT": { + "dtype": "float16", + "shape": [84], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 1, 15, 15], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [15, 20] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_useful_float32_d256v512_t2_bias0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [2, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [2, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [2, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 2], "data": { "kind": "values", "values": [0, 1] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 2] } } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [2, 512] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_useful_float32_d256v512_t2_bias1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [2, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [2, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [2, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 2], "data": { "kind": "values", "values": [0, 1] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 2] } }, + "biasT": { + "dtype": "float32", + "shape": [1024], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 1, 2, 2], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [2, 512] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_useful_float16_d256v512_t2_bias0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [2, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [2, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [2, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 2], "data": { "kind": "values", "values": [0, 1] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 2] } } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [2, 512] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_useful_float16_d256v512_t2_bias1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [2, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [2, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [2, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 2], "data": { "kind": "values", "values": [0, 1] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 2] } }, + "biasT": { + "dtype": "float16", + "shape": [1024], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 1, 2, 2], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [2, 512] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_useful_float32_d256v512_t3_bias0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [3, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [3, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [3, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 3], "data": { "kind": "values", "values": [0, 1, 2] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 3] } } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [3, 512] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_useful_float32_d256v512_t3_bias1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [3, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [3, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [3, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 3], "data": { "kind": "values", "values": [0, 1, 2] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 3] } }, + "biasT": { + "dtype": "float32", + "shape": [1024], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 1, 3, 3], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [3, 512] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_useful_float16_d256v512_t3_bias0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [3, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [3, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [3, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 3], "data": { "kind": "values", "values": [0, 1, 2] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 3] } } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [3, 512] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_useful_float16_d256v512_t3_bias1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [3, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [3, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [3, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 3], "data": { "kind": "values", "values": [0, 1, 2] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 3] } }, + "biasT": { + "dtype": "float16", + "shape": [1024], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 1, 3, 3], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [3, 512] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_useful_float32_d256v512_t8_bias0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [8, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [8, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [8, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [1, 8], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5, 6, 7] } + }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 8] } } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [8, 512] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_useful_float32_d256v512_t8_bias1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [8, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [8, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [8, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [1, 8], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5, 6, 7] } + }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 8] } }, + "biasT": { + "dtype": "float32", + "shape": [1024], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 1, 8, 8], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [8, 512] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_useful_float16_d256v512_t8_bias0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [8, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [8, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [8, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [1, 8], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5, 6, 7] } + }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 8] } } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [8, 512] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_useful_float16_d256v512_t8_bias1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [8, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [8, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [8, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [1, 8], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5, 6, 7] } + }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 8] } }, + "biasT": { + "dtype": "float16", + "shape": [1024], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 1, 8, 8], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [8, 512] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_useful_float32_d256v512_t15_bias0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [15, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [15, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [15, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [1, 15], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14] } + }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 15] } } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [15, 512] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_useful_float32_d256v512_t15_bias1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [15, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [15, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [15, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [1, 15], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14] } + }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 15] } }, + "biasT": { + "dtype": "float32", + "shape": [1024], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 1, 15, 15], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [15, 512] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_useful_float16_d256v512_t15_bias0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [15, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [15, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [15, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [1, 15], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14] } + }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 15] } } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [15, 512] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_useful_float16_d256v512_t15_bias1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [15, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [15, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [15, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [1, 15], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14] } + }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 15] } }, + "biasT": { + "dtype": "float16", + "shape": [1024], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 1, 15, 15], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [15, 512] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_useful_float32_d16v20_t1_bias0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [1, 16], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [1, 16], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [1, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [1, 20] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_useful_float32_d16v20_t1_bias1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [1, 16], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [1, 16], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [1, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } }, + "biasT": { + "dtype": "float32", + "shape": [52], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 1, 1, 1], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [1, 20] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_useful_float16_d16v20_t1_bias0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [1, 16], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [1, 16], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [1, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [1, 20] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_useful_float16_d16v20_t1_bias1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [1, 16], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [1, 16], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [1, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } }, + "biasT": { + "dtype": "float16", + "shape": [52], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 1, 1, 1], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [1, 20] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_useful_float32_d16v20_t3_bias0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [3, 16], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [3, 16], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [3, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 3], "data": { "kind": "values", "values": [0, 1, 2] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 3] } } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [3, 20] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_useful_float32_d16v20_t3_bias1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [3, 16], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [3, 16], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [3, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 3], "data": { "kind": "values", "values": [0, 1, 2] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 3] } }, + "biasT": { + "dtype": "float32", + "shape": [52], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 1, 3, 3], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [3, 20] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_useful_float16_d16v20_t3_bias0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [3, 16], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [3, 16], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [3, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 3], "data": { "kind": "values", "values": [0, 1, 2] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 3] } } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [3, 20] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_useful_float16_d16v20_t3_bias1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [3, 16], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [3, 16], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [3, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 3], "data": { "kind": "values", "values": [0, 1, 2] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 3] } }, + "biasT": { + "dtype": "float16", + "shape": [52], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 1, 3, 3], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [3, 20] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_useful_float32_d16v20_t7_bias0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [7, 16], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [7, 16], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [7, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [1, 7], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5, 6] } + }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 7] } } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [7, 20] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_useful_float32_d16v20_t7_bias1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [7, 16], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [7, 16], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [7, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [1, 7], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5, 6] } + }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 7] } }, + "biasT": { + "dtype": "float32", + "shape": [52], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 1, 7, 7], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [7, 20] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_useful_float16_d16v20_t7_bias0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [7, 16], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [7, 16], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [7, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [1, 7], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5, 6] } + }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 7] } } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [7, 20] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_useful_float16_d16v20_t7_bias1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [7, 16], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [7, 16], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [7, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [1, 7], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5, 6] } + }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 7] } }, + "biasT": { + "dtype": "float16", + "shape": [52], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 1, 7, 7], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [7, 20] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_reuse_float32_d32v20_t4_bias0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [4, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [4, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [4, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 4], "data": { "kind": "values", "values": [0, 1, 2, 3] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 4] } } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [4, 20] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_reuse_float32_d32v20_t4_bias1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [4, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [4, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [4, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 4], "data": { "kind": "values", "values": [0, 1, 2, 3] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 4] } }, + "biasT": { + "dtype": "float32", + "shape": [84], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 1, 4, 4], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [4, 20] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_reuse_float16_d32v20_t4_bias0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [4, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [4, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [4, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 4], "data": { "kind": "values", "values": [0, 1, 2, 3] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 4] } } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [4, 20] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_reuse_float16_d32v20_t4_bias1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [4, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [4, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [4, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 4], "data": { "kind": "values", "values": [0, 1, 2, 3] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 4] } }, + "biasT": { + "dtype": "float16", + "shape": [84], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 1, 4, 4], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [4, 20] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_reuse_float32_d32v20_t7_bias0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [7, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [7, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [7, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [1, 7], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5, 6] } + }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 7] } } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [7, 20] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_reuse_float32_d32v20_t7_bias1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [7, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [7, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [7, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [1, 7], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5, 6] } + }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 7] } }, + "biasT": { + "dtype": "float32", + "shape": [84], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 1, 7, 7], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [7, 20] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_reuse_float16_d32v20_t7_bias0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [7, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [7, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [7, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [1, 7], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5, 6] } + }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 7] } } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [7, 20] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_reuse_float16_d32v20_t7_bias1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [7, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [7, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [7, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [1, 7], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5, 6] } + }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 7] } }, + "biasT": { + "dtype": "float16", + "shape": [84], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 1, 7, 7], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [7, 20] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_reuse_float32_d256v512_t4_bias0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [4, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [4, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [4, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 4], "data": { "kind": "values", "values": [0, 1, 2, 3] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 4] } } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [4, 512] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_reuse_float32_d256v512_t4_bias1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [4, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [4, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [4, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 4], "data": { "kind": "values", "values": [0, 1, 2, 3] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 4] } }, + "biasT": { + "dtype": "float32", + "shape": [1024], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 1, 4, 4], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [4, 512] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_reuse_float16_d256v512_t4_bias0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [4, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [4, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [4, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 4], "data": { "kind": "values", "values": [0, 1, 2, 3] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 4] } } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [4, 512] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_reuse_float16_d256v512_t4_bias1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [4, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [4, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [4, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 4], "data": { "kind": "values", "values": [0, 1, 2, 3] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 4] } }, + "biasT": { + "dtype": "float16", + "shape": [1024], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 1, 4, 4], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [4, 512] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_reuse_float32_d256v512_t7_bias0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [7, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [7, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [7, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [1, 7], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5, 6] } + }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 7] } } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [7, 512] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_reuse_float32_d256v512_t7_bias1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [7, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [7, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [7, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [1, 7], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5, 6] } + }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 7] } }, + "biasT": { + "dtype": "float32", + "shape": [1024], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 1, 7, 7], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [7, 512] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_reuse_float16_d256v512_t7_bias0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [7, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [7, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [7, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [1, 7], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5, 6] } + }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 7] } } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [7, 512] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_reuse_float16_d256v512_t7_bias1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [7, 256], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [7, 256], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [7, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [1, 7], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5, 6] } + }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 7] } }, + "biasT": { + "dtype": "float16", + "shape": [1024], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 1, 7, 7], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [7, 512] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_reuse_float32_d16v20_t4_bias0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [4, 16], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [4, 16], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [4, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 4], "data": { "kind": "values", "values": [0, 1, 2, 3] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 4] } } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [4, 20] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_reuse_float32_d16v20_t4_bias1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [4, 16], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [4, 16], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [4, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 4], "data": { "kind": "values", "values": [0, 1, 2, 3] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 4] } }, + "biasT": { + "dtype": "float32", + "shape": [52], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 1, 4, 4], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [4, 20] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_reuse_float16_d16v20_t4_bias0", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [4, 16], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [4, 16], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [4, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 4], "data": { "kind": "values", "values": [0, 1, 2, 3] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 4] } } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [4, 20] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_reuse_float16_d16v20_t4_bias1", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [4, 16], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [4, 16], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [4, 20], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 4], "data": { "kind": "values", "values": [0, 1, 2, 3] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 4] } }, + "biasT": { + "dtype": "float16", + "shape": [52], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 1, 4, 4], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [4, 20] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_wide_float32_d16v128_t33_bias0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 256] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_wide_float32_d16v128_t33_bias1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float32", + "shape": [320], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 256] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_wide_float16_d16v128_t33_bias0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [33, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [33, 256] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_wide_float16_d16v128_t33_bias1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [33, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [33, 256], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float16", + "shape": [320], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [33, 256] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_wide_float32_d16v512_t33_bias0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 1024], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 1024] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_wide_float32_d16v512_t33_bias1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 1024], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float32", + "shape": [1088], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 1024] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_wide_float16_d16v512_t33_bias0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [33, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [33, 1024], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [33, 1024] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_wide_float16_d16v512_t33_bias1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [33, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [33, 1024], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float16", + "shape": [1088], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [33, 1024] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_wide_float32_d32v256_t33_bias0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 512] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_wide_float32_d32v256_t33_bias1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float32", + "shape": [640], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 512] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_wide_float16_d32v256_t33_bias0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [33, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [33, 512] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_wide_float16_d32v256_t33_bias1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [33, 512], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float16", + "shape": [640], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [33, 512] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_wide_float32_d32v1024_t33_bias0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 2048], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 2048] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_wide_float32_d32v1024_t33_bias1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [33, 2048], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float32", + "shape": [2176], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [33, 2048] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_wide_float16_d32v1024_t33_bias0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [33, 2048], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [33, 2048] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_wide_float16_d32v1024_t33_bias1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [33, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [33, 2048], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [3, 17], + "data": { + "kind": "values", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v20_b3t33h3_bias0attn0_input_tokenOffsetT" } + } + }, + "cumulativeSequenceLengthT": { + "dtype": "int32", + "shape": [4], + "data": { "kind": "values", "values": [0, 1, 16, 33] } + }, + "biasT": { + "dtype": "float16", + "shape": [2176], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [3, 1, 17, 17], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [33, 2048] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_wide_boundary_float32_d16v512_t1_bias0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [1, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [1, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [1, 1024], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [1, 1024] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_wide_boundary_float32_d16v512_t1_bias1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [1, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [1, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [1, 1024], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } }, + "biasT": { + "dtype": "float32", + "shape": [1088], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 1, 1, 1], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [1, 1024] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_wide_boundary_float16_d16v512_t1_bias0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [1, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [1, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [1, 1024], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [1, 1024] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_wide_boundary_float16_d16v512_t1_bias1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [1, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [1, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [1, 1024], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } }, + "biasT": { + "dtype": "float16", + "shape": [1088], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 1, 1, 1], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [1, 1024] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_wide_boundary_float32_d16v512_t4_bias0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [4, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [4, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [4, 1024], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 4], "data": { "kind": "values", "values": [0, 1, 2, 3] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 4] } } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [4, 1024] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_wide_boundary_float32_d16v512_t4_bias1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [4, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [4, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [4, 1024], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 4], "data": { "kind": "values", "values": [0, 1, 2, 3] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 4] } }, + "biasT": { + "dtype": "float32", + "shape": [1088], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 1, 4, 4], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [4, 1024] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_wide_boundary_float16_d16v512_t4_bias0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [4, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [4, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [4, 1024], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 4], "data": { "kind": "values", "values": [0, 1, 2, 3] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 4] } } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [4, 1024] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_wide_boundary_float16_d16v512_t4_bias1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [4, 32], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [4, 32], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [4, 1024], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 4], "data": { "kind": "values", "values": [0, 1, 2, 3] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 4] } }, + "biasT": { + "dtype": "float16", + "shape": [1088], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 1, 4, 4], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [4, 1024] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_wide_boundary_float32_d32v1024_t1_bias0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [1, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [1, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [1, 2048], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [1, 2048] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_wide_boundary_float32_d32v1024_t1_bias1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [1, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [1, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [1, 2048], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } }, + "biasT": { + "dtype": "float32", + "shape": [2176], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 1, 1, 1], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [1, 2048] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_wide_boundary_float16_d32v1024_t1_bias0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [1, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [1, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [1, 2048], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [1, 2048] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_wide_boundary_float16_d32v1024_t1_bias1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [1, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [1, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [1, 2048], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 1], "data": { "kind": "values", "values": [0] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 1] } }, + "biasT": { + "dtype": "float16", + "shape": [2176], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 1, 1, 1], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [1, 2048] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_wide_boundary_float32_d32v1024_t8_bias0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [8, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [8, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [8, 2048], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [1, 8], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5, 6, 7] } + }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 8] } } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [8, 2048] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_wide_boundary_float32_d32v1024_t8_bias1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float32", + "shape": [8, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float32", + "shape": [8, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float32", + "shape": [8, 2048], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [1, 8], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5, 6, 7] } + }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 8] } }, + "biasT": { + "dtype": "float32", + "shape": [2176], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float32", + "shape": [1, 1, 8, 8], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float32", "shape": [8, 2048] } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "unequal_wide_boundary_float16_d32v1024_t8_bias0", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [8, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [8, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [8, 2048], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [1, 8], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5, 6, 7] } + }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 8] } } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [8, 2048] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_wide_boundary_float16_d32v1024_t8_bias1", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "dtype": "float16", + "shape": [8, 64], + "data": { + "kind": "cycle", + "values": [-0.5, 0.25, 0.75, -0.125, 0.375, -0.75, 0.5, 0.125, -0.25, 0.625, -0.375] + } + }, + "keyT": { + "dtype": "float16", + "shape": [8, 64], + "data": { + "kind": "cycle", + "values": [0.375, -0.25, 0.125, 0.5, -0.625, 0.75, -0.5, 0.25, -0.125, 0.875, -0.375, 0.625, 0.0625] + } + }, + "valueT": { + "dtype": "float16", + "shape": [8, 2048], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_float32_d16v12_b3t9h2_bias0attn0_input_valueT" } + } + }, + "tokenOffsetT": { + "dtype": "int32", + "shape": [1, 8], + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5, 6, 7] } + }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 8] } }, + "biasT": { + "dtype": "float16", + "shape": [2176], + "data": { "kind": "cycle", "values": [0.125, -0.25, 0.375, -0.125, 0.0625] } + }, + "attentionBiasT": { + "dtype": "float16", + "shape": [1, 1, 8, 8], + "data": { "kind": "cycle", "values": [0.25, -0.5, 0.375, 0.125, -0.25, 0.0625, -0.125] } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [8, 2048] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "unequal_f16_wide_value_bias_sum_exceeds_f16_finite_range", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { "dtype": "float16", "shape": [4, 32], "data": { "kind": "constant", "value": 0.0009765625 } }, + "keyT": { "dtype": "float16", "shape": [4, 32], "data": { "kind": "constant", "value": 65504.0 } }, + "valueT": { + "dtype": "float16", + "shape": [4, 1024], + "data": { + "kind": "cycle", + "values": { "$ref": "#/fixtureArrays/unequal_f16_bias_key_sum_exceeds_f16_finite_range_input_valueT" } + } + }, + "tokenOffsetT": { "dtype": "int32", "shape": [1, 4], "data": { "kind": "values", "values": [0, 1, 2, 3] } }, + "cumulativeSequenceLengthT": { "dtype": "int32", "shape": [2], "data": { "kind": "values", "values": [0, 4] } }, + "biasT": { + "dtype": "float16", + "shape": [1088], + "data": { + "kind": "values", + "values": [0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 65504.0, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25, 0.25] + } + } + }, + "outputs": { "outputT": { "dtype": "float16", "shape": [4, 1024] } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-wide-float32-t7-h1-d192-packed", + "attrs": { "num_heads": 1, "scale": 0 }, + "inputs": { + "queryT": { + "shape": [7, 1, 3, 192], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.157, "cosStep": 0.211, "scale": 0.4 } + }, + "tokenOffsetT": { + "shape": [1, 10], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5, 6, 7, 8, 9] } + }, + "cumulativeSequenceLengthT": { "shape": [2], "dtype": "int32", "data": { "kind": "values", "values": [0, 7] } }, + "attentionBiasT": { + "shape": [1, 1, 10, 10], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.20800000000000002, "cosStep": 0.25, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [7, 192], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-wide-float32-t7-h1-d256-separate", + "attrs": { "num_heads": 1, "scale": 0.071 }, + "inputs": { + "queryT": { + "shape": [7, 256], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.174, "cosStep": 0.237, "scale": 0.4 } + }, + "keyT": { + "shape": [7, 256], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.191, "cosStep": 0.25, "scale": 0.4 } + }, + "valueT": { + "shape": [7, 256], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.20800000000000002, "cosStep": 0.263, "scale": 0.4 } + }, + "biasT": { + "shape": [768], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.225, "cosStep": 0.276, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 9], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_34" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 7] } + } + }, + "outputs": { "outputT": { "shape": [7, 256], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-wide-float32-t16-h2-d256-separate", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "shape": [16, 512], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.191, "cosStep": 0.263, "scale": 0.4 } + }, + "keyT": { + "shape": [16, 512], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.20800000000000002, "cosStep": 0.276, "scale": 0.4 } + }, + "valueT": { + "shape": [16, 512], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.225, "cosStep": 0.289, "scale": 0.4 } + }, + "biasT": { + "shape": [1536], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.242, "cosStep": 0.302, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 18], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_35" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 16] } + }, + "attentionBiasT": { + "shape": [1, 1, 18, 18], + "dtype": "float32", + "data": { + "kind": "fillFloat32", + "sinStep": 0.29300000000000004, + "cosStep": 0.34099999999999997, + "scale": 0.1 + } + } + }, + "outputs": { "outputT": { "shape": [16, 512], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-wide-float32-t31-h1-d192-packed", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "shape": [31, 1, 3, 192], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.20800000000000002, "cosStep": 0.289, "scale": 0.4 } + }, + "tokenOffsetT": { + "shape": [3, 32], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_36" } } + }, + "cumulativeSequenceLengthT": { + "shape": [4], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 0, 2, 31] } + }, + "attentionBiasT": { + "shape": [3, 1, 32, 32], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.259, "cosStep": 0.32799999999999996, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [31, 192], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-wide-float32-t33-h2-d320-separate", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "shape": [33, 640], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.225, "cosStep": 0.315, "scale": 0.4 } + }, + "keyT": { + "shape": [33, 640], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.242, "cosStep": 0.32799999999999996, "scale": 0.4 } + }, + "valueT": { + "shape": [33, 640], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.259, "cosStep": 0.34099999999999997, "scale": 0.4 } + }, + "biasT": { + "shape": [1920], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.276, "cosStep": 0.354, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 35], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_37" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 33] } + } + }, + "outputs": { "outputT": { "shape": [33, 640], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-wide-float32-t64-h2-d256-packed", + "attrs": { "num_heads": 2, "scale": 0 }, + "inputs": { + "queryT": { + "shape": [64, 2, 3, 256], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.242, "cosStep": 0.34099999999999997, "scale": 0.4 } + }, + "biasT": { + "shape": [1536], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.259, "cosStep": 0.354, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 66], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_38" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 64] } + }, + "attentionBiasT": { + "shape": [2, 2, 66, 66], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.31000000000000005, "cosStep": 0.393, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [64, 512], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-wide-float32-t129-h4-d192-separate", + "attrs": { "num_heads": 4, "scale": 0.071 }, + "inputs": { + "queryT": { + "shape": [129, 768], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.259, "cosStep": 0.367, "scale": 0.4 } + }, + "keyT": { + "shape": [129, 768], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.276, "cosStep": 0.38, "scale": 0.4 } + }, + "valueT": { + "shape": [129, 768], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.29300000000000004, "cosStep": 0.393, "scale": 0.4 } + }, + "tokenOffsetT": { + "shape": [2, 131], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_39" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 129] } + }, + "attentionBiasT": { + "shape": [1, 1, 131, 131], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.344, "cosStep": 0.432, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [129, 768], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-wide-float32-t257-h2-d256-separate", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "shape": [257, 512], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.276, "cosStep": 0.393, "scale": 0.4 } + }, + "keyT": { + "shape": [257, 512], + "dtype": "float32", + "data": { + "kind": "fillFloat32", + "sinStep": 0.29300000000000004, + "cosStep": 0.40599999999999997, + "scale": 0.4 + } + }, + "valueT": { + "shape": [257, 512], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.31000000000000005, "cosStep": 0.419, "scale": 0.4 } + }, + "biasT": { + "shape": [1536], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.327, "cosStep": 0.432, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 259], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_40" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 257] } + } + }, + "outputs": { "outputT": { "shape": [257, 512], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-wide-float32-t513-h2-d256-packed", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "shape": [513, 2, 3, 256], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.29300000000000004, "cosStep": 0.419, "scale": 0.4 } + }, + "biasT": { + "shape": [1536], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.31000000000000005, "cosStep": 0.432, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 515], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_41" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 513] } + }, + "attentionBiasT": { + "shape": [1, 1, 515, 515], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.361, "cosStep": 0.471, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [513, 512], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-wide-float32-t129-h1-d384-separate", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "shape": [129, 384], + "dtype": "float32", + "data": { + "kind": "fillFloat32", + "sinStep": 0.31000000000000005, + "cosStep": 0.44499999999999995, + "scale": 0.4 + } + }, + "keyT": { + "shape": [129, 384], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.327, "cosStep": 0.45799999999999996, "scale": 0.4 } + }, + "valueT": { + "shape": [129, 384], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.344, "cosStep": 0.471, "scale": 0.4 } + }, + "tokenOffsetT": { + "shape": [2, 131], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_39" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 129] } + }, + "attentionBiasT": { + "shape": [2, 1, 131, 131], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.395, "cosStep": 0.51, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [129, 384], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-wide-float32-t257-h2-d512-packed", + "attrs": { "num_heads": 2, "scale": 0 }, + "inputs": { + "queryT": { + "shape": [257, 2, 3, 512], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.327, "cosStep": 0.471, "scale": 0.4 } + }, + "biasT": { + "shape": [3072], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.344, "cosStep": 0.484, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 259], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_40" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 257] } + } + }, + "outputs": { "outputT": { "shape": [257, 1024], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-wide-float32-t33-h2-d128-packed", + "attrs": { "num_heads": 2, "scale": 0.071 }, + "inputs": { + "queryT": { + "shape": [33, 2, 3, 128], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.344, "cosStep": 0.497, "scale": 0.4 } + }, + "biasT": { + "shape": [768], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.361, "cosStep": 0.51, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 35], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_37" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 33] } + }, + "attentionBiasT": { + "shape": [2, 2, 35, 35], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.41200000000000003, "cosStep": 0.5489999999999999, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [33, 256], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-wide-float32-t33-h2-d160-separate", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "shape": [33, 320], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.361, "cosStep": 0.523, "scale": 0.4 } + }, + "keyT": { + "shape": [33, 320], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.378, "cosStep": 0.536, "scale": 0.4 } + }, + "valueT": { + "shape": [33, 320], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.395, "cosStep": 0.5489999999999999, "scale": 0.4 } + }, + "tokenOffsetT": { + "shape": [2, 35], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_37" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 33] } + }, + "attentionBiasT": { + "shape": [1, 1, 35, 35], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.44600000000000006, "cosStep": 0.588, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [33, 320], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-wide-float32-t33-h2-d224-separate", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "shape": [33, 448], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.378, "cosStep": 0.5489999999999999, "scale": 0.4 } + }, + "keyT": { + "shape": [33, 448], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.395, "cosStep": 0.5619999999999999, "scale": 0.4 } + }, + "valueT": { + "shape": [33, 448], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.41200000000000003, "cosStep": 0.575, "scale": 0.4 } + }, + "biasT": { + "shape": [1344], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.42900000000000005, "cosStep": 0.588, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 35], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_37" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 33] } + } + }, + "outputs": { "outputT": { "shape": [33, 448], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-wide-float16-t7-h1-d192-packed", + "attrs": { "num_heads": 1, "scale": 0 }, + "inputs": { + "queryT": { + "shape": [7, 1, 3, 192], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.157, "cosStep": 0.211, "scale": 0.4 } + }, + "tokenOffsetT": { + "shape": [1, 10], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 2, 3, 4, 5, 6, 7, 8, 9] } + }, + "cumulativeSequenceLengthT": { "shape": [2], "dtype": "int32", "data": { "kind": "values", "values": [0, 7] } }, + "attentionBiasT": { + "shape": [1, 1, 10, 10], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.20800000000000002, "cosStep": 0.25, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [7, 192], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-wide-float16-t7-h1-d256-separate", + "attrs": { "num_heads": 1, "scale": 0.071 }, + "inputs": { + "queryT": { + "shape": [7, 256], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.174, "cosStep": 0.237, "scale": 0.4 } + }, + "keyT": { + "shape": [7, 256], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.191, "cosStep": 0.25, "scale": 0.4 } + }, + "valueT": { + "shape": [7, 256], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.20800000000000002, "cosStep": 0.263, "scale": 0.4 } + }, + "biasT": { + "shape": [768], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.225, "cosStep": 0.276, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 9], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_34" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 7] } + } + }, + "outputs": { "outputT": { "shape": [7, 256], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-wide-float16-t16-h2-d256-separate", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "shape": [16, 512], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.191, "cosStep": 0.263, "scale": 0.4 } + }, + "keyT": { + "shape": [16, 512], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.20800000000000002, "cosStep": 0.276, "scale": 0.4 } + }, + "valueT": { + "shape": [16, 512], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.225, "cosStep": 0.289, "scale": 0.4 } + }, + "biasT": { + "shape": [1536], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.242, "cosStep": 0.302, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 18], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_35" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 16] } + }, + "attentionBiasT": { + "shape": [1, 1, 18, 18], + "dtype": "float16", + "data": { + "kind": "fillFloat32", + "sinStep": 0.29300000000000004, + "cosStep": 0.34099999999999997, + "scale": 0.1 + } + } + }, + "outputs": { "outputT": { "shape": [16, 512], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-wide-float16-t31-h1-d192-packed", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "shape": [31, 1, 3, 192], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.20800000000000002, "cosStep": 0.289, "scale": 0.4 } + }, + "tokenOffsetT": { + "shape": [3, 32], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_36" } } + }, + "cumulativeSequenceLengthT": { + "shape": [4], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 0, 2, 31] } + }, + "attentionBiasT": { + "shape": [3, 1, 32, 32], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.259, "cosStep": 0.32799999999999996, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [31, 192], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-wide-float16-t33-h2-d320-separate", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "shape": [33, 640], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.225, "cosStep": 0.315, "scale": 0.4 } + }, + "keyT": { + "shape": [33, 640], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.242, "cosStep": 0.32799999999999996, "scale": 0.4 } + }, + "valueT": { + "shape": [33, 640], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.259, "cosStep": 0.34099999999999997, "scale": 0.4 } + }, + "biasT": { + "shape": [1920], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.276, "cosStep": 0.354, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 35], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_37" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 33] } + } + }, + "outputs": { "outputT": { "shape": [33, 640], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-wide-float16-t64-h2-d256-packed", + "attrs": { "num_heads": 2, "scale": 0 }, + "inputs": { + "queryT": { + "shape": [64, 2, 3, 256], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.242, "cosStep": 0.34099999999999997, "scale": 0.4 } + }, + "biasT": { + "shape": [1536], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.259, "cosStep": 0.354, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 66], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_38" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 64] } + }, + "attentionBiasT": { + "shape": [2, 2, 66, 66], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.31000000000000005, "cosStep": 0.393, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [64, 512], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-wide-float16-t129-h4-d192-separate", + "attrs": { "num_heads": 4, "scale": 0.071 }, + "inputs": { + "queryT": { + "shape": [129, 768], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.259, "cosStep": 0.367, "scale": 0.4 } + }, + "keyT": { + "shape": [129, 768], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.276, "cosStep": 0.38, "scale": 0.4 } + }, + "valueT": { + "shape": [129, 768], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.29300000000000004, "cosStep": 0.393, "scale": 0.4 } + }, + "tokenOffsetT": { + "shape": [2, 131], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_39" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 129] } + }, + "attentionBiasT": { + "shape": [1, 1, 131, 131], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.344, "cosStep": 0.432, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [129, 768], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-wide-float16-t257-h2-d256-separate", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "shape": [257, 512], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.276, "cosStep": 0.393, "scale": 0.4 } + }, + "keyT": { + "shape": [257, 512], + "dtype": "float16", + "data": { + "kind": "fillFloat32", + "sinStep": 0.29300000000000004, + "cosStep": 0.40599999999999997, + "scale": 0.4 + } + }, + "valueT": { + "shape": [257, 512], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.31000000000000005, "cosStep": 0.419, "scale": 0.4 } + }, + "biasT": { + "shape": [1536], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.327, "cosStep": 0.432, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 259], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_40" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 257] } + } + }, + "outputs": { "outputT": { "shape": [257, 512], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-wide-float16-t513-h2-d256-packed", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "shape": [513, 2, 3, 256], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.29300000000000004, "cosStep": 0.419, "scale": 0.4 } + }, + "biasT": { + "shape": [1536], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.31000000000000005, "cosStep": 0.432, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 515], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_41" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 513] } + }, + "attentionBiasT": { + "shape": [1, 1, 515, 515], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.361, "cosStep": 0.471, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [513, 512], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-wide-float16-t129-h1-d384-separate", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "shape": [129, 384], + "dtype": "float16", + "data": { + "kind": "fillFloat32", + "sinStep": 0.31000000000000005, + "cosStep": 0.44499999999999995, + "scale": 0.4 + } + }, + "keyT": { + "shape": [129, 384], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.327, "cosStep": 0.45799999999999996, "scale": 0.4 } + }, + "valueT": { + "shape": [129, 384], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.344, "cosStep": 0.471, "scale": 0.4 } + }, + "tokenOffsetT": { + "shape": [2, 131], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_39" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 129] } + }, + "attentionBiasT": { + "shape": [2, 1, 131, 131], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.395, "cosStep": 0.51, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [129, 384], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-wide-float16-t257-h2-d512-packed", + "attrs": { "num_heads": 2, "scale": 0 }, + "inputs": { + "queryT": { + "shape": [257, 2, 3, 512], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.327, "cosStep": 0.471, "scale": 0.4 } + }, + "biasT": { + "shape": [3072], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.344, "cosStep": 0.484, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 259], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_40" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 257] } + } + }, + "outputs": { "outputT": { "shape": [257, 1024], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-wide-float16-t33-h2-d128-packed", + "attrs": { "num_heads": 2, "scale": 0.071 }, + "inputs": { + "queryT": { + "shape": [33, 2, 3, 128], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.344, "cosStep": 0.497, "scale": 0.4 } + }, + "biasT": { + "shape": [768], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.361, "cosStep": 0.51, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 35], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_37" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 33] } + }, + "attentionBiasT": { + "shape": [2, 2, 35, 35], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.41200000000000003, "cosStep": 0.5489999999999999, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [33, 256], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-wide-float16-t33-h2-d160-separate", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "shape": [33, 320], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.361, "cosStep": 0.523, "scale": 0.4 } + }, + "keyT": { + "shape": [33, 320], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.378, "cosStep": 0.536, "scale": 0.4 } + }, + "valueT": { + "shape": [33, 320], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.395, "cosStep": 0.5489999999999999, "scale": 0.4 } + }, + "tokenOffsetT": { + "shape": [2, 35], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_37" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 33] } + }, + "attentionBiasT": { + "shape": [1, 1, 35, 35], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.44600000000000006, "cosStep": 0.588, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [33, 320], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-wide-float16-t33-h2-d224-separate", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "shape": [33, 448], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.378, "cosStep": 0.5489999999999999, "scale": 0.4 } + }, + "keyT": { + "shape": [33, 448], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.395, "cosStep": 0.5619999999999999, "scale": 0.4 } + }, + "valueT": { + "shape": [33, 448], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.41200000000000003, "cosStep": 0.575, "scale": 0.4 } + }, + "biasT": { + "shape": [1344], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.42900000000000005, "cosStep": 0.588, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 35], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_37" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 33] } + } + }, + "outputs": { "outputT": { "shape": [33, 448], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-fragments-float32-t17-h1-d80-packed", + "attrs": { "num_heads": 1, "scale": 0 }, + "inputs": { + "queryT": { + "shape": [17, 1, 3, 80], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.157, "cosStep": 0.211, "scale": 0.4 } + }, + "tokenOffsetT": { + "shape": [3, 15], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/fragment_schedule_634ea297a37d" } } + }, + "cumulativeSequenceLengthT": { + "shape": [4], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 0, 5, 17] } + }, + "attentionBiasT": { + "shape": [1, 1, 15, 15], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.20800000000000002, "cosStep": 0.25, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [17, 80], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-fragments-float32-t33-h2-d112-separate", + "attrs": { "num_heads": 2, "scale": 0.071 }, + "inputs": { + "queryT": { + "shape": [33, 224], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.174, "cosStep": 0.237, "scale": 0.4 } + }, + "keyT": { + "shape": [33, 224], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.191, "cosStep": 0.25, "scale": 0.4 } + }, + "valueT": { + "shape": [33, 224], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.20800000000000002, "cosStep": 0.263, "scale": 0.4 } + }, + "biasT": { + "shape": [672], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.225, "cosStep": 0.276, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 35], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_37" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 33] } + } + }, + "outputs": { "outputT": { "shape": [33, 224], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-fragments-float32-t65-h3-d144-packed", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "shape": [65, 3, 3, 144], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.191, "cosStep": 0.263, "scale": 0.4 } + }, + "biasT": { + "shape": [1296], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.20800000000000002, "cosStep": 0.276, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 67], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/fragment_schedule_326ff2747fbb" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 65] } + }, + "attentionBiasT": { + "shape": [1, 1, 67, 67], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.259, "cosStep": 0.315, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [65, 432], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-fragments-float32-t129-h1-d176-separate", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "shape": [129, 176], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.20800000000000002, "cosStep": 0.289, "scale": 0.4 } + }, + "keyT": { + "shape": [129, 176], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.225, "cosStep": 0.302, "scale": 0.4 } + }, + "valueT": { + "shape": [129, 176], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.242, "cosStep": 0.315, "scale": 0.4 } + }, + "tokenOffsetT": { + "shape": [2, 131], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_39" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 129] } + }, + "attentionBiasT": { + "shape": [2, 1, 131, 131], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.29300000000000004, "cosStep": 0.354, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [129, 176], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-fragments-float32-t257-h2-d208-packed", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "shape": [257, 2, 3, 208], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.225, "cosStep": 0.315, "scale": 0.4 } + }, + "biasT": { + "shape": [1248], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.242, "cosStep": 0.32799999999999996, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 259], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_40" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 257] } + } + }, + "outputs": { "outputT": { "shape": [257, 416], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-fragments-float32-t17-h3-d240-separate", + "attrs": { "num_heads": 3, "scale": 0 }, + "inputs": { + "queryT": { + "shape": [17, 720], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.242, "cosStep": 0.34099999999999997, "scale": 0.4 } + }, + "keyT": { + "shape": [17, 720], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.259, "cosStep": 0.354, "scale": 0.4 } + }, + "valueT": { + "shape": [17, 720], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.276, "cosStep": 0.367, "scale": 0.4 } + }, + "biasT": { + "shape": [2160], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.29300000000000004, "cosStep": 0.38, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [3, 15], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/fragment_schedule_634ea297a37d" } } + }, + "cumulativeSequenceLengthT": { + "shape": [4], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 0, 5, 17] } + }, + "attentionBiasT": { + "shape": [3, 3, 15, 15], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.344, "cosStep": 0.419, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [17, 720], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-fragments-float32-t33-h1-d272-packed", + "attrs": { "num_heads": 1, "scale": 0.071 }, + "inputs": { + "queryT": { + "shape": [33, 1, 3, 272], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.259, "cosStep": 0.367, "scale": 0.4 } + }, + "tokenOffsetT": { + "shape": [2, 35], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_37" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 33] } + }, + "attentionBiasT": { + "shape": [1, 1, 35, 35], + "dtype": "float32", + "data": { + "kind": "fillFloat32", + "sinStep": 0.31000000000000005, + "cosStep": 0.40599999999999997, + "scale": 0.1 + } + } + }, + "outputs": { "outputT": { "shape": [33, 272], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-fragments-float32-t65-h2-d304-separate", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "shape": [65, 608], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.276, "cosStep": 0.393, "scale": 0.4 } + }, + "keyT": { + "shape": [65, 608], + "dtype": "float32", + "data": { + "kind": "fillFloat32", + "sinStep": 0.29300000000000004, + "cosStep": 0.40599999999999997, + "scale": 0.4 + } + }, + "valueT": { + "shape": [65, 608], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.31000000000000005, "cosStep": 0.419, "scale": 0.4 } + }, + "biasT": { + "shape": [1824], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.327, "cosStep": 0.432, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 67], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/fragment_schedule_326ff2747fbb" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 65] } + } + }, + "outputs": { "outputT": { "shape": [65, 608], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-fragments-float32-t129-h3-d336-packed", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "shape": [129, 3, 3, 336], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.29300000000000004, "cosStep": 0.419, "scale": 0.4 } + }, + "biasT": { + "shape": [3024], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.31000000000000005, "cosStep": 0.432, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 131], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_39" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 129] } + }, + "attentionBiasT": { + "shape": [1, 1, 131, 131], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.361, "cosStep": 0.471, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [129, 1008], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-fragments-float32-t257-h1-d368-separate", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "shape": [257, 368], + "dtype": "float32", + "data": { + "kind": "fillFloat32", + "sinStep": 0.31000000000000005, + "cosStep": 0.44499999999999995, + "scale": 0.4 + } + }, + "keyT": { + "shape": [257, 368], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.327, "cosStep": 0.45799999999999996, "scale": 0.4 } + }, + "valueT": { + "shape": [257, 368], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.344, "cosStep": 0.471, "scale": 0.4 } + }, + "tokenOffsetT": { + "shape": [2, 259], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_40" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 257] } + }, + "attentionBiasT": { + "shape": [2, 1, 259, 259], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.395, "cosStep": 0.51, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [257, 368], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-fragments-float32-t17-h2-d400-packed", + "attrs": { "num_heads": 2, "scale": 0 }, + "inputs": { + "queryT": { + "shape": [17, 2, 3, 400], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.327, "cosStep": 0.471, "scale": 0.4 } + }, + "biasT": { + "shape": [2400], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.344, "cosStep": 0.484, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [3, 15], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/fragment_schedule_634ea297a37d" } } + }, + "cumulativeSequenceLengthT": { + "shape": [4], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 0, 5, 17] } + } + }, + "outputs": { "outputT": { "shape": [17, 800], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-fragments-float32-t33-h3-d432-separate", + "attrs": { "num_heads": 3, "scale": 0.071 }, + "inputs": { + "queryT": { + "shape": [33, 1296], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.344, "cosStep": 0.497, "scale": 0.4 } + }, + "keyT": { + "shape": [33, 1296], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.361, "cosStep": 0.51, "scale": 0.4 } + }, + "valueT": { + "shape": [33, 1296], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.378, "cosStep": 0.523, "scale": 0.4 } + }, + "biasT": { + "shape": [3888], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.395, "cosStep": 0.536, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 35], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_37" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 33] } + }, + "attentionBiasT": { + "shape": [2, 3, 35, 35], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.44600000000000006, "cosStep": 0.575, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [33, 1296], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-fragments-float32-t65-h1-d464-packed", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "shape": [65, 1, 3, 464], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.361, "cosStep": 0.523, "scale": 0.4 } + }, + "tokenOffsetT": { + "shape": [2, 67], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/fragment_schedule_326ff2747fbb" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 65] } + }, + "attentionBiasT": { + "shape": [1, 1, 67, 67], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.41200000000000003, "cosStep": 0.5619999999999999, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [65, 464], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-fragments-float32-t129-h2-d496-separate", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "shape": [129, 992], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.378, "cosStep": 0.5489999999999999, "scale": 0.4 } + }, + "keyT": { + "shape": [129, 992], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.395, "cosStep": 0.5619999999999999, "scale": 0.4 } + }, + "valueT": { + "shape": [129, 992], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.41200000000000003, "cosStep": 0.575, "scale": 0.4 } + }, + "biasT": { + "shape": [2976], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.42900000000000005, "cosStep": 0.588, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 131], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_39" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 129] } + } + }, + "outputs": { "outputT": { "shape": [129, 992], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-fragments-float32-t129-h3-d528-packed", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "shape": [129, 3, 3, 528], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.395, "cosStep": 0.575, "scale": 0.4 } + }, + "biasT": { + "shape": [4752], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.41200000000000003, "cosStep": 0.588, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 131], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_39" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 129] } + }, + "attentionBiasT": { + "shape": [1, 1, 131, 131], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.4630000000000001, "cosStep": 0.627, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [129, 1584], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-fragments-float32-t513-h2-d176-separate", + "attrs": { "num_heads": 2, "scale": 0 }, + "inputs": { + "queryT": { + "shape": [513, 352], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.41200000000000003, "cosStep": 0.601, "scale": 0.4 } + }, + "keyT": { + "shape": [513, 352], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.42900000000000005, "cosStep": 0.614, "scale": 0.4 } + }, + "valueT": { + "shape": [513, 352], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.44600000000000006, "cosStep": 0.627, "scale": 0.4 } + }, + "tokenOffsetT": { + "shape": [3, 345], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/fragment_schedule_eb0c4a513a9e" } } + }, + "cumulativeSequenceLengthT": { + "shape": [4], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 0, 171, 513] } + }, + "attentionBiasT": { + "shape": [3, 2, 345, 345], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.497, "cosStep": 0.6659999999999999, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [513, 352], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-fragments-float32-t513-h2-d304-packed", + "attrs": { "num_heads": 2, "scale": 0.071 }, + "inputs": { + "queryT": { + "shape": [513, 2, 3, 304], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.42900000000000005, "cosStep": 0.627, "scale": 0.4 } + }, + "biasT": { + "shape": [1824], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.44600000000000006, "cosStep": 0.64, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 515], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_41" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 513] } + } + }, + "outputs": { "outputT": { "shape": [513, 608], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-fragments-float32-t7-h1-d144-separate", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "shape": [7, 144], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.44600000000000006, "cosStep": 0.653, "scale": 0.4 } + }, + "keyT": { + "shape": [7, 144], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.4630000000000001, "cosStep": 0.6659999999999999, "scale": 0.4 } + }, + "valueT": { + "shape": [7, 144], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.48, "cosStep": 0.6789999999999999, "scale": 0.4 } + }, + "biasT": { + "shape": [432], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.497, "cosStep": 0.692, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 9], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_34" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 7] } + }, + "attentionBiasT": { + "shape": [2, 1, 9, 9], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.548, "cosStep": 0.731, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [7, 144], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-fragments-float32-t7-h1-d272-packed", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "shape": [7, 1, 3, 272], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.4630000000000001, "cosStep": 0.6789999999999999, "scale": 0.4 } + }, + "tokenOffsetT": { + "shape": [2, 9], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_34" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 7] } + }, + "attentionBiasT": { + "shape": [1, 1, 9, 9], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.514, "cosStep": 0.718, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [7, 272], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-fragments-float32-t33-h2-d256-separate", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "shape": [33, 512], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.48, "cosStep": 0.705, "scale": 0.4 } + }, + "keyT": { + "shape": [33, 512], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.497, "cosStep": 0.718, "scale": 0.4 } + }, + "valueT": { + "shape": [33, 512], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.514, "cosStep": 0.731, "scale": 0.4 } + }, + "biasT": { + "shape": [1536], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.531, "cosStep": 0.744, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 35], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_37" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 33] } + } + }, + "outputs": { "outputT": { "shape": [33, 512], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-fragments-float16-t17-h1-d80-packed", + "attrs": { "num_heads": 1, "scale": 0 }, + "inputs": { + "queryT": { + "shape": [17, 1, 3, 80], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.157, "cosStep": 0.211, "scale": 0.4 } + }, + "tokenOffsetT": { + "shape": [3, 15], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/fragment_schedule_634ea297a37d" } } + }, + "cumulativeSequenceLengthT": { + "shape": [4], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 0, 5, 17] } + }, + "attentionBiasT": { + "shape": [1, 1, 15, 15], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.20800000000000002, "cosStep": 0.25, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [17, 80], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-fragments-float16-t33-h2-d112-separate", + "attrs": { "num_heads": 2, "scale": 0.071 }, + "inputs": { + "queryT": { + "shape": [33, 224], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.174, "cosStep": 0.237, "scale": 0.4 } + }, + "keyT": { + "shape": [33, 224], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.191, "cosStep": 0.25, "scale": 0.4 } + }, + "valueT": { + "shape": [33, 224], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.20800000000000002, "cosStep": 0.263, "scale": 0.4 } + }, + "biasT": { + "shape": [672], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.225, "cosStep": 0.276, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 35], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_37" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 33] } + } + }, + "outputs": { "outputT": { "shape": [33, 224], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-fragments-float16-t65-h3-d144-packed", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "shape": [65, 3, 3, 144], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.191, "cosStep": 0.263, "scale": 0.4 } + }, + "biasT": { + "shape": [1296], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.20800000000000002, "cosStep": 0.276, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 67], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/fragment_schedule_326ff2747fbb" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 65] } + }, + "attentionBiasT": { + "shape": [1, 1, 67, 67], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.259, "cosStep": 0.315, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [65, 432], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-fragments-float16-t129-h1-d176-separate", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "shape": [129, 176], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.20800000000000002, "cosStep": 0.289, "scale": 0.4 } + }, + "keyT": { + "shape": [129, 176], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.225, "cosStep": 0.302, "scale": 0.4 } + }, + "valueT": { + "shape": [129, 176], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.242, "cosStep": 0.315, "scale": 0.4 } + }, + "tokenOffsetT": { + "shape": [2, 131], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_39" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 129] } + }, + "attentionBiasT": { + "shape": [2, 1, 131, 131], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.29300000000000004, "cosStep": 0.354, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [129, 176], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-fragments-float16-t257-h2-d208-packed", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "shape": [257, 2, 3, 208], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.225, "cosStep": 0.315, "scale": 0.4 } + }, + "biasT": { + "shape": [1248], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.242, "cosStep": 0.32799999999999996, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 259], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_40" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 257] } + } + }, + "outputs": { "outputT": { "shape": [257, 416], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-fragments-float16-t17-h3-d240-separate", + "attrs": { "num_heads": 3, "scale": 0 }, + "inputs": { + "queryT": { + "shape": [17, 720], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.242, "cosStep": 0.34099999999999997, "scale": 0.4 } + }, + "keyT": { + "shape": [17, 720], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.259, "cosStep": 0.354, "scale": 0.4 } + }, + "valueT": { + "shape": [17, 720], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.276, "cosStep": 0.367, "scale": 0.4 } + }, + "biasT": { + "shape": [2160], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.29300000000000004, "cosStep": 0.38, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [3, 15], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/fragment_schedule_634ea297a37d" } } + }, + "cumulativeSequenceLengthT": { + "shape": [4], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 0, 5, 17] } + }, + "attentionBiasT": { + "shape": [3, 3, 15, 15], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.344, "cosStep": 0.419, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [17, 720], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-fragments-float16-t33-h1-d272-packed", + "attrs": { "num_heads": 1, "scale": 0.071 }, + "inputs": { + "queryT": { + "shape": [33, 1, 3, 272], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.259, "cosStep": 0.367, "scale": 0.4 } + }, + "tokenOffsetT": { + "shape": [2, 35], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_37" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 33] } + }, + "attentionBiasT": { + "shape": [1, 1, 35, 35], + "dtype": "float16", + "data": { + "kind": "fillFloat32", + "sinStep": 0.31000000000000005, + "cosStep": 0.40599999999999997, + "scale": 0.1 + } + } + }, + "outputs": { "outputT": { "shape": [33, 272], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-fragments-float16-t65-h2-d304-separate", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "shape": [65, 608], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.276, "cosStep": 0.393, "scale": 0.4 } + }, + "keyT": { + "shape": [65, 608], + "dtype": "float16", + "data": { + "kind": "fillFloat32", + "sinStep": 0.29300000000000004, + "cosStep": 0.40599999999999997, + "scale": 0.4 + } + }, + "valueT": { + "shape": [65, 608], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.31000000000000005, "cosStep": 0.419, "scale": 0.4 } + }, + "biasT": { + "shape": [1824], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.327, "cosStep": 0.432, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 67], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/fragment_schedule_326ff2747fbb" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 65] } + } + }, + "outputs": { "outputT": { "shape": [65, 608], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-fragments-float16-t129-h3-d336-packed", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "shape": [129, 3, 3, 336], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.29300000000000004, "cosStep": 0.419, "scale": 0.4 } + }, + "biasT": { + "shape": [3024], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.31000000000000005, "cosStep": 0.432, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 131], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_39" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 129] } + }, + "attentionBiasT": { + "shape": [1, 1, 131, 131], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.361, "cosStep": 0.471, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [129, 1008], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-fragments-float16-t257-h1-d368-separate", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "shape": [257, 368], + "dtype": "float16", + "data": { + "kind": "fillFloat32", + "sinStep": 0.31000000000000005, + "cosStep": 0.44499999999999995, + "scale": 0.4 + } + }, + "keyT": { + "shape": [257, 368], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.327, "cosStep": 0.45799999999999996, "scale": 0.4 } + }, + "valueT": { + "shape": [257, 368], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.344, "cosStep": 0.471, "scale": 0.4 } + }, + "tokenOffsetT": { + "shape": [2, 259], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_40" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 257] } + }, + "attentionBiasT": { + "shape": [2, 1, 259, 259], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.395, "cosStep": 0.51, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [257, 368], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-fragments-float16-t17-h2-d400-packed", + "attrs": { "num_heads": 2, "scale": 0 }, + "inputs": { + "queryT": { + "shape": [17, 2, 3, 400], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.327, "cosStep": 0.471, "scale": 0.4 } + }, + "biasT": { + "shape": [2400], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.344, "cosStep": 0.484, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [3, 15], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/fragment_schedule_634ea297a37d" } } + }, + "cumulativeSequenceLengthT": { + "shape": [4], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 0, 5, 17] } + } + }, + "outputs": { "outputT": { "shape": [17, 800], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-fragments-float16-t33-h3-d432-separate", + "attrs": { "num_heads": 3, "scale": 0.071 }, + "inputs": { + "queryT": { + "shape": [33, 1296], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.344, "cosStep": 0.497, "scale": 0.4 } + }, + "keyT": { + "shape": [33, 1296], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.361, "cosStep": 0.51, "scale": 0.4 } + }, + "valueT": { + "shape": [33, 1296], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.378, "cosStep": 0.523, "scale": 0.4 } + }, + "biasT": { + "shape": [3888], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.395, "cosStep": 0.536, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 35], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_37" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 33] } + }, + "attentionBiasT": { + "shape": [2, 3, 35, 35], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.44600000000000006, "cosStep": 0.575, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [33, 1296], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-fragments-float16-t65-h1-d464-packed", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "shape": [65, 1, 3, 464], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.361, "cosStep": 0.523, "scale": 0.4 } + }, + "tokenOffsetT": { + "shape": [2, 67], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/fragment_schedule_326ff2747fbb" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 65] } + }, + "attentionBiasT": { + "shape": [1, 1, 67, 67], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.41200000000000003, "cosStep": 0.5619999999999999, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [65, 464], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-fragments-float16-t129-h2-d496-separate", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "shape": [129, 992], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.378, "cosStep": 0.5489999999999999, "scale": 0.4 } + }, + "keyT": { + "shape": [129, 992], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.395, "cosStep": 0.5619999999999999, "scale": 0.4 } + }, + "valueT": { + "shape": [129, 992], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.41200000000000003, "cosStep": 0.575, "scale": 0.4 } + }, + "biasT": { + "shape": [2976], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.42900000000000005, "cosStep": 0.588, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 131], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_39" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 129] } + } + }, + "outputs": { "outputT": { "shape": [129, 992], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-fragments-float16-t129-h3-d528-packed", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "shape": [129, 3, 3, 528], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.395, "cosStep": 0.575, "scale": 0.4 } + }, + "biasT": { + "shape": [4752], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.41200000000000003, "cosStep": 0.588, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 131], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_39" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 129] } + }, + "attentionBiasT": { + "shape": [1, 1, 131, 131], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.4630000000000001, "cosStep": 0.627, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [129, 1584], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-fragments-float16-t513-h2-d176-separate", + "attrs": { "num_heads": 2, "scale": 0 }, + "inputs": { + "queryT": { + "shape": [513, 352], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.41200000000000003, "cosStep": 0.601, "scale": 0.4 } + }, + "keyT": { + "shape": [513, 352], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.42900000000000005, "cosStep": 0.614, "scale": 0.4 } + }, + "valueT": { + "shape": [513, 352], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.44600000000000006, "cosStep": 0.627, "scale": 0.4 } + }, + "tokenOffsetT": { + "shape": [3, 345], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/fragment_schedule_eb0c4a513a9e" } } + }, + "cumulativeSequenceLengthT": { + "shape": [4], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 0, 171, 513] } + }, + "attentionBiasT": { + "shape": [3, 2, 345, 345], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.497, "cosStep": 0.6659999999999999, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [513, 352], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-fragments-float16-t513-h2-d304-packed", + "attrs": { "num_heads": 2, "scale": 0.071 }, + "inputs": { + "queryT": { + "shape": [513, 2, 3, 304], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.42900000000000005, "cosStep": 0.627, "scale": 0.4 } + }, + "biasT": { + "shape": [1824], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.44600000000000006, "cosStep": 0.64, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 515], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_41" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 513] } + } + }, + "outputs": { "outputT": { "shape": [513, 608], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-fragments-float16-t7-h1-d144-separate", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "shape": [7, 144], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.44600000000000006, "cosStep": 0.653, "scale": 0.4 } + }, + "keyT": { + "shape": [7, 144], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.4630000000000001, "cosStep": 0.6659999999999999, "scale": 0.4 } + }, + "valueT": { + "shape": [7, 144], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.48, "cosStep": 0.6789999999999999, "scale": 0.4 } + }, + "biasT": { + "shape": [432], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.497, "cosStep": 0.692, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 9], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_34" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 7] } + }, + "attentionBiasT": { + "shape": [2, 1, 9, 9], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.548, "cosStep": 0.731, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [7, 144], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-fragments-float16-t7-h1-d272-packed", + "attrs": { "num_heads": 1 }, + "inputs": { + "queryT": { + "shape": [7, 1, 3, 272], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.4630000000000001, "cosStep": 0.6789999999999999, "scale": 0.4 } + }, + "tokenOffsetT": { + "shape": [2, 9], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_34" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 7] } + }, + "attentionBiasT": { + "shape": [1, 1, 9, 9], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.514, "cosStep": 0.718, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [7, 272], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-fragments-float16-t33-h2-d256-separate", + "attrs": { "num_heads": 2 }, + "inputs": { + "queryT": { + "shape": [33, 512], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.48, "cosStep": 0.705, "scale": 0.4 } + }, + "keyT": { + "shape": [33, 512], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.497, "cosStep": 0.718, "scale": 0.4 } + }, + "valueT": { + "shape": [33, 512], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.514, "cosStep": 0.731, "scale": 0.4 } + }, + "biasT": { + "shape": [1536], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.531, "cosStep": 0.744, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 35], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_37" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 33] } + } + }, + "outputs": { "outputT": { "shape": [33, 512], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-fragments-float32-t17-h1-d288-packed", + "attrs": { "num_heads": 1, "scale": 0 }, + "inputs": { + "queryT": { + "shape": [17, 1, 3, 288], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.157, "cosStep": 0.211, "scale": 0.4 } + }, + "tokenOffsetT": { + "shape": [3, 15], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/fragment_schedule_634ea297a37d" } } + }, + "cumulativeSequenceLengthT": { + "shape": [4], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 0, 5, 17] } + }, + "attentionBiasT": { + "shape": [1, 1, 15, 15], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.20800000000000002, "cosStep": 0.25, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [17, 288], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-fragments-float32-t65-h2-d352-separate", + "attrs": { "num_heads": 2, "scale": 0.071 }, + "inputs": { + "queryT": { + "shape": [65, 704], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.174, "cosStep": 0.237, "scale": 0.4 } + }, + "keyT": { + "shape": [65, 704], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.191, "cosStep": 0.25, "scale": 0.4 } + }, + "valueT": { + "shape": [65, 704], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.20800000000000002, "cosStep": 0.263, "scale": 0.4 } + }, + "biasT": { + "shape": [2112], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.225, "cosStep": 0.276, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 67], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/fragment_schedule_326ff2747fbb" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 65] } + } + }, + "outputs": { "outputT": { "shape": [65, 704], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-fragments-float32-t129-h3-d480-packed", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "shape": [129, 3, 3, 480], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.191, "cosStep": 0.263, "scale": 0.4 } + }, + "biasT": { + "shape": [4320], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.20800000000000002, "cosStep": 0.276, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 131], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_39" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 129] } + }, + "attentionBiasT": { + "shape": [1, 1, 131, 131], + "dtype": "float32", + "data": { "kind": "fillFloat32", "sinStep": 0.259, "cosStep": 0.315, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [129, 1440], "dtype": "float32" } }, + "tolerance": 0.00002, + "relTolerance": 0.00002 + }, + { + "name": "pmha-fragments-float16-t17-h1-d288-packed", + "attrs": { "num_heads": 1, "scale": 0 }, + "inputs": { + "queryT": { + "shape": [17, 1, 3, 288], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.157, "cosStep": 0.211, "scale": 0.4 } + }, + "tokenOffsetT": { + "shape": [3, 15], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/fragment_schedule_634ea297a37d" } } + }, + "cumulativeSequenceLengthT": { + "shape": [4], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 0, 5, 17] } + }, + "attentionBiasT": { + "shape": [1, 1, 15, 15], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.20800000000000002, "cosStep": 0.25, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [17, 288], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-fragments-float16-t65-h2-d352-separate", + "attrs": { "num_heads": 2, "scale": 0.071 }, + "inputs": { + "queryT": { + "shape": [65, 704], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.174, "cosStep": 0.237, "scale": 0.4 } + }, + "keyT": { + "shape": [65, 704], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.191, "cosStep": 0.25, "scale": 0.4 } + }, + "valueT": { + "shape": [65, 704], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.20800000000000002, "cosStep": 0.263, "scale": 0.4 } + }, + "biasT": { + "shape": [2112], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.225, "cosStep": 0.276, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 67], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/fragment_schedule_326ff2747fbb" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 65] } + } + }, + "outputs": { "outputT": { "shape": [65, 704], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + }, + { + "name": "pmha-fragments-float16-t129-h3-d480-packed", + "attrs": { "num_heads": 3 }, + "inputs": { + "queryT": { + "shape": [129, 3, 3, 480], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.191, "cosStep": 0.263, "scale": 0.4 } + }, + "biasT": { + "shape": [4320], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.20800000000000002, "cosStep": 0.276, "scale": 0.1 } + }, + "tokenOffsetT": { + "shape": [2, 131], + "dtype": "int32", + "data": { "kind": "values", "values": { "$ref": "#/fixtureArrays/wide_schedule_39" } } + }, + "cumulativeSequenceLengthT": { + "shape": [3], + "dtype": "int32", + "data": { "kind": "values", "values": [0, 1, 129] } + }, + "attentionBiasT": { + "shape": [1, 1, 131, 131], + "dtype": "float16", + "data": { "kind": "fillFloat32", "sinStep": 0.259, "cosStep": 0.315, "scale": 0.1 } + } + }, + "outputs": { "outputT": { "shape": [129, 1440], "dtype": "float16" } }, + "tolerance": 0.001, + "relTolerance": 0.005 + } + ] +}