Gitiles
Code Review Sign In
review.mlplatform.org / tosa / reference_model / 1a083b598d26bb968ef2d6e3403bf808e26766b1 / . / verif / conformance / tosa_ext_profile_ops_info.json
blob: ac25269814248ee24a47287048454b964d6ed0ae [file] [log] [blame]
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]1{
2 "abs": {
3 "group": "ew_unary",
4 "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ],
5 "generation": {
6 "tosa-bi": {
7 "supports_all": [
8 "tosa-bi"
9 ],
10 "selector": "tosa-bi",
11 "generator_args": [
12 [
13 "--target-dtype",
14 "int32",
15 "--tensor-dim-range",
16 "16,64",
17 "--target-rank",
18 "1",
19 "--target-rank",
20 "2",
21 "--target-rank",
22 "3"
23 ],
24 [
25 "--target-dtype",
26 "int32",
27 "--tensor-dim-range",
28 "1,16",
29 "--target-rank",
30 "4",
31 "--target-rank",
32 "5"
33 ],
34 [
35 "--target-dtype",
36 "int32",
37 "--target-shape",
38 "1,2,65533,1",
39 "--target-shape",
40 "1,65539,1,1,1"
41 ]
42 ]
43 },
44 "8k_level": {
45 "supports_all": [
46 "tosa-bi"
47 ],
48 "selector": "tosa-bi",
49 "no_negative_tests": "true",
50 "generator_args": [
51 [
52 "--target-dtype",
53 "int32",
54 "--tensor-dim-range",
55 "1,10",
56 "--target-rank",
57 "6"
58 ]
59 ]
60 },
61 "tosa-mi": {
62 "supports_all": [
63 "tosa-mi"
64 ],
65 "selector": "tosa-mi",
66 "generator_args": [
67 [
68 "--target-dtype",
69 "fp32",
70 "--target-dtype",
71 "fp16",
72 "--fp-values-range",
73 "-max,max",
74 "--tensor-dim-range",
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]75 "31,66",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]76 "--target-rank",
77 "0",
78 "--target-rank",
79 "1",
80 "--target-rank",
81 "2",
82 "--target-rank",
83 "3"
84 ],
85 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]86 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]87 "--target-dtype",
88 "fp32",
89 "--target-dtype",
90 "fp16",
91 "--fp-values-range",
92 "-max,max",
93 "--tensor-dim-range",
94 "1,16",
95 "--target-rank",
96 "4",
97 "--target-rank",
Jeremy Johnson074b98c2024-05-30 10:38:35 +0100[diff] [blame]98 "5",
99 "--target-rank",
100 "6"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]101 ]
102 ]
103 },
104 "tosa-ext-bf16": {
105 "supports_all": [
106 "tosa-ext-bf16"
107 ],
108 "selector": "tosa-mi",
109 "no_negative_tests": "true",
110 "generator_args": [
111 [
112 "--target-dtype",
113 "bf16",
114 "--fp-values-range",
115 "-max,max",
116 "--tensor-dim-range",
117 "16,64",
118 "--target-rank",
119 "0",
120 "--target-rank",
121 "1",
122 "--target-rank",
123 "2",
124 "--target-rank",
125 "3"
126 ],
127 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]128 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]129 "--target-dtype",
130 "bf16",
131 "--fp-values-range",
132 "-max,max",
133 "--tensor-dim-range",
134 "1,16",
135 "--target-rank",
136 "4",
137 "--target-rank",
Jeremy Johnson074b98c2024-05-30 10:38:35 +0100[diff] [blame]138 "5",
139 "--target-rank",
140 "6"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]141 ],
142 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]143 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]144 "--target-dtype",
145 "bf16",
146 "--fp-values-range",
147 "-max,max",
148 "--target-shape",
149 "1,65531,2,1",
150 "--target-shape",
151 "65538,1,1,1"
152 ]
153 ]
154 }
155 },
156 "selection": {
157 "tosa-bi": {
158 "params": {},
159 "permutes": [
160 "shape",
161 "type"
162 ]
163 },
164 "tosa-mi": {
165 "generator_select": "true"
166 }
167 }
168 },
169 "add": {
170 "group": "ew_binary",
171 "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ],
172 "generation": {
173 "tosa-bi-mi": {
174 "supports_any": [
175 "tosa-bi",
176 "tosa-mi"
177 ],
178 "generator_args": [
179 [
180 "--target-dtype",
181 "int32",
182 "--tensor-dim-range",
183 "16,64",
184 "--target-rank",
185 "1",
186 "--target-rank",
187 "2",
188 "--target-rank",
189 "3"
190 ],
191 [
192 "--target-dtype",
193 "int32",
194 "--tensor-dim-range",
195 "1,16",
196 "--target-rank",
197 "4",
198 "--target-rank",
199 "5"
200 ],
201 [
202 "--target-dtype",
203 "int32",
204 "--target-shape",
205 "1,2,65533,1",
206 "--target-shape",
207 "1,65539,1,1,1"
208 ]
209 ]
210 },
211 "8k_level": {
212 "supports_all": [
213 "tosa-bi"
214 ],
215 "no_negative_tests": "true",
216 "generator_args": [
217 [
218 "--target-dtype",
219 "int32",
220 "--tensor-dim-range",
221 "1,10",
222 "--target-rank",
223 "6"
224 ]
225 ]
226 },
227 "tosa-mi": {
228 "supports_all": [
229 "tosa-mi"
230 ],
231 "selector": "tosa-mi",
232 "generator_args": [
233 [
234 "--target-dtype",
235 "fp32",
236 "--target-dtype",
237 "fp16",
238 "--fp-values-range",
239 "-max,max",
240 "--tensor-dim-range",
241 "16,64",
242 "--target-rank",
243 "0",
244 "--target-rank",
245 "1",
246 "--target-rank",
247 "2",
248 "--target-rank",
249 "3"
250 ],
251 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]252 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]253 "--target-dtype",
254 "fp32",
255 "--fp-values-range",
256 "-max,max",
257 "--tensor-dim-range",
258 "1,16",
259 "--target-rank",
260 "4",
261 "--target-rank",
262 "5"
263 ],
264 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]265 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]266 "--target-dtype",
267 "fp16",
268 "--fp-values-range",
269 "-max,max",
270 "--target-shape",
271 "2,1,1,65532",
272 "--target-shape",
273 "65542,2,1,1,1"
274 ]
275 ]
276 },
277 "tosa-ext-bf16": {
278 "supports_all": [
279 "tosa-ext-bf16"
280 ],
281 "selector": "tosa-mi",
282 "no_negative_tests": "true",
283 "generator_args": [
284 [
285 "--target-dtype",
286 "bf16",
287 "--fp-values-range",
288 "-max,max",
289 "--tensor-dim-range",
290 "16,64",
291 "--target-rank",
292 "0",
293 "--target-rank",
294 "1",
295 "--target-rank",
296 "2",
297 "--target-rank",
298 "3"
299 ]
300 ]
301 }
302 },
303 "selection": {
304 "default": {
305 "params": {},
306 "permutes": [
307 "shape",
308 "type"
309 ]
310 },
311 "tosa-mi": {
312 "generator_select": "true"
313 }
314 }
315 },
316 "add_shape": {
317 "group": "shape",
318 "generation": {
319 "tosa-bi-mi": {
320 "supports_any": [
321 "tosa-bi",
322 "tosa-mi"
323 ],
324 "generator_args": [
325 [
326 "--target-dtype",
327 "shape",
328 "--tensor-dim-range",
329 "1,16",
330 "--target-rank",
331 "1"
332 ]
333 ]
334 }
335 },
336 "selection": {
337 "default": {
338 "params": {},
339 "permutes": [
340 "shape"
341 ]
342 }
343 }
344 },
345 "argmax": {
346 "group": "tensor",
347 "support_for": [
348 "lazy_data_gen"
349 ],
350 "generation": {
351 "tosa-bi": {
352 "supports_all": [
353 "tosa-bi"
354 ],
355 "generator_args": [
356 [
357 "--target-dtype",
358 "int8",
359 "--tensor-dim-range",
360 "32,64",
361 "--target-rank",
362 "1",
363 "--target-rank",
364 "2",
365 "--target-rank",
366 "3",
367 "--num-rand-permutations",
368 "2"
369 ],
370 [
371 "--target-dtype",
372 "int8",
373 "--tensor-dim-range",
374 "1,32",
375 "--target-rank",
376 "4",
377 "--num-rand-permutations",
378 "2"
379 ]
380 ]
381 },
382 "8k_level": {
383 "supports_all": [
384 "tosa-bi"
385 ],
386 "from_version": "v0.70.0",
387 "no_negative_tests": "true",
388 "generator_args": [
389 [
390 "--target-dtype",
391 "int8",
392 "--tensor-dim-range",
393 "1,10",
394 "--target-rank",
395 "5",
396 "--target-rank",
397 "6",
398 "--num-rand-permutations",
399 "2"
400 ]
401 ]
402 },
403 "tosa-mi": {
404 "supports_all": [
405 "tosa-mi"
406 ],
407 "generator_args": [
408 [
409 "--target-dtype",
410 "fp32",
411 "--target-dtype",
412 "fp16",
413 "--fp-values-range",
414 "32,64",
415 "--target-rank",
416 "1",
417 "--target-rank",
418 "2",
419 "--target-rank",
420 "3",
421 "--num-rand-permutations",
422 "2"
423 ],
424 [
425 "--target-dtype",
426 "fp32",
427 "--target-dtype",
428 "fp16",
429 "--fp-values-range",
430 "-max,max",
431 "--tensor-dim-range",
432 "1,32",
433 "--target-rank",
434 "4",
435 "--num-rand-permutations",
436 "2"
437 ],
438 [
439 "--target-dtype",
440 "fp32",
441 "--fp-values-range",
442 "-max,max",
443 "--target-shape",
444 "1,3,65535,1",
445 "--target-shape",
446 "1,65537,1,2",
447 "--num-rand-permutations",
448 "2"
449 ]
450 ]
451 },
452 "tosa-ext-bf16": {
453 "supports_all": [
454 "tosa-ext-bf16"
455 ],
456 "no_negative_tests": "true",
457 "generator_args": [
458 [
459 "--target-dtype",
460 "bf16",
461 "--fp-values-range",
462 "-max,max",
463 "--tensor-dim-range",
464 "32,64",
465 "--target-rank",
466 "1",
467 "--target-rank",
468 "2",
469 "--target-rank",
470 "3",
471 "--num-rand-permutations",
472 "2"
473 ],
474 [
475 "--target-dtype",
476 "bf16",
477 "--fp-values-range",
478 "-max,max",
479 "--tensor-dim-range",
480 "1,32",
481 "--target-rank",
482 "4",
483 "--num-rand-permutations",
484 "2"
485 ]
486 ]
487 },
488 "tosa-ext-fp8e4m3": {
489 "supports_all": [
490 "tosa-ext-fp8e4m3"
491 ],
492 "no_negative_tests": "true",
493 "from_version": "v0.100.0",
494 "generator_args": [
495 [
496 "--target-dtype",
497 "fp8e4m3",
498 "--fp-values-range",
499 "-max,max",
500 "--tensor-dim-range",
501 "32,64",
502 "--target-rank",
503 "1",
504 "--target-rank",
505 "2",
506 "--target-rank",
507 "3",
508 "--num-rand-permutations",
509 "2"
510 ]
511 ]
512 },
513 "tosa-ext-fp8e5m2": {
514 "supports_all": [
515 "tosa-ext-fp8e5m2"
516 ],
517 "no_negative_tests": "true",
518 "from_version": "v0.100.0",
519 "generator_args": [
520 [
521 "--target-dtype",
522 "fp8e5m2",
523 "--fp-values-range",
524 "-max,max",
525 "--tensor-dim-range",
526 "32,64",
527 "--target-rank",
528 "1",
529 "--target-rank",
530 "2",
531 "--target-rank",
532 "3",
533 "--num-rand-permutations",
534 "2"
535 ]
536 ]
537 },
538 "tosa-ext-int16": {
539 "supports_all": [
540 "tosa-ext-int16"
541 ],
542 "no_negative_tests": "true",
543 "from_version": "v0.100.0",
544 "generator_args": [
545 [
546 "--target-dtype",
547 "int16",
548 "--tensor-dim-range",
549 "32,64",
550 "--target-rank",
551 "1",
552 "--target-rank",
553 "2",
554 "--target-rank",
555 "3",
556 "--num-rand-permutations",
557 "2"
558 ],
559 [
560 "--target-dtype",
561 "int16",
562 "--tensor-dim-range",
563 "1,32",
564 "--target-rank",
565 "4",
566 "--num-rand-permutations",
567 "2"
568 ]
569 ]
570 },
571 "8k_level_ext-int16": {
572 "supports_all": [
573 "tosa-ext-int16"
574 ],
575 "from_version": "v0.70.0",
576 "no_negative_tests": "true",
577 "generator_args": [
578 [
579 "--target-dtype",
580 "int16",
581 "--tensor-dim-range",
582 "1,10",
583 "--target-rank",
584 "5",
585 "--target-rank",
586 "6",
587 "--num-rand-permutations",
588 "2"
589 ]
590 ]
591 }
592 },
593 "selection": {
594 "default": {
595 "params": {},
596 "permutes": [
597 "shape",
598 "type"
599 ]
600 }
601 }
602 },
603 "arithmetic_right_shift": {
604 "group": "ew_binary",
605 "generation": {
606 "tosa-bi": {
607 "supports_all": [
608 "tosa-bi"
609 ],
610 "generator_args": [
611 [
612 "--target-dtype",
613 "int8",
614 "--target-dtype",
615 "int16",
616 "--target-dtype",
617 "int32",
618 "--tensor-dim-range",
619 "16,64",
620 "--target-rank",
621 "1",
622 "--target-rank",
623 "2",
624 "--target-rank",
625 "3"
626 ],
627 [
628 "--target-dtype",
629 "int8",
630 "--target-dtype",
631 "int16",
632 "--target-dtype",
633 "int32",
634 "--tensor-dim-range",
635 "1,16",
636 "--target-rank",
637 "4",
638 "--target-rank",
639 "5"
640 ],
641 [
642 "--target-dtype",
643 "int8",
644 "--target-shape",
645 "1,2,65533,1",
646 "--target-shape",
647 "1,65539,1,1,1"
648 ]
649 ]
650 },
651 "8k_level": {
652 "supports_all": [
653 "tosa-bi"
654 ],
655 "no_negative_tests": "true",
656 "generator_args": [
657 [
658 "--target-dtype",
659 "int8",
660 "--target-dtype",
661 "int16",
662 "--target-dtype",
663 "int32",
664 "--tensor-dim-range",
665 "1,10",
666 "--target-rank",
667 "6"
668 ]
669 ]
670 }
671 },
672 "selection": {
673 "default": {
674 "params": {},
675 "permutes": [
676 "shape",
677 "type",
678 "rounding"
679 ]
680 }
681 }
682 },
683 "avg_pool2d": {
684 "group": "tensor",
685 "support_for": [
686 "lazy_data_gen"
687 ],
688 "generation": {
689 "tosa-bi": {
690 "supports_all": [
691 "tosa-bi"
692 ],
693 "selector": "tosa-bi",
694 "generator_args": [
695 [
696 "--target-dtype",
697 "int8",
698 "--target-shape",
699 "1,49,42,28",
700 "--target-shape",
701 "1,11,44,3",
702 "--allow-pooling-and-conv-oversizes"
703 ],
704 [
705 "--target-dtype",
706 "int8",
707 "--target-shape",
708 "1,65535,5,1",
709 "--target-shape",
710 "1,3,65537,1",
711 "--allow-pooling-and-conv-oversizes"
712 ]
713 ]
714 },
715 "8k_level": {
716 "supports_all": [
717 "tosa-bi"
718 ],
719 "no_negative_tests": "true",
720 "selector": "8k_level",
721 "generator_args": [
722 [
723 "--target-dtype",
724 "int8",
725 "--tensor-dim-range",
726 "3,10",
727 "--level-8k-sizes"
728 ]
729 ]
730 },
731 "tosa-mi": {
732 "supports_all": [
733 "tosa-mi"
734 ],
735 "selector": "tosa-mi",
736 "generator_args": [
737 [
738 "--target-dtype",
739 "fp32",
740 "--target-dtype",
741 "fp16",
742 "--fp-values-range",
743 "-max,max",
744 "--target-shape",
745 "4,11,7,39",
746 "--target-shape",
747 "5,21,36,17",
748 "--allow-pooling-and-conv-oversizes"
749 ],
750 [
751 "--target-dtype",
752 "fp32",
753 "--fp-values-range",
754 "-max,max",
755 "--target-shape",
756 "1,2,65533,2",
757 "--target-shape",
758 "1,65539,2,1",
759 "--allow-pooling-and-conv-oversizes"
760 ]
761 ]
762 },
763 "tosa-ext-bf16": {
764 "supports_all": [
765 "tosa-ext-bf16"
766 ],
767 "selector": "tosa-mi",
768 "no_negative_tests": "true",
769 "generator_args": [
770 [
771 "--target-dtype",
772 "bf16",
773 "--fp-values-range",
774 "-max,max",
775 "--target-shape",
776 "4,11,7,39",
777 "--target-shape",
778 "5,21,36,17",
779 "--allow-pooling-and-conv-oversizes"
780 ]
781 ]
782 },
783 "tosa-ext-fp8e4m3": {
784 "supports_all": [
785 "tosa-ext-fp8e4m3"
786 ],
787 "selector": "tosa-mi",
788 "no_negative_tests": "true",
789 "from_version": "v0.100.0",
790 "generator_args": [
791 [
792 "--target-dtype",
793 "fp8e4m3",
794 "--fp-values-range",
795 "-max,max",
796 "--tensor-dim-range",
797 "32,64",
798 "--num-rand-permutations",
799 "2"
800 ]
801 ]
802 },
803 "tosa-ext-fp8e5m2": {
804 "supports_all": [
805 "tosa-ext-fp8e5m2"
806 ],
807 "selector": "tosa-mi",
808 "no_negative_tests": "true",
809 "from_version": "v0.100.0",
810 "generator_args": [
811 [
812 "--target-dtype",
813 "fp8e5m2",
814 "--fp-values-range",
815 "-max,max",
816 "--tensor-dim-range",
817 "32,64",
818 "--num-rand-permutations",
819 "2"
820 ]
821 ]
822 },
823 "tosa-ext-int16": {
824 "supports_all": [
825 "tosa-ext-int16"
826 ],
827 "selector": "tosa-bi",
828 "no_negative_tests": "true",
829 "generator_args": [
830 [
831 "--target-dtype",
832 "int16",
833 "--target-shape",
834 "1,49,42,28",
835 "--target-shape",
836 "1,11,44,3",
837 "--allow-pooling-and-conv-oversizes"
838 ]
839 ]
840 },
841 "8k_level_ext_int16": {
842 "supports_all": [
843 "tosa-ext-int16"
844 ],
845 "no_negative_tests": "true",
846 "selector": "8k_level",
847 "generator_args": [
848 [
849 "--target-dtype",
850 "int16",
851 "--tensor-dim-range",
852 "3,10",
853 "--level-8k-sizes"
854 ]
855 ]
856 }
857 },
858 "selection": {
859 "tosa-bi": {
860 "params": {
861 "shape": [],
862 "type": [
863 "i8",
864 "i16"
865 ],
866 "kernel": [],
867 "stride": [],
868 "pad": []
869 },
870 "permutes": [
871 "shape",
872 "type",
873 "pad"
874 ]
875 },
876 "tosa-mi": {
877 "params": {},
878 "permutes": [
879 "shape",
880 "type"
881 ],
882 "sparsity": {
883 "pad": 17
884 }
885 },
886 "8k_level": {
887 "all": "true"
888 }
889 }
890 },
891 "bitwise_and": {
892 "group": "ew_binary",
893 "generation": {
894 "standard": {
895 "supports_all": [
896 "tosa-bi"
897 ],
898 "generator_args": [
899 [
900 "--target-dtype",
901 "int8",
902 "--target-dtype",
903 "int16",
904 "--target-dtype",
905 "int32",
906 "--tensor-dim-range",
907 "1,64",
908 "--target-rank",
909 "1",
910 "--target-rank",
911 "2",
912 "--target-rank",
913 "3"
914 ],
915 [
916 "--target-dtype",
917 "int8",
918 "--tensor-dim-range",
919 "1,16",
920 "--target-rank",
921 "4",
922 "--target-rank",
923 "5"
924 ],
925 [
926 "--target-dtype",
927 "int8",
928 "--target-shape",
929 "1,2,65536,1",
930 "--target-shape",
931 "1,65539,1,1,1"
932 ]
933 ]
934 },
935 "8k_level": {
936 "supports_all": [
937 "tosa-bi"
938 ],
939 "no_negative_tests": "true",
940 "generator_args": [
941 [
942 "--target-dtype",
943 "int8",
944 "--target-dtype",
945 "int16",
946 "--tensor-dim-range",
947 "1,10",
948 "--target-rank",
949 "6"
950 ]
951 ]
952 }
953 },
954 "selection": {
955 "default": {
956 "params": {},
957 "permutes": [
958 "shape",
959 "type"
960 ]
961 }
962 }
963 },
964 "bitwise_not": {
965 "group": "ew_unary",
966 "generation": {
967 "standard": {
968 "supports_all": [
969 "tosa-bi"
970 ],
971 "generator_args": [
972 [
973 "--target-dtype",
974 "int8",
975 "--target-dtype",
976 "int16",
977 "--target-dtype",
978 "int32",
979 "--tensor-dim-range",
980 "2,64",
981 "--target-rank",
982 "1",
983 "--target-rank",
984 "2",
985 "--target-rank",
986 "3"
987 ],
988 [
989 "--target-dtype",
990 "int8",
991 "--tensor-dim-range",
992 "1,16",
993 "--target-rank",
994 "4",
995 "--target-rank",
996 "5"
997 ],
998 [
999 "--target-dtype",
1000 "int8",
1001 "--target-shape",
1002 "1,2,65537,1",
1003 "--target-shape",
1004 "1,65538,1,1,1"
1005 ]
1006 ]
1007 },
1008 "8k_level": {
1009 "supports_all": [
1010 "tosa-bi"
1011 ],
1012 "no_negative_tests": "true",
1013 "generator_args": [
1014 [
1015 "--target-dtype",
1016 "int16",
1017 "--target-dtype",
1018 "int32",
1019 "--tensor-dim-range",
1020 "1,16",
1021 "--target-rank",
1022 "4",
1023 "--target-rank",
1024 "5"
1025 ],
1026 [
1027 "--target-dtype",
1028 "int8",
1029 "--target-dtype",
1030 "int16",
1031 "--target-dtype",
1032 "int32",
1033 "--tensor-dim-range",
1034 "1,10",
1035 "--target-rank",
1036 "6"
1037 ]
1038 ]
1039 }
1040 },
1041 "selection": {
1042 "default": {
1043 "params": {},
1044 "permutes": [
1045 "shape",
1046 "type"
1047 ]
1048 }
1049 }
1050 },
1051 "bitwise_or": {
1052 "group": "ew_binary",
1053 "generation": {
1054 "standard": {
1055 "supports_all": [
1056 "tosa-bi"
1057 ],
1058 "selector": "tosa-bi",
1059 "generator_args": [
1060 [
1061 "--target-dtype",
1062 "int8",
1063 "--target-dtype",
1064 "int16",
1065 "--target-dtype",
1066 "int32",
1067 "--tensor-dim-range",
1068 "3,64",
1069 "--target-rank",
1070 "1",
1071 "--target-rank",
1072 "2",
1073 "--target-rank",
1074 "3"
1075 ],
1076 [
1077 "--target-dtype",
1078 "int8",
1079 "--tensor-dim-range",
1080 "1,16",
1081 "--target-rank",
1082 "4",
1083 "--target-rank",
1084 "5"
1085 ],
1086 [
1087 "--target-dtype",
1088 "int8",
1089 "--target-shape",
1090 "1,2,65538,1",
1091 "--target-shape",
1092 "1,65537,1,1,1"
1093 ]
1094 ]
1095 },
1096 "8k_level": {
1097 "supports_all": [
1098 "tosa-bi"
1099 ],
1100 "no_negative_tests": "true",
1101 "generator_args": [
1102 [
1103 "--target-dtype",
1104 "int8",
1105 "--target-dtype",
1106 "int16",
1107 "--tensor-dim-range",
1108 "1,10",
1109 "--target-rank",
1110 "6"
1111 ]
1112 ]
1113 }
1114 },
1115 "selection": {
1116 "default": {
1117 "params": {},
1118 "permutes": [
1119 "shape",
1120 "type"
1121 ]
1122 }
1123 }
1124 },
1125 "bitwise_xor": {
1126 "group": "ew_binary",
1127 "generation": {
1128 "standard": {
1129 "supports_all": [
1130 "tosa-bi"
1131 ],
1132 "generator_args": [
1133 [
1134 "--target-dtype",
1135 "int8",
1136 "--target-dtype",
1137 "int16",
1138 "--target-dtype",
1139 "int32",
1140 "--tensor-dim-range",
1141 "4,64",
1142 "--target-rank",
1143 "1",
1144 "--target-rank",
1145 "2",
1146 "--target-rank",
1147 "3"
1148 ],
1149 [
1150 "--target-dtype",
1151 "int8",
1152 "--tensor-dim-range",
1153 "1,16",
1154 "--target-rank",
1155 "4",
1156 "--target-rank",
1157 "5"
1158 ],
1159 [
1160 "--target-dtype",
1161 "int8",
1162 "--target-shape",
1163 "1,2,65536,1",
1164 "--target-shape",
1165 "1,65536,1,1,1"
1166 ]
1167 ]
1168 },
1169 "8k_level": {
1170 "supports_all": [
1171 "tosa-bi"
1172 ],
1173 "no_negative_tests": "true",
1174 "generator_args": [
1175 [
1176 "--target-dtype",
1177 "int8",
1178 "--target-dtype",
1179 "int16",
1180 "--tensor-dim-range",
1181 "1,10",
1182 "--target-rank",
1183 "6"
1184 ]
1185 ]
1186 }
1187 },
1188 "selection": {
1189 "default": {
1190 "params": {},
1191 "permutes": [
1192 "shape",
1193 "type"
1194 ]
1195 }
1196 }
1197 },
1198 "cast": {
1199 "group": "type_conversion",
1200 "support_for": [
1201 "lazy_data_gen"
1202 ],
1203 "generation": {
1204 "tosa-bi": {
1205 "supports_all": [
1206 "tosa-bi"
1207 ],
1208 "selector": "tosa-bi",
1209 "generator_args": [
1210 [
1211 "--target-dtype",
1212 "int8",
1213 "--target-dtype",
1214 "int16",
1215 "--target-dtype",
1216 "int32",
1217 "--target-dtype",
1218 "bool",
1219 "--tensor-dim-range",
1220 "16,64",
1221 "--target-rank",
1222 "1",
1223 "--target-rank",
1224 "2",
1225 "--target-rank",
1226 "3"
1227 ],
1228 [
1229 "--target-dtype",
1230 "int8",
1231 "--target-dtype",
1232 "int16",
1233 "--target-dtype",
1234 "int32",
1235 "--target-dtype",
1236 "bool",
1237 "--tensor-dim-range",
1238 "1,16",
1239 "--target-rank",
1240 "4",
1241 "--target-rank",
1242 "5"
1243 ],
1244 [
1245 "--target-dtype",
1246 "int8",
1247 "--target-shape",
1248 "1,1,65535,1,1",
1249 "--target-shape",
1250 "1,65538,1,2"
1251 ]
1252 ]
1253 },
1254 "8k_level": {
1255 "supports_all": [
1256 "tosa-bi"
1257 ],
1258 "selector": "tosa-bi",
1259 "no_negative_tests": "true",
1260 "generator_args": [
1261 [
1262 "--target-dtype",
1263 "int8",
1264 "--target-dtype",
1265 "int16",
1266 "--target-dtype",
1267 "int32",
1268 "--target-dtype",
1269 "bool",
1270 "--tensor-dim-range",
1271 "1,10",
1272 "--target-rank",
1273 "6"
1274 ]
1275 ]
1276 },
1277 "tosa-mi": {
1278 "supports_all": [
1279 "tosa-mi"
1280 ],
1281 "selector": "tosa-mi",
1282 "negative_dim_range": "1,10",
1283 "generator_args": [
1284 [
1285 "--target-dtype",
1286 "fp32",
1287 "--target-dtype",
1288 "fp16",
1289 "--target-dtype",
1290 "int8",
1291 "--target-dtype",
1292 "int16",
1293 "--target-dtype",
1294 "int32",
1295 "--fp-values-range",
1296 "-max,max",
1297 "--tensor-dim-range",
1298 "16,64",
1299 "--target-rank",
1300 "1",
1301 "--target-rank",
1302 "2",
1303 "--target-rank",
1304 "3"
1305 ],
1306 [
1307 "--target-dtype",
1308 "fp32",
1309 "--target-dtype",
1310 "fp16",
1311 "--target-dtype",
1312 "int8",
1313 "--target-dtype",
1314 "int16",
1315 "--target-dtype",
1316 "int32",
1317 "--fp-values-range",
1318 "-max,max",
1319 "--tensor-dim-range",
1320 "1,16",
1321 "--target-rank",
1322 "4",
1323 "--target-rank",
1324 "5"
1325 ],
1326 [
1327 "--target-dtype",
1328 "fp16",
1329 "--fp-values-range",
1330 "-max,max",
1331 "--target-shape",
1332 "1,1,1,65533,1",
1333 "--target-shape",
1334 "2,65538,1,1"
1335 ]
1336 ]
1337 },
1338 "tosa-ext-bf16": {
1339 "supports_all": [
1340 "tosa-ext-bf16"
1341 ],
1342 "selector": "tosa-ext-bf16",
1343 "no_negative_tests": "true",
1344 "negative_dim_range": "1,10",
1345 "generator_args": [
1346 [
1347 "--target-dtype",
1348 "fp32",
1349 "--target-dtype",
1350 "bf16",
1351 "--target-dtype",
1352 "int8",
1353 "--target-dtype",
1354 "int16",
1355 "--target-dtype",
1356 "int32",
1357 "--fp-values-range",
1358 "-max,max",
1359 "--tensor-dim-range",
1360 "16,64",
1361 "--target-rank",
1362 "1",
1363 "--target-rank",
1364 "2",
1365 "--target-rank",
1366 "3"
1367 ],
1368 [
1369 "--target-dtype",
1370 "fp32",
1371 "--target-dtype",
1372 "bf16",
1373 "--target-dtype",
1374 "int8",
1375 "--target-dtype",
1376 "int16",
1377 "--target-dtype",
1378 "int32",
1379 "--fp-values-range",
1380 "-max,max",
1381 "--tensor-dim-range",
1382 "1,16",
1383 "--target-rank",
1384 "4",
1385 "--target-rank",
1386 "5"
1387 ]
1388 ]
1389 },
1390 "tosa-ext-bf16-fp8e4m3": {
1391 "supports_all": [
1392 "tosa-ext-bf16",
1393 "tosa-ext-fp8e4m3"
1394 ],
1395 "selector": "tosa-ext-bf16-fp8e4m3",
1396 "no_negative_tests": "true",
1397 "negative_dim_range": "1,10",
1398 "generator_args": [
1399 [
1400 "--target-dtype",
1401 "bf16",
1402 "--target-dtype",
1403 "fp8e4m3",
1404 "--fp-values-range",
1405 "-max,max",
1406 "--tensor-dim-range",
1407 "16,64",
1408 "--target-rank",
1409 "1",
1410 "--target-rank",
1411 "2",
1412 "--target-rank",
1413 "3"
1414 ],
1415 [
1416 "--target-dtype",
1417 "bf16",
1418 "--target-dtype",
1419 "fp8e4m3",
1420 "--fp-values-range",
1421 "-max,max",
1422 "--tensor-dim-range",
1423 "1,16",
1424 "--target-rank",
1425 "4",
1426 "--target-rank",
1427 "5"
1428 ]
1429 ]
1430 },
1431 "tosa-ext-bf16-fp8e5m2": {
1432 "supports_all": [
1433 "tosa-ext-bf16",
1434 "tosa-ext-fp8e5m2"
1435 ],
1436 "selector": "tosa-ext-bf16-fp8e5m2",
1437 "no_negative_tests": "true",
1438 "negative_dim_range": "1,10",
1439 "generator_args": [
1440 [
1441 "--target-dtype",
1442 "bf16",
1443 "--target-dtype",
1444 "fp8e5m2",
1445 "--fp-values-range",
1446 "-max,max",
1447 "--tensor-dim-range",
1448 "16,64",
1449 "--target-rank",
1450 "1",
1451 "--target-rank",
1452 "2",
1453 "--target-rank",
1454 "3"
1455 ],
1456 [
1457 "--target-dtype",
1458 "bf16",
1459 "--target-dtype",
1460 "fp8e5m2",
1461 "--fp-values-range",
1462 "-max,max",
1463 "--tensor-dim-range",
1464 "1,16",
1465 "--target-rank",
1466 "4",
1467 "--target-rank",
1468 "5"
1469 ]
1470 ]
1471 },
1472 "tosa-ext-fp8e4m3": {
1473 "supports_all": [
1474 "tosa-ext-fp8e4m3"
1475 ],
1476 "no_negative_tests": "true",
1477 "selector": "tosa-ext-fp8e4m3",
1478 "from_version": "v0.100.0",
1479 "generator_args": [
1480 [
1481 "--target-dtype",
1482 "fp8e4m3",
1483 "--target-dtype",
1484 "fp16",
1485 "--target-dtype",
1486 "fp32",
1487 "--fp-values-range",
1488 "-max,max",
1489 "--tensor-dim-range",
1490 "32,64",
1491 "--target-rank",
1492 "1",
1493 "--target-rank",
1494 "2",
1495 "--target-rank",
1496 "3",
1497 "--num-rand-permutations",
1498 "2"
1499 ]
1500 ]
1501 },
1502 "tosa-ext-fp8e5m2": {
1503 "supports_all": [
1504 "tosa-ext-fp8e5m2"
1505 ],
1506 "selector": "tosa-ext-fp8e5m2",
1507 "no_negative_tests": "true",
1508 "from_version": "v0.100.0",
1509 "generator_args": [
1510 [
1511 "--target-dtype",
1512 "fp8e5m2",
1513 "--target-dtype",
1514 "fp16",
1515 "--target-dtype",
1516 "fp32",
1517 "--fp-values-range",
1518 "-max,max",
1519 "--tensor-dim-range",
1520 "32,64",
1521 "--target-rank",
1522 "1",
1523 "--target-rank",
1524 "2",
1525 "--target-rank",
1526 "3",
1527 "--num-rand-permutations",
1528 "2"
1529 ]
1530 ]
1531 }
1532 },
1533 "selection": {
1534 "tosa-bi": {
1535 "params": {
1536 "shape": [],
1537 "type": [],
1538 "output_type": [
1539 "outi8",
1540 "outi16",
1541 "outi32",
1542 "outb"
1543 ]
1544 },
1545 "permutes": [
1546 "shape",
1547 "type",
1548 "output_type"
1549 ],
1550 "exclude_patterns": [
1551 "^((?!ERRORIF).)*_out(f16|f32)"
1552 ]
1553 },
1554 "tosa-mi": {
1555 "params": {
1556 "shape": [],
1557 "type": [],
1558 "output_type": [
1559 "outi8",
1560 "outi16",
1561 "outi32",
1562 "outf16",
1563 "outf32"
1564 ]
1565 },
1566 "permutes": [
1567 "shape",
1568 "type",
1569 "output_type"
1570 ],
1571 "exclude_patterns": [
1572 "^((?!ERRORIF).)*_(b|i8|i16|i32|bf16|f8e4m3|f8e5m2)_out(b|i8|i16|i32|bf16|f8e4m3|f8e5m2)"
1573 ]
1574 },
1575 "tosa-ext-bf16": {
1576 "params": {
1577 "shape": [],
1578 "type": [],
1579 "output_type": [
1580 "outi8",
1581 "outi16",
1582 "outi32",
1583 "outf32",
1584 "outbf16"
1585 ]
1586 },
1587 "permutes": [
1588 "shape",
1589 "type",
1590 "output_type"
1591 ],
1592 "exclude_patterns": [
1593 "^((?!ERRORIF).)*_(b|i8|i16|i32|f16|f32)_out(b|i8|i16|i32|f16|f32)",
1594 "^((?!ERRORIF).)*_bf16_out(b|f16|f8e4m3|f8e5m2)",
1595 "^((?!ERRORIF).)*_(b|f16|f8e4m3|f8e5m2)_outbf16"
1596 ]
1597 },
1598 "tosa-ext-fp8e4m3": {
1599 "params": {
1600 "shape": [],
1601 "type": [],
1602 "output_type": [
1603 "outf16",
1604 "outf32",
1605 "outf8e4m3"
1606 ]
1607 },
1608 "permutes": [
1609 "shape",
1610 "type",
1611 "output_type"
1612 ],
1613 "exclude_patterns": [
1614 "^((?!ERRORIF).)*_(b|i8|i16|i32|f16|bf16|f32)_out(b|i8|i16|i32|f16|bf16|f32|f8e5m2)",
1615 "^((?!ERRORIF).)*_f8e4m3_out(b|i8|i16|i32|bf16|f8e4m3|f8e5m2)",
1616 "^((?!ERRORIF).)*_(b|i8|i16|i32|bf16|f8e4m3|f8e5m2)_outf8e4m3"
1617 ]
1618 },
1619 "tosa-ext-fp8e5m2": {
1620 "params": {
1621 "shape": [],
1622 "type": [],
1623 "output_type": [
1624 "outf16",
1625 "outf32",
1626 "outf8e5m2"
1627 ]
1628 },
1629 "permutes": [
1630 "shape",
1631 "type",
1632 "output_type"
1633 ],
1634 "exclude_patterns": [
1635 "^((?!ERRORIF).)*_(b|i8|i16|i32|f16|bf16|f32)_out(b|i8|i16|i32|f16|bf16|f32|f8e4m3)",
1636 "^((?!ERRORIF).)*_f8e5m2_out(b|i8|i16|i32|bf16|f8e4m3|f8e5m2)",
1637 "^((?!ERRORIF).)*_(b|i8|i16|i32|bf16|f8e4m3|f8e5m2)_outf8e5m2"
1638 ]
1639 },
1640 "tosa-ext-bf16-fp8e4m3": {
1641 "params": {
1642 "shape": [],
1643 "type": [],
1644 "output_type": [
1645 "outf8e4m3",
1646 "outbf16"
1647 ]
1648 },
1649 "permutes": [
1650 "shape",
1651 "type",
1652 "output_type"
1653 ],
1654 "exclude_patterns": [
1655 "^((?!ERRORIF).)*_(b|i8|i16|f16|f32|f8e5m2)_out(b|i8|i16|i32|f16|f32|f8e5m2)",
1656 "^((?!ERRORIF).)*_(bf16|f8e4m3)_out(b|i8|i16|i32|f16|f32|f8e5m2)",
1657 "^((?!ERRORIF).)*_(b|i8 d|i16|i32|f16|f32|f8e5m2)_out(bf16|f8e4m3)"
1658 ]
1659 },
1660 "tosa-ext-bf16-fp8e5m2": {
1661 "params": {
1662 "shape": [],
1663 "type": [],
1664 "output_type": [
1665 "outf8e5m2",
1666 "outbf16"
1667 ]
1668 },
1669 "permutes": [
1670 "shape",
1671 "type",
1672 "output_type"
1673 ],
1674 "exclude_patterns": [
1675 "^((?!ERRORIF).)*_(b|i8|i16|f16|f32|f8e4m3)_out(b|i8|i16|i32|f16|f32|f8e4m3)",
1676 "^((?!ERRORIF).)*_(bf16|f8e5m2)_out(b|i8|i16|i32|f16|f32|f8e4m3)",
1677 "^((?!ERRORIF).)*_(b|i8 d|i16|i32|f16|f32|f8e4m3)_out(bf16|f8e5m2)"
1678 ]
1679 }
1680 }
1681 },
1682 "ceil": {
1683 "group": "ew_unary",
1684 "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ],
1685 "generation": {
1686 "tosa-mi": {
1687 "supports_all": [
1688 "tosa-mi"
1689 ],
1690 "generator_args": [
1691 [
1692 "--target-dtype",
1693 "fp32",
1694 "--target-dtype",
1695 "fp16",
1696 "--fp-values-range",
1697 "-max,max",
1698 "--tensor-dim-range",
1699 "24,64",
1700 "--target-rank",
1701 "0",
1702 "--target-rank",
1703 "1",
1704 "--target-rank",
1705 "2",
1706 "--target-rank",
1707 "3"
1708 ],
1709 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]1710 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]1711 "--target-dtype",
1712 "fp32",
1713 "--target-dtype",
1714 "fp16",
1715 "--fp-values-range",
1716 "-max,max",
1717 "--tensor-dim-range",
1718 "1,16",
1719 "--target-rank",
1720 "4",
1721 "--target-rank",
Jeremy Johnson074b98c2024-05-30 10:38:35 +0100[diff] [blame]1722 "5",
1723 "--target-rank",
1724 "6"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]1725 ],
1726 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]1727 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]1728 "--target-dtype",
1729 "fp16",
1730 "--fp-values-range",
1731 "-max,max",
1732 "--target-shape",
1733 "2,1,65530,1",
1734 "--target-shape",
1735 "1,1,65539,1"
1736 ]
1737 ]
1738 },
1739 "tosa-ext-bf16": {
1740 "supports_all": [
1741 "tosa-ext-bf16"
1742 ],
1743 "no_negative_tests": "true",
1744 "generator_args": [
1745 [
1746 "--target-dtype",
1747 "bf16",
1748 "--fp-values-range",
1749 "-max,max",
1750 "--tensor-dim-range",
1751 "4,64",
1752 "--target-rank",
1753 "0",
1754 "--target-rank",
1755 "1",
1756 "--target-rank",
1757 "2",
1758 "--target-rank",
1759 "3"
1760 ],
1761 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]1762 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]1763 "--target-dtype",
1764 "bf16",
1765 "--fp-values-range",
1766 "-max,max",
1767 "--tensor-dim-range",
1768 "1,16",
1769 "--target-rank",
1770 "4",
1771 "--target-rank",
Jeremy Johnson074b98c2024-05-30 10:38:35 +0100[diff] [blame]1772 "5",
1773 "--target-rank",
1774 "6"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]1775 ]
1776 ]
1777 }
1778 },
1779 "selection": {
1780 "default": {
1781 "generator_select": "true"
1782 }
1783 }
1784 },
1785 "clamp": {
1786 "group": "activation",
1787 "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ],
1788 "generation": {
1789 "tosa-bi": {
1790 "supports_all": [
1791 "tosa-bi"
1792 ],
1793 "generator_args": [
1794 [
1795 "--target-dtype",
1796 "int8",
1797 "--tensor-dim-range",
1798 "20,64",
1799 "--target-rank",
1800 "1",
1801 "--target-rank",
1802 "2",
1803 "--target-rank",
1804 "3"
1805 ],
1806 [
1807 "--target-dtype",
1808 "int8",
1809 "--tensor-dim-range",
1810 "1,20",
1811 "--target-rank",
1812 "4",
1813 "--target-rank",
1814 "5"
1815 ],
1816 [
1817 "--target-dtype",
1818 "int8",
1819 "--target-shape",
1820 "1,2,1,65536",
1821 "--target-shape",
1822 "1,3,65537,1,1"
1823 ]
1824 ]
1825 },
1826 "8k_level": {
1827 "supports_all": [
1828 "tosa-bi"
1829 ],
1830 "no_negative_tests": "true",
1831 "generator_args": [
1832 [
1833 "--target-dtype",
1834 "int8",
1835 "--tensor-dim-range",
1836 "1,10",
1837 "--target-rank",
1838 "6"
1839 ]
1840 ]
1841 },
1842 "tosa-mi": {
1843 "supports_all": [
1844 "tosa-mi"
1845 ],
1846 "selector": "tosa-mi",
1847 "generator_args": [
1848 [
1849 "--target-dtype",
1850 "fp32",
1851 "--target-dtype",
1852 "fp16",
1853 "--fp-values-range",
1854 "-max,max",
1855 "--tensor-dim-range",
1856 "20,64",
1857 "--target-rank",
1858 "0",
1859 "--target-rank",
1860 "1",
1861 "--target-rank",
1862 "2",
1863 "--target-rank",
1864 "3"
1865 ],
1866 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]1867 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]1868 "--target-dtype",
1869 "fp32",
1870 "--target-dtype",
1871 "fp16",
1872 "--fp-values-range",
1873 "-max,max",
1874 "--tensor-dim-range",
1875 "1,20",
1876 "--target-rank",
1877 "4",
1878 "--target-rank",
Jeremy Johnson4423bb72024-06-05 15:30:25 +0100[diff] [blame]1879 "5",
1880 "--target-rank",
1881 "6"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]1882 ],
1883 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]1884 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]1885 "--target-dtype",
1886 "fp32",
1887 "--fp-values-range",
1888 "-max,max",
1889 "--target-shape",
1890 "1,2,1,65536",
1891 "--target-shape",
1892 "1,3,65537,1,1"
1893 ]
1894 ]
1895 },
1896 "tosa-ext-bf16": {
1897 "supports_all": [
1898 "tosa-ext-bf16"
1899 ],
1900 "selector": "tosa-mi",
1901 "no_negative_tests": "true",
1902 "generator_args": [
1903 [
1904 "--target-dtype",
1905 "bf16",
1906 "--fp-values-range",
1907 "-max,max",
1908 "--tensor-dim-range",
1909 "20,64",
1910 "--target-rank",
1911 "0",
1912 "--target-rank",
1913 "1",
1914 "--target-rank",
1915 "2",
1916 "--target-rank",
1917 "3"
1918 ],
1919 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]1920 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]1921 "--target-dtype",
1922 "bf16",
1923 "--fp-values-range",
1924 "-max,max",
1925 "--tensor-dim-range",
1926 "1,20",
1927 "--target-rank",
1928 "4",
1929 "--target-rank",
Jeremy Johnson4423bb72024-06-05 15:30:25 +0100[diff] [blame]1930 "5",
1931 "--target-rank",
1932 "6"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]1933 ]
1934 ]
1935 },
1936 "tosa-ext-int16": {
1937 "supports_all": [
1938 "tosa-ext-int16"
1939 ],
1940 "no_negative_tests": "true",
1941 "generator_args": [
1942 [
1943 "--target-dtype",
1944 "int16",
1945 "--tensor-dim-range",
1946 "20,64",
1947 "--target-rank",
1948 "1",
1949 "--target-rank",
1950 "2",
1951 "--target-rank",
1952 "3"
1953 ],
1954 [
1955 "--target-dtype",
1956 "int16",
1957 "--tensor-dim-range",
1958 "1,20",
1959 "--target-rank",
1960 "4",
1961 "--target-rank",
1962 "5"
1963 ]
1964 ]
1965 },
1966 "8k_level_ext_int16": {
1967 "supports_all": [
1968 "tosa-ext-int16"
1969 ],
1970 "no_negative_tests": "true",
1971 "generator_args": [
1972 [
1973 "--target-dtype",
1974 "int16",
1975 "--tensor-dim-range",
1976 "1,10",
1977 "--target-rank",
1978 "6"
1979 ]
1980 ]
1981 }
1982 },
1983 "selection": {
1984 "default": {
1985 "params": {},
1986 "permutes": [
1987 "shape",
1988 "type"
1989 ]
1990 },
1991 "tosa-mi": {
1992 "generator_select": "true"
1993 }
1994 }
1995 },
1996 "clz": {
1997 "group": "ew_unary",
1998 "generation": {
1999 "standard": {
2000 "supports_all": [
2001 "tosa-bi"
2002 ],
2003 "selector": "tosa-bi",
2004 "generator_args": [
2005 [
2006 "--target-dtype",
2007 "int32",
2008 "--tensor-dim-range",
2009 "16,64",
2010 "--target-rank",
2011 "1",
2012 "--target-rank",
2013 "2",
2014 "--target-rank",
2015 "3"
2016 ],
2017 [
2018 "--target-dtype",
2019 "int32",
2020 "--tensor-dim-range",
2021 "1,16",
2022 "--target-rank",
2023 "4",
2024 "--target-rank",
2025 "5"
2026 ],
2027 [
2028 "--target-dtype",
2029 "int32",
2030 "--target-shape",
2031 "1,2,65533,1",
2032 "--target-shape",
2033 "1,65539,1,1,1"
2034 ]
2035 ]
2036 },
2037 "8k_level": {
2038 "supports_all": [
2039 "tosa-bi"
2040 ],
2041 "no_negative_tests": "true",
2042 "generator_args": [
2043 [
2044 "--target-dtype",
2045 "int32",
2046 "--tensor-dim-range",
2047 "1,10",
2048 "--target-rank",
2049 "6"
2050 ]
2051 ]
2052 }
2053 },
2054 "selection": {
2055 "default": {
2056 "params": {},
2057 "permutes": [
2058 "shape",
2059 "type"
2060 ]
2061 }
2062 }
2063 },
2064 "concat": {
2065 "group": "data_layout",
2066 "support_for": [
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]2067 "lazy_data_gen",
2068 "random_const_inputs",
2069 "stable_random_gen"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]2070 ],
2071 "generation": {
2072 "tosa-bi-mi": {
2073 "supports_any": [
2074 "tosa-bi",
2075 "tosa-mi"
2076 ],
2077 "generator_args": [
2078 [
2079 "--target-dtype",
2080 "bool",
2081 "--tensor-dim-range",
2082 "16,64",
2083 "--target-rank",
2084 "1",
2085 "--target-rank",
2086 "2",
2087 "--target-rank",
2088 "3",
2089 "--num-const-inputs-concat",
2090 "3"
2091 ],
2092 [
2093 "--target-dtype",
2094 "bool",
2095 "--tensor-dim-range",
2096 "1,16",
2097 "--target-rank",
2098 "4",
2099 "--target-rank",
2100 "5",
2101 "--num-const-inputs-concat",
2102 "1"
2103 ]
2104 ]
2105 },
2106 "tosa-bi": {
2107 "supports_all": [
2108 "tosa-bi"
2109 ],
2110 "generator_args": [
2111 [
2112 "--target-dtype",
2113 "int8",
2114 "--target-dtype",
2115 "int32",
2116 "--tensor-dim-range",
2117 "16,64",
2118 "--target-rank",
2119 "1",
2120 "--target-rank",
2121 "2",
2122 "--target-rank",
2123 "3",
2124 "--num-const-inputs-concat",
2125 "3"
2126 ],
2127 [
2128 "--target-dtype",
2129 "int8",
2130 "--target-dtype",
2131 "int32",
2132 "--tensor-dim-range",
2133 "1,16",
2134 "--target-rank",
2135 "4",
2136 "--target-rank",
2137 "5",
2138 "--num-const-inputs-concat",
2139 "1"
2140 ],
2141 [
2142 "--target-dtype",
2143 "int8",
2144 "--target-shape",
2145 "1,1,65535,1,1",
2146 "--target-shape",
2147 "1,65538,1,2",
2148 "--num-const-inputs-concat",
2149 "2"
2150 ]
2151 ]
2152 },
2153 "tosa-ext-int16": {
2154 "supports_all": [
2155 "tosa-ext-int16"
2156 ],
2157 "no_negative_tests": "true",
2158 "generator_args": [
2159 [
2160 "--target-dtype",
2161 "int16",
2162 "--tensor-dim-range",
2163 "16,64",
2164 "--target-rank",
2165 "1",
2166 "--target-rank",
2167 "2",
2168 "--target-rank",
2169 "3",
2170 "--num-const-inputs-concat",
2171 "3"
2172 ],
2173 [
2174 "--target-dtype",
2175 "int16",
2176 "--tensor-dim-range",
2177 "1,16",
2178 "--target-rank",
2179 "4",
2180 "--target-rank",
2181 "5",
2182 "--num-const-inputs-concat",
2183 "1"
2184 ]
2185 ]
2186 },
2187 "8k_level": {
2188 "supports_all": [
2189 "tosa-bi"
2190 ],
2191 "no_negative_tests": "true",
2192 "generator_args": [
2193 [
2194 "--target-dtype",
2195 "int8",
2196 "--target-dtype",
2197 "int16",
2198 "--target-dtype",
2199 "int32",
2200 "--target-dtype",
2201 "bool",
2202 "--tensor-dim-range",
2203 "1,10",
2204 "--target-rank",
2205 "6",
2206 "--num-const-inputs-concat",
2207 "2"
2208 ]
2209 ]
2210 },
2211 "tosa-mi": {
2212 "supports_all": [
2213 "tosa-mi"
2214 ],
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]2215 "selector": "tosa-mi",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]2216 "no_negative_tests": "true",
2217 "generator_args": [
2218 [
2219 "--target-dtype",
2220 "fp32",
2221 "--target-dtype",
2222 "fp16",
2223 "--fp-values-range",
2224 "-max,max",
2225 "--tensor-dim-range",
2226 "16,64",
2227 "--target-rank",
2228 "1",
2229 "--target-rank",
2230 "2",
2231 "--target-rank",
2232 "3",
2233 "--num-const-inputs-concat",
2234 "3"
2235 ],
2236 [
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]2237 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]2238 "--target-dtype",
2239 "fp32",
2240 "--target-dtype",
2241 "fp16",
2242 "--fp-values-range",
2243 "-max,max",
2244 "--tensor-dim-range",
2245 "1,16",
2246 "--target-rank",
2247 "4",
2248 "--target-rank",
2249 "5",
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]2250 "--target-rank",
2251 "6",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]2252 "--num-const-inputs-concat",
2253 "1"
2254 ],
2255 [
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]2256 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]2257 "--target-dtype",
2258 "fp32",
2259 "--fp-values-range",
2260 "-max,max",
2261 "--target-shape",
2262 "1,1,65534,1,1",
2263 "--target-shape",
2264 "1,65537,1,2",
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]2265 "--target-shape",
2266 "13,11,7",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]2267 "--num-const-inputs-concat",
2268 "2"
2269 ]
2270 ]
2271 },
2272 "tosa-ext-bf16": {
2273 "supports_all": [
2274 "tosa-ext-bf16"
2275 ],
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]2276 "selector": "tosa-mi",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]2277 "no_negative_tests": "true",
2278 "generator_args": [
2279 [
2280 "--target-dtype",
2281 "bf16",
2282 "--fp-values-range",
2283 "-max,max",
2284 "--tensor-dim-range",
2285 "16,64",
2286 "--target-rank",
2287 "1",
2288 "--target-rank",
2289 "2",
2290 "--target-rank",
2291 "3",
2292 "--num-const-inputs-concat",
2293 "3"
2294 ],
2295 [
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]2296 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]2297 "--target-dtype",
2298 "bf16",
2299 "--fp-values-range",
2300 "-max,max",
2301 "--tensor-dim-range",
2302 "1,16",
2303 "--target-rank",
2304 "4",
2305 "--target-rank",
2306 "5",
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]2307 "--target-rank",
2308 "6",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]2309 "--num-const-inputs-concat",
2310 "1"
2311 ]
2312 ]
2313 },
2314 "tosa-ext-fp8e4m3": {
2315 "supports_all": [
2316 "tosa-ext-fp8e4m3"
2317 ],
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]2318 "selector": "tosa-mi",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]2319 "no_negative_tests": "true",
2320 "from_version": "v0.100.0",
2321 "generator_args": [
2322 [
2323 "--target-dtype",
2324 "fp8e4m3",
2325 "--fp-values-range",
2326 "-max,max",
2327 "--tensor-dim-range",
2328 "32,64",
2329 "--target-rank",
2330 "1",
2331 "--target-rank",
2332 "2",
2333 "--target-rank",
2334 "3",
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]2335 "--num-const-inputs-concat",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]2336 "2"
2337 ]
2338 ]
2339 },
2340 "tosa-ext-fp8e5m2": {
2341 "supports_all": [
2342 "tosa-ext-fp8e5m2"
2343 ],
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]2344 "selector": "tosa-mi",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]2345 "no_negative_tests": "true",
2346 "from_version": "v0.100.0",
2347 "generator_args": [
2348 [
2349 "--target-dtype",
2350 "fp8e5m2",
2351 "--fp-values-range",
2352 "-max,max",
2353 "--tensor-dim-range",
2354 "32,64",
2355 "--target-rank",
2356 "1",
2357 "--target-rank",
2358 "2",
2359 "--target-rank",
2360 "3",
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]2361 "--num-const-inputs-concat",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]2362 "2"
2363 ]
2364 ]
2365 }
2366 },
2367 "selection": {
2368 "default": {
2369 "params": {},
2370 "permutes": [
2371 "shape",
2372 "type"
2373 ]
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]2374 },
2375 "tosa-mi": {
2376 "generator_select": true,
2377 "maximum": 3
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]2378 }
2379 }
2380 },
2381 "concat_shape": {
2382 "group": "shape",
2383 "generation": {
2384 "tosa-bi-mi": {
2385 "supports_any": [
2386 "tosa-bi",
2387 "tosa-mi"
2388 ],
2389 "no_negative_tests": "true",
2390 "generator_args": [
2391 [
2392 "--target-dtype",
2393 "shape",
2394 "--target-rank",
2395 "1",
2396 "--target-shape",
2397 "1",
2398 "--num-const-inputs-concat",
2399 "2"
2400 ]
2401 ]
2402 }
2403 },
2404 "selection": {
2405 "default": {
2406 "params": {},
2407 "permutes": []
2408 }
2409 }
2410 },
2411 "cond_if": {
2412 "group": "control_flow",
2413 "gen_filter": "^cond_if",
2414 "generation": {
2415 "standard": {
2416 "supports_any": [
2417 "tosa-bi",
2418 "tosa-mi"
2419 ],
2420 "generator_args": [
2421 [
2422 "--target-dtype",
2423 "bool",
2424 "--target-dtype",
2425 "int8",
2426 "--target-dtype",
2427 "int16",
2428 "--target-dtype",
2429 "int32",
2430 "--target-rank",
2431 "1",
2432 "--target-rank",
2433 "2",
2434 "--target-rank",
2435 "3",
2436 "--target-rank",
2437 "4"
2438 ],
2439 [
2440 "--target-dtype",
2441 "int32",
2442 "--target-shape",
2443 "3,22,17,1,2"
2444 ]
2445 ]
2446 }
2447 },
2448 "selection": {
2449 "default": {
2450 "params": {},
2451 "permutes": [
2452 "variant",
2453 "shape",
2454 "cond"
2455 ]
2456 }
2457 }
2458 },
2459 "const": {
2460 "group": "data_nodes",
2461 "support_for": [
2462 "lazy_data_gen"
2463 ],
2464 "generation": {
2465 "tosa-bi": {
2466 "supports_all": [
2467 "tosa-bi"
2468 ],
2469 "no_negative_tests": "true",
2470 "generator_args": [
2471 [
2472 "--target-dtype",
2473 "int8",
2474 "--target-dtype",
2475 "int16",
2476 "--target-dtype",
2477 "int32",
2478 "--target-dtype",
2479 "bool",
2480 "--target-rank",
2481 "1",
2482 "--target-rank",
2483 "2",
2484 "--target-rank",
2485 "3"
2486 ],
2487 [
2488 "--target-dtype",
2489 "int8",
2490 "--target-dtype",
2491 "int16",
2492 "--target-dtype",
2493 "int32",
2494 "--target-dtype",
2495 "bool",
2496 "--tensor-dim-range",
2497 "1,16",
2498 "--target-rank",
2499 "4",
2500 "--target-rank",
2501 "5"
2502 ],
2503 [
2504 "--target-dtype",
2505 "int8",
2506 "--target-shape",
2507 "1,1,65537,1,1",
2508 "--target-shape",
2509 "1,65530,1,2"
2510 ]
2511 ]
2512 },
2513 "tosa-mi": {
2514 "supports_all": [
2515 "tosa-mi"
2516 ],
2517 "no_negative_tests": "true",
2518 "generator_args": [
2519 [
2520 "--target-dtype",
2521 "fp32",
2522 "--target-dtype",
2523 "fp16",
2524 "--target-dtype",
2525 "bf16",
2526 "--fp-values-range",
2527 "-max,max",
2528 "--target-rank",
2529 "1",
2530 "--target-rank",
2531 "2",
2532 "--target-rank",
2533 "3"
2534 ],
2535 [
2536 "--target-dtype",
2537 "fp32",
2538 "--target-dtype",
2539 "fp16",
2540 "--target-dtype",
2541 "bf16",
2542 "--fp-values-range",
2543 "-max,max",
2544 "--tensor-dim-range",
2545 "1,16",
2546 "--target-rank",
2547 "4",
2548 "--target-rank",
2549 "5"
2550 ],
2551 [
2552 "--target-dtype",
2553 "fp32",
2554 "--fp-values-range",
2555 "-max,max",
2556 "--target-shape",
2557 "1,65540,1,1,1",
2558 "--target-shape",
2559 "1,2,1,65529"
2560 ]
2561 ]
2562 },
2563 "tosa-ext-bf16": {
2564 "supports_all": [
2565 "tosa-ext-bf16"
2566 ],
2567 "no_negative_tests": "true",
2568 "generator_args": [
2569 [
2570 "--target-dtype",
2571 "bf16",
2572 "--fp-values-range",
2573 "-max,max",
2574 "--target-rank",
2575 "1",
2576 "--target-rank",
2577 "2",
2578 "--target-rank",
2579 "3"
2580 ],
2581 [
2582 "--target-dtype",
2583 "bf16",
2584 "--fp-values-range",
2585 "-max,max",
2586 "--tensor-dim-range",
2587 "1,16",
2588 "--target-rank",
2589 "4",
2590 "--target-rank",
2591 "5"
2592 ]
2593 ]
2594 },
2595 "tosa-ext-fp8e4m3": {
2596 "supports_all": [
2597 "tosa-ext-fp8e4m3"
2598 ],
2599 "no_negative_tests": "true",
2600 "from_version": "v0.100.0",
2601 "generator_args": [
2602 [
2603 "--target-dtype",
2604 "fp8e4m3",
2605 "--fp-values-range",
2606 "-max,max",
2607 "--tensor-dim-range",
2608 "32,64",
2609 "--target-rank",
2610 "1",
2611 "--target-rank",
2612 "2",
2613 "--target-rank",
2614 "3",
2615 "--num-rand-permutations",
2616 "2"
2617 ]
2618 ]
2619 },
2620 "tosa-ext-fp8e5m2": {
2621 "supports_all": [
2622 "tosa-ext-fp8e5m2"
2623 ],
2624 "no_negative_tests": "true",
2625 "from_version": "v0.100.0",
2626 "generator_args": [
2627 [
2628 "--target-dtype",
2629 "fp8e5m2",
2630 "--fp-values-range",
2631 "-max,max",
2632 "--tensor-dim-range",
2633 "32,64",
2634 "--target-rank",
2635 "1",
2636 "--target-rank",
2637 "2",
2638 "--target-rank",
2639 "3",
2640 "--num-rand-permutations",
2641 "2"
2642 ]
2643 ]
2644 },
2645 "tosa-ext-int16": {
2646 "supports_all": [
2647 "tosa-ext-int16"
2648 ],
2649 "no_negative_tests": "true",
2650 "generator_args": [
2651 [
2652 "--target-dtype",
2653 "int48",
2654 "--target-rank",
2655 "1",
2656 "--target-rank",
2657 "2",
2658 "--target-rank",
2659 "3"
2660 ],
2661 [
2662 "--target-dtype",
2663 "int48",
2664 "--tensor-dim-range",
2665 "1,16",
2666 "--target-rank",
2667 "4",
2668 "--target-rank",
2669 "5"
2670 ]
2671 ]
2672 }
2673 },
2674 "selection": {
2675 "default": {
2676 "params": {},
2677 "permutes": [
2678 "shape",
2679 "type"
2680 ]
2681 }
2682 }
2683 },
2684 "const_shape": {
2685 "group": "shape",
2686 "generation": {
2687 "standard": {
2688 "supports_any": [
2689 "tosa-bi",
2690 "tosa-mi"
2691 ],
2692 "no_negative_tests": "true",
2693 "generator_args": [
2694 [
2695 "--target-dtype",
2696 "shape",
2697 "--target-rank",
2698 "1",
2699 "--target-shape",
2700 "1"
2701 ]
2702 ]
2703 }
2704 },
2705 "selection": {
2706 "default": {
2707 "params": {},
2708 "permutes": []
2709 }
2710 }
2711 },
2712 "conv2d": {
2713 "group": "tensor",
2714 "support_for": [
2715 "lazy_data_gen",
2716 "stable_random_gen"
2717 ],
2718 "gen_filter": "^conv2d",
2719 "generation": {
2720 "tosa-bi": {
2721 "supports_all": [
2722 "tosa-bi"
2723 ],
2724 "selector": "tosa-bi",
2725 "generator_args": [
2726 [
2727 "--target-dtype",
2728 "int8",
2729 "--target-shape",
2730 "1,49,42,28",
2731 "--target-shape",
2732 "1,11,44,13",
2733 "--tensor-dim-range",
2734 "1,16",
2735 "--allow-pooling-and-conv-oversizes"
2736 ],
2737 [
2738 "--target-dtype",
2739 "int8",
2740 "--target-shape",
2741 "1,65535,3,1",
2742 "--target-shape",
2743 "1,2,65536,1",
2744 "--max-conv-dilation",
2745 "1",
2746 "--tensor-dim-range",
2747 "1,8",
2748 "--allow-pooling-and-conv-oversizes"
2749 ]
2750 ]
2751 },
2752 "tosa-ext-int16": {
2753 "supports_all": [
2754 "tosa-ext-int16"
2755 ],
2756 "selector": "tosa-ext-int16",
2757 "no_negative_tests": "true",
2758 "generator_args": [
2759 [
2760 "--target-dtype",
2761 "int16",
2762 "--target-shape",
2763 "1,49,42,28",
2764 "--target-shape",
2765 "1,11,44,13",
2766 "--tensor-dim-range",
2767 "1,16",
2768 "--allow-pooling-and-conv-oversizes"
2769 ]
2770 ]
2771 },
2772 "8k_level": {
2773 "supports_all": [
2774 "tosa-bi"
2775 ],
2776 "no_negative_tests": "true",
2777 "selector": "8k_level",
2778 "generator_args": [
2779 [
2780 "--target-dtype",
2781 "int8",
2782 "--tensor-dim-range",
2783 "2,5",
2784 "--level-8k-sizes"
2785 ]
2786 ]
2787 },
2788 "tosa-mi": {
2789 "supports_all": [
2790 "tosa-mi"
2791 ],
2792 "selector": "tosa-mi",
2793 "negative_dim_range": "1,10",
2794 "generator_args": [
2795 [
2796 "--target-dtype",
2797 "fp32",
2798 "--target-dtype",
2799 "fp16",
2800 "--fp-values-range",
2801 "-max,max",
2802 "--target-shape",
2803 "1,34,19,27",
2804 "--target-shape",
2805 "1,16,51,7",
2806 "--tensor-dim-range",
2807 "1,16",
2808 "--allow-pooling-and-conv-oversizes"
2809 ],
2810 [
2811 "--target-dtype",
2812 "fp32",
2813 "--fp-values-range",
2814 "-max,max",
2815 "--target-shape",
2816 "1,65537,1,3",
2817 "--target-shape",
2818 "1,2,65530,2",
2819 "--tensor-dim-range",
2820 "1,16",
2821 "--max-conv-dilation",
2822 "1",
2823 "--allow-pooling-and-conv-oversizes"
2824 ],
2825 [
2826 "--target-dtype",
2827 "fp32",
2828 "--target-dtype",
2829 "fp16",
2830 "--fp-values-range",
2831 "-max,max",
2832 "--conv-kernel",
2833 "7,7",
2834 "--conv-kernel",
2835 "9,9",
2836 "--target-shape",
2837 "1,127,127,1",
2838 "--target-shape",
2839 "1,256,256,1",
2840 "--target-shape",
2841 "1,1080,720,1",
2842 "--tensor-dim-range",
2843 "1,8"
2844 ]
2845 ]
2846 },
2847 "tosa-ext-bf16": {
2848 "supports_all": [
2849 "tosa-ext-bf16"
2850 ],
2851 "selector": "tosa-mi",
2852 "no_negative_tests": "true",
2853 "negative_dim_range": "1,10",
2854 "generator_args": [
2855 [
2856 "--target-dtype",
2857 "bf16",
2858 "--fp-values-range",
2859 "-max,max",
2860 "--target-shape",
2861 "1,34,19,27",
2862 "--target-shape",
2863 "1,16,51,7",
2864 "--tensor-dim-range",
2865 "1,16",
2866 "--allow-pooling-and-conv-oversizes"
2867 ],
2868 [
2869 "--target-dtype",
2870 "bf16",
2871 "--fp-values-range",
2872 "-max,max",
2873 "--conv-kernel",
2874 "7,7",
2875 "--conv-kernel",
2876 "9,9",
2877 "--target-shape",
2878 "1,127,127,1",
2879 "--target-shape",
2880 "1,256,256,1",
2881 "--target-shape",
2882 "1,1080,720,1",
2883 "--tensor-dim-range",
2884 "1,8"
2885 ]
2886 ]
2887 },
2888 "tosa-ext-fp8e4m3": {
2889 "supports_all": [
2890 "tosa-ext-fp8e4m3"
2891 ],
2892 "selector": "tosa-mi",
2893 "no_negative_tests": "true",
2894 "from_version": "v0.100.0",
2895 "generator_args": [
2896 [
2897 "--target-dtype",
2898 "fp8e4m3",
2899 "--fp-values-range",
2900 "-max,max",
2901 "--tensor-dim-range",
2902 "32,64",
2903 "--num-rand-permutations",
2904 "2"
2905 ]
2906 ]
2907 },
2908 "tosa-ext-fp8e5m2": {
2909 "supports_all": [
2910 "tosa-ext-fp8e5m2"
2911 ],
2912 "selector": "tosa-mi",
2913 "no_negative_tests": "true",
2914 "from_version": "v0.100.0",
2915 "generator_args": [
2916 [
2917 "--target-dtype",
2918 "fp8e5m2",
2919 "--fp-values-range",
2920 "-max,max",
2921 "--tensor-dim-range",
2922 "32,64",
2923 "--num-rand-permutations",
2924 "2"
2925 ]
2926 ]
2927 },
2928 "8k_level_ext-int16": {
2929 "supports_all": [
2930 "tosa-ext-int16"
2931 ],
2932 "no_negative_tests": "true",
2933 "selector": "8k_level",
2934 "generator_args": [
2935 [
2936 "--target-dtype",
2937 "int16",
2938 "--tensor-dim-range",
2939 "2,5",
2940 "--level-8k-sizes"
2941 ]
2942 ]
2943 }
2944 },
2945 "selection": {
2946 "tosa-bi": {
2947 "params": {
2948 "shape": [],
2949 "type": [
2950 "i8xi8"
2951 ],
2952 "kernel": [],
2953 "stride": [],
2954 "pad": [],
2955 "dilation": []
2956 },
2957 "permutes": [
2958 "kernel",
2959 "shape",
2960 "type",
2961 "pad"
2962 ]
2963 },
2964 "tosa-ext-int16": {
2965 "params": {
2966 "shape": [],
2967 "type": [
2968 "i16xi8"
2969 ],
2970 "kernel": [],
2971 "stride": [],
2972 "pad": [],
2973 "dilation": []
2974 },
2975 "permutes": [
2976 "kernel",
2977 "shape",
2978 "type",
2979 "pad"
2980 ]
2981 },
2982 "tosa-ext-int4": {
2983 "params": {
2984 "shape": [],
2985 "type": [
2986 "i8xi4"
2987 ],
2988 "kernel": [],
2989 "stride": [],
2990 "pad": [],
2991 "dilation": []
2992 },
2993 "permutes": [
2994 "kernel",
2995 "shape",
2996 "type",
2997 "pad"
2998 ]
2999 },
3000 "8k_level": {
3001 "all": "true"
3002 },
3003 "tosa-mi": {
3004 "generator_select": "true",
3005 "permutes": [
3006 "shape",
3007 "type",
3008 "kernel",
3009 "acc_type"
3010 ],
3011 "full_params": [
3012 "stride",
3013 "dilation"
3014 ],
3015 "specifics": {
3016 "pad": [
3017 "(0, 0, 0, 0)"
3018 ]
3019 },
3020 "groups": [
3021 "s"
3022 ],
3023 "maximum": 3
3024 }
3025 }
3026 },
3027 "conv3d": {
3028 "group": "tensor",
3029 "support_for": [
3030 "lazy_data_gen"
3031 ],
3032 "gen_filter": "^conv3d",
3033 "generation": {
3034 "tosa-bi": {
3035 "supports_all": [
3036 "tosa-bi"
3037 ],
3038 "selector": "tosa-bi",
3039 "generator_args": [
3040 [
3041 "--target-dtype",
3042 "int8",
3043 "--target-shape",
3044 "1,9,21,14,1",
3045 "--target-shape",
3046 "1,9,11,12,3",
3047 "--tensor-dim-range",
3048 "1,4",
3049 "--allow-pooling-and-conv-oversizes"
3050 ],
3051 [
3052 "--target-dtype",
3053 "int8",
3054 "--target-shape",
3055 "1,1,65535,2,1",
3056 "--target-shape",
3057 "1,1,2,65536,1",
3058 "--max-conv-dilation",
3059 "1",
3060 "--tensor-dim-range",
3061 "1,2",
3062 "--allow-pooling-and-conv-oversizes"
3063 ]
3064 ]
3065 },
3066 "8k_level": {
3067 "supports_all": [
3068 "tosa-bi"
3069 ],
3070 "no_negative_tests": "true",
3071 "selector": "8k_level",
3072 "generator_args": [
3073 [
3074 "--target-dtype",
3075 "int8",
3076 "--tensor-dim-range",
3077 "2,5",
3078 "--level-8k-sizes"
3079 ]
3080 ]
3081 },
3082 "8k_level_ext_int16": {
3083 "supports_all": [
3084 "tosa-bi"
3085 ],
3086 "no_negative_tests": "true",
3087 "selector": "8k_level",
3088 "generator_args": [
3089 [
3090 "--target-dtype",
3091 "int16",
3092 "--tensor-dim-range",
3093 "2,5",
3094 "--level-8k-sizes"
3095 ]
3096 ]
3097 },
3098 "tosa-mi": {
3099 "supports_all": [
3100 "tosa-mi"
3101 ],
3102 "selector": "tosa-mi",
3103 "negative_dim_range": "1,10",
3104 "generator_args": [
3105 [
3106 "--target-dtype",
3107 "fp32",
3108 "--target-dtype",
3109 "fp16",
3110 "--fp-values-range",
3111 "-max,max",
3112 "--target-shape",
3113 "1,7,18,5,4",
3114 "--target-shape",
3115 "1,6,12,17,3",
3116 "--tensor-dim-range",
3117 "1,4",
3118 "--allow-pooling-and-conv-oversizes"
3119 ],
3120 [
3121 "--target-dtype",
3122 "fp32",
3123 "--fp-values-range",
3124 "-max,max",
3125 "--target-shape",
3126 "1,2,65539,1,2",
3127 "--target-shape",
3128 "1,65532,1,1,3",
3129 "--tensor-dim-range",
3130 "1,2",
3131 "--max-conv-dilation",
3132 "1",
3133 "--allow-pooling-and-conv-oversizes"
3134 ]
3135 ]
3136 },
3137 "tosa-ext-bf16": {
3138 "supports_all": [
3139 "tosa-ext-bf16"
3140 ],
3141 "selector": "tosa-mi",
3142 "no_negative_tests": "true",
3143 "negative_dim_range": "1,10",
3144 "generator_args": [
3145 [
3146 "--target-dtype",
3147 "bf16",
3148 "--fp-values-range",
3149 "-max,max",
3150 "--target-shape",
3151 "1,7,18,5,4",
3152 "--target-shape",
3153 "1,6,12,17,3",
3154 "--tensor-dim-range",
3155 "1,4",
3156 "--allow-pooling-and-conv-oversizes"
3157 ]
3158 ]
3159 },
3160 "tosa-ext-fp8e4m3": {
3161 "supports_all": [
3162 "tosa-ext-fp8e4m3"
3163 ],
3164 "selector": "tosa-mi",
3165 "no_negative_tests": "true",
3166 "from_version": "v0.100.0",
3167 "generator_args": [
3168 [
3169 "--target-dtype",
3170 "fp8e4m3",
3171 "--fp-values-range",
3172 "-max,max",
3173 "--target-shape",
3174 "1,7,18,5,4",
3175 "--target-shape",
3176 "1,6,12,17,3",
3177 "--tensor-dim-range",
3178 "1,4",
3179 "--num-rand-permutations",
3180 "2"
3181 ]
3182 ]
3183 },
3184 "tosa-ext-fp8e5m2": {
3185 "supports_all": [
3186 "tosa-ext-fp8e5m2"
3187 ],
3188 "selector": "tosa-mi",
3189 "no_negative_tests": "true",
3190 "from_version": "v0.100.0",
3191 "generator_args": [
3192 [
3193 "--target-dtype",
3194 "fp8e5m2",
3195 "--fp-values-range",
3196 "-max,max",
3197 "--target-shape",
3198 "1,7,18,5,4",
3199 "--target-shape",
3200 "1,6,12,17,3",
3201 "--tensor-dim-range",
3202 "1,4",
3203 "--num-rand-permutations",
3204 "2"
3205 ]
3206 ]
3207 },
3208 "tosa-ext-int16": {
3209 "supports_all": [
3210 "tosa-ext-int16"
3211 ],
3212 "selector": "tosa-ext-int16",
3213 "no_negative_tests": "true",
3214 "generator_args": [
3215 [
3216 "--target-dtype",
3217 "int16",
3218 "--target-shape",
3219 "1,9,21,14,1",
3220 "--target-shape",
3221 "1,9,11,12,3",
3222 "--tensor-dim-range",
3223 "1,4",
3224 "--allow-pooling-and-conv-oversizes"
3225 ]
3226 ]
3227 }
3228 },
3229 "selection": {
3230 "tosa-bi": {
3231 "params": {
3232 "shape": [],
3233 "type": [
3234 "i8xi8"
3235 ],
3236 "kernel": [],
3237 "stride": [],
3238 "pad": [],
3239 "dilation": []
3240 },
3241 "permutes": [
3242 "kernel",
3243 "shape",
3244 "type",
3245 "pad",
3246 "stride"
3247 ],
3248 "exclude_patterns": [
3249 "^((?!ERRORIF).)*_(i16xi8|i8xi4)_.*"
3250 ]
3251 },
3252 "tosa-ext-int16": {
3253 "params": {
3254 "shape": [],
3255 "type": [
3256 "i16xi8"
3257 ],
3258 "kernel": [],
3259 "stride": [],
3260 "pad": [],
3261 "dilation": []
3262 },
3263 "permutes": [
3264 "kernel",
3265 "shape",
3266 "type",
3267 "pad",
3268 "stride"
3269 ],
3270 "exclude_patterns": [
3271 "^((?!ERRORIF).)*_(i8xi8|i8xi4)_.*"
3272 ]
3273 },
3274 "tosa-ext-int4": {
3275 "params": {
3276 "shape": [],
3277 "type": [
3278 "i8xi4"
3279 ],
3280 "kernel": [],
3281 "stride": [],
3282 "pad": [],
3283 "dilation": []
3284 },
3285 "permutes": [
3286 "kernel",
3287 "shape",
3288 "type",
3289 "pad",
3290 "stride"
3291 ],
3292 "exclude_patterns": [
3293 "^((?!ERRORIF).)*_(i8xi8|i16xi8)_.*"
3294 ]
3295 },
3296 "8k_level": {
3297 "all": "true"
3298 },
3299 "tosa-mi": {
3300 "params": {
3301 "shape": [],
3302 "type": [],
3303 "kernel": [],
3304 "stride": [],
3305 "pad": [],
3306 "dilation": []
3307 },
3308 "permutes": [
3309 "kernel",
3310 "shape",
3311 "type",
3312 "pad",
3313 "stride"
3314 ]
3315 }
3316 }
3317 },
3318 "cos": {
3319 "group": "ew_unary",
Jeremy Johnson074b98c2024-05-30 10:38:35 +0100[diff] [blame]3320 "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ],
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]3321 "generation": {
3322 "tosa-mi": {
3323 "supports_all": [
3324 "tosa-mi"
3325 ],
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]3326 "generator_args": [
3327 [
3328 "--target-dtype",
3329 "fp32",
3330 "--target-dtype",
3331 "fp16",
3332 "--fp-values-range",
3333 "-max,max",
3334 "--tensor-dim-range",
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]3335 "35,64",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]3336 "--target-rank",
3337 "1",
3338 "--target-rank",
3339 "2",
3340 "--target-rank",
3341 "3"
3342 ],
3343 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]3344 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]3345 "--target-dtype",
Jeremy Johnson074b98c2024-05-30 10:38:35 +0100[diff] [blame]3346 "fp32",
3347 "--target-dtype",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]3348 "fp16",
3349 "--fp-values-range",
3350 "-max,max",
3351 "--tensor-dim-range",
3352 "1,15",
3353 "--target-rank",
3354 "4",
3355 "--target-rank",
Jeremy Johnson074b98c2024-05-30 10:38:35 +0100[diff] [blame]3356 "5",
3357 "--target-rank",
3358 "6"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]3359 ],
3360 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]3361 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]3362 "--target-dtype",
3363 "fp32",
3364 "--fp-values-range",
3365 "-max,max",
3366 "--target-shape",
3367 "2,1,65537,1",
3368 "--target-shape",
3369 "3,1,65539,2,1"
3370 ]
3371 ]
3372 },
3373 "tosa-ext-bf16": {
3374 "supports_all": [
3375 "tosa-ext-bf16"
3376 ],
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]3377 "no_negative_tests": "true",
3378 "generator_args": [
3379 [
3380 "--target-dtype",
3381 "bf16",
3382 "--fp-values-range",
3383 "-max,max",
3384 "--tensor-dim-range",
3385 "15,64",
3386 "--target-rank",
3387 "1",
3388 "--target-rank",
3389 "2",
3390 "--target-rank",
3391 "3"
Jeremy Johnson074b98c2024-05-30 10:38:35 +0100[diff] [blame]3392 ],
3393 [
3394 "--no-special-tests",
3395 "--target-dtype",
3396 "bf16",
3397 "--fp-values-range",
3398 "-max,max",
3399 "--tensor-dim-range",
3400 "1,13",
3401 "--target-rank",
3402 "4",
3403 "--target-rank",
3404 "5",
3405 "--target-rank",
3406 "6"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]3407 ]
3408 ]
3409 }
3410 },
3411 "selection": {
Jeremy Johnson074b98c2024-05-30 10:38:35 +0100[diff] [blame]3412 "default": {
3413 "generator_select": "true"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]3414 }
3415 }
3416 },
3417 "depthwise_conv2d": {
3418 "group": "tensor",
3419 "support_for": [
3420 "lazy_data_gen"
3421 ],
3422 "gen_filter": "^depthwise_conv2d",
3423 "generation": {
3424 "tosa-bi": {
3425 "supports_all": [
3426 "tosa-bi"
3427 ],
3428 "selector": "tosa-bi",
3429 "generator_args": [
3430 [
3431 "--target-dtype",
3432 "int8",
3433 "--target-shape",
3434 "1,9,44,3",
3435 "--target-shape",
3436 "1,11,33,4",
3437 "--tensor-dim-range",
3438 "1,16",
3439 "--allow-pooling-and-conv-oversizes"
3440 ],
3441 [
3442 "--target-dtype",
3443 "int8",
3444 "--target-shape",
3445 "1,65532,2,1",
3446 "--target-shape",
3447 "1,4,65537,1",
3448 "--max-conv-dilation",
3449 "1",
3450 "--tensor-dim-range",
3451 "1,16",
3452 "--allow-pooling-and-conv-oversizes"
3453 ]
3454 ]
3455 },
3456 "8k_level": {
3457 "supports_all": [
3458 "tosa-bi"
3459 ],
3460 "no_negative_tests": "true",
3461 "selector": "8k_level",
3462 "generator_args": [
3463 [
3464 "--target-dtype",
3465 "int8",
3466 "--target-dtype",
3467 "int16",
3468 "--tensor-dim-range",
3469 "2,5",
3470 "--level-8k-sizes"
3471 ]
3472 ]
3473 },
3474 "tosa-mi": {
3475 "supports_all": [
3476 "tosa-mi"
3477 ],
3478 "selector": "tosa-mi",
3479 "negative_dim_range": "1,10",
3480 "generator_args": [
3481 [
3482 "--target-dtype",
3483 "fp32",
3484 "--target-dtype",
3485 "fp16",
3486 "--fp-values-range",
3487 "-max,max",
3488 "--target-shape",
3489 "1,17,31,4",
3490 "--target-shape",
3491 "1,37,11,5",
3492 "--tensor-dim-range",
3493 "1,32",
3494 "--allow-pooling-and-conv-oversizes"
3495 ],
3496 [
3497 "--target-dtype",
3498 "fp32",
3499 "--fp-values-range",
3500 "-max,max",
3501 "--target-shape",
3502 "1,1,65531,2",
3503 "--target-shape",
3504 "1,65540,1,3",
3505 "--tensor-dim-range",
3506 "1,16",
3507 "--max-conv-dilation",
3508 "1",
3509 "--allow-pooling-and-conv-oversizes"
3510 ]
3511 ]
3512 },
3513 "tosa-ext-bf16": {
3514 "supports_all": [
3515 "tosa-ext-bf16"
3516 ],
3517 "selector": "tosa-ext-bf16",
3518 "no_negative_tests": "true",
3519 "negative_dim_range": "1,10",
3520 "generator_args": [
3521 [
3522 "--target-dtype",
3523 "bf16",
3524 "--fp-values-range",
3525 "-max,max",
3526 "--target-shape",
3527 "1,17,31,4",
3528 "--target-shape",
3529 "1,37,11,5",
3530 "--tensor-dim-range",
3531 "1,32",
3532 "--allow-pooling-and-conv-oversizes"
3533 ]
3534 ]
3535 },
3536 "tosa-ext-fp8e4m3": {
3537 "supports_all": [
3538 "tosa-ext-fp8e4m3"
3539 ],
3540 "selector": "tosa-ext-fp8e4m3",
3541 "no_negative_tests": "true",
3542 "from_version": "v0.100.0",
3543 "generator_args": [
3544 [
3545 "--target-dtype",
3546 "fp8e4m3",
3547 "--fp-values-range",
3548 "-max,max",
3549 "--tensor-dim-range",
3550 "32,64",
3551 "--num-rand-permutations",
3552 "2"
3553 ]
3554 ]
3555 },
3556 "tosa-ext-fp8e5m2": {
3557 "supports_all": [
3558 "tosa-ext-fp8e5m2"
3559 ],
3560 "selector": "tosa-ext-fp8e5m2",
3561 "no_negative_tests": "true",
3562 "from_version": "v0.100.0",
3563 "generator_args": [
3564 [
3565 "--target-dtype",
3566 "fp8e5m2",
3567 "--fp-values-range",
3568 "-max,max",
3569 "--tensor-dim-range",
3570 "32,64",
3571 "--num-rand-permutations",
3572 "2"
3573 ]
3574 ]
3575 },
3576 "tosa-ext-int16": {
3577 "supports_all": [
3578 "tosa-ext-int16"
3579 ],
3580 "selector": "tosa-bi",
3581 "no_negative_tests": "true",
3582 "generator_args": [
3583 [
3584 "--target-dtype",
3585 "int16",
3586 "--target-shape",
3587 "1,9,44,3",
3588 "--target-shape",
3589 "1,11,33,4",
3590 "--tensor-dim-range",
3591 "1,16",
3592 "--allow-pooling-and-conv-oversizes"
3593 ]
3594 ]
3595 }
3596 },
3597 "selection": {
3598 "tosa-bi": {
3599 "params": {
3600 "shape": [],
3601 "type": [
3602 "i8xi8",
3603 "i16xi8",
3604 "i8xi4"
3605 ],
3606 "kernel": [],
3607 "stride": [],
3608 "pad": [],
3609 "dilation": []
3610 },
3611 "permutes": [
3612 "kernel",
3613 "shape",
3614 "type",
3615 "pad"
3616 ]
3617 },
3618 "8k_level": {
3619 "all": "true"
3620 },
3621 "tosa-mi": {
IanTayler1a083b52024-05-10 09:35:24 +0100[diff] [blame^]3622 "generator_select": "true",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]3623 "permutes": [
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]3624 "shape",
3625 "type",
IanTayler1a083b52024-05-10 09:35:24 +0100[diff] [blame^]3626 "kernel",
3627 "acc_type"
3628 ],
3629 "full_params": [
3630 "stride",
3631 "dilation"
3632 ],
3633 "specifics": {
3634 "pad": [
3635 "(0, 0, 0, 0)"
3636 ]
3637 },
3638 "groups": [
3639 "s"
3640 ],
3641 "maximum": 3
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]3642 },
3643 "tosa-ext-bf16": {
3644 "params": {
3645 "shape": [],
3646 "type": [
3647 "bf16xbf16"
3648 ],
3649 "kernel": [],
3650 "stride": [],
3651 "pad": [],
3652 "dilation": []
3653 },
3654 "permutes": [
3655 "kernel",
3656 "shape",
3657 "type",
3658 "pad"
3659 ]
3660 },
3661 "tosa-ext-fp8e4m3": {
3662 "params": {
3663 "shape": [],
3664 "type": [
3665 "f8e4m3xf8e4m3"
3666 ],
3667 "kernel": [],
3668 "stride": [],
3669 "pad": [],
3670 "dilation": []
3671 },
3672 "permutes": [
3673 "kernel",
3674 "shape",
3675 "type",
3676 "pad"
3677 ]
3678 },
3679 "tosa-ext-fp8e5m2": {
3680 "params": {
3681 "shape": [],
3682 "type": [
3683 "f8e5m2xf8e5m2"
3684 ],
3685 "kernel": [],
3686 "stride": [],
3687 "pad": [],
3688 "dilation": []
3689 },
3690 "permutes": [
3691 "kernel",
3692 "shape",
3693 "type",
3694 "pad"
3695 ]
3696 }
3697 }
3698 },
3699 "dim": {
3700 "group": "data_layout",
3701 "generation": {
3702 "tosa-bi-mi": {
3703 "supports_any": [
3704 "tosa-bi",
3705 "tosa-mi"
3706 ],
3707 "from_version": "v0.80.0",
3708 "generator_args": [
3709 [
3710 "--target-dtype",
3711 "bool",
3712 "--tensor-dim-range",
3713 "1,64",
3714 "--target-rank",
3715 "1",
3716 "--target-rank",
3717 "2",
3718 "--target-rank",
3719 "3"
3720 ]
3721 ]
3722 },
3723 "tosa-bi": {
3724 "supports_all": [
3725 "tosa-bi"
3726 ],
3727 "from_version": "v0.80.0",
3728 "generator_args": [
3729 [
3730 "--target-dtype",
3731 "int8",
3732 "--target-dtype",
3733 "int16",
3734 "--target-dtype",
3735 "int32",
3736 "--tensor-dim-range",
3737 "1,64",
3738 "--target-rank",
3739 "1",
3740 "--target-rank",
3741 "2",
3742 "--target-rank",
3743 "3"
3744 ]
3745 ]
3746 },
3747 "8k_level": {
3748 "supports_all": [
3749 "tosa-bi"
3750 ],
3751 "from_version": "v0.80.0",
3752 "no_negative_tests": "true",
3753 "generator_args": [
3754 [
3755 "--target-dtype",
3756 "int8",
3757 "--tensor-dim-range",
3758 "1,10",
3759 "--target-rank",
3760 "6"
3761 ]
3762 ]
3763 },
3764 "tosa-mi": {
3765 "supports_all": [
3766 "tosa-mi"
3767 ],
3768 "from_version": "v0.80.0",
3769 "generator_args": [
3770 [
3771 "--target-dtype",
3772 "fp32",
3773 "--target-dtype",
3774 "fp16",
3775 "--fp-values-range",
3776 "-2.0,2.0",
3777 "--tensor-dim-range",
3778 "1,65",
3779 "--target-rank",
3780 "1",
3781 "--target-rank",
3782 "2",
3783 "--target-rank",
3784 "3"
3785 ]
3786 ]
3787 },
3788 "tosa-ext-bf16": {
3789 "supports_all": [
3790 "tosa-ext-bf16"
3791 ],
3792 "no_negative_tests": "true",
3793 "from_version": "v0.80.0",
3794 "generator_args": [
3795 [
3796 "--target-dtype",
3797 "bf16",
3798 "--fp-values-range",
3799 "-2.0,2.0",
3800 "--tensor-dim-range",
3801 "1,65",
3802 "--target-rank",
3803 "1",
3804 "--target-rank",
3805 "2",
3806 "--target-rank",
3807 "3"
3808 ]
3809 ]
3810 },
3811 "tosa-ext-fp8e4m3": {
3812 "supports_all": [
3813 "tosa-ext-fp8e4m3"
3814 ],
3815 "no_negative_tests": "true",
3816 "from_version": "v0.100.0",
3817 "generator_args": [
3818 [
3819 "--target-dtype",
3820 "fp8e4m3",
3821 "--fp-values-range",
3822 "-max,max",
3823 "--tensor-dim-range",
3824 "32,64",
3825 "--target-rank",
3826 "1",
3827 "--target-rank",
3828 "2",
3829 "--target-rank",
3830 "3",
3831 "--num-rand-permutations",
3832 "2"
3833 ]
3834 ]
3835 },
3836 "tosa-ext-fp8e5m2": {
3837 "supports_all": [
3838 "tosa-ext-fp8e5m2"
3839 ],
3840 "no_negative_tests": "true",
3841 "from_version": "v0.100.0",
3842 "generator_args": [
3843 [
3844 "--target-dtype",
3845 "fp8e5m2",
3846 "--fp-values-range",
3847 "-max,max",
3848 "--tensor-dim-range",
3849 "32,64",
3850 "--target-rank",
3851 "1",
3852 "--target-rank",
3853 "2",
3854 "--target-rank",
3855 "3",
3856 "--num-rand-permutations",
3857 "2"
3858 ]
3859 ]
3860 }
3861 },
3862 "selection": {
3863 "default": {
3864 "params": {},
3865 "permutes": [
3866 "shape",
3867 "type"
3868 ]
3869 }
3870 }
3871 },
3872 "div_shape": {
3873 "group": "shape",
3874 "generation": {
3875 "tosa-bi-mi": {
3876 "supports_all": [
3877 "tosa-bi",
3878 "tosa-mi"
3879 ],
3880 "generator_args": [
3881 [
3882 "--target-dtype",
3883 "shape",
3884 "--tensor-dim-range",
3885 "1,16",
3886 "--target-rank",
3887 "1"
3888 ]
3889 ]
3890 }
3891 },
3892 "selection": {
3893 "default": {
3894 "params": {},
3895 "permutes": [
3896 "shape"
3897 ]
3898 }
3899 }
3900 },
3901 "equal": {
3902 "group": "comparison",
3903 "support_for": [
3904 "lazy_data_gen"
3905 ],
3906 "generation": {
3907 "tosa-bi": {
3908 "supports_all": [
3909 "tosa-bi"
3910 ],
3911 "generator_args": [
3912 [
3913 "--target-dtype",
3914 "int32",
3915 "--tensor-dim-range",
3916 "16,64",
3917 "--target-rank",
3918 "1",
3919 "--target-rank",
3920 "2",
3921 "--target-rank",
3922 "3"
3923 ],
3924 [
3925 "--target-dtype",
3926 "int32",
3927 "--tensor-dim-range",
3928 "1,16",
3929 "--target-rank",
3930 "4",
3931 "--target-rank",
3932 "5"
3933 ],
3934 [
3935 "--target-dtype",
3936 "int32",
3937 "--target-shape",
3938 "1,2,65501,1",
3939 "--target-shape",
3940 "1,65541,1,1,1"
3941 ]
3942 ]
3943 },
3944 "8k_level": {
3945 "supports_all": [
3946 "tosa-bi"
3947 ],
3948 "no_negative_tests": "true",
3949 "generator_args": [
3950 [
3951 "--target-dtype",
3952 "int32",
3953 "--tensor-dim-range",
3954 "1,10",
3955 "--target-rank",
3956 "6"
3957 ]
3958 ]
3959 },
3960 "tosa-mi": {
3961 "supports_all": [
3962 "tosa-mi"
3963 ],
3964 "generator_args": [
3965 [
3966 "--target-dtype",
3967 "fp32",
3968 "--target-dtype",
3969 "fp16",
3970 "--fp-values-range",
3971 "-max,max",
3972 "--tensor-dim-range",
3973 "18,64",
3974 "--target-rank",
3975 "1",
3976 "--target-rank",
3977 "2",
3978 "--target-rank",
3979 "3"
3980 ],
3981 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]3982 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]3983 "--target-dtype",
3984 "fp32",
3985 "--fp-values-range",
3986 "-max,max",
3987 "--tensor-dim-range",
3988 "1,18",
3989 "--target-rank",
3990 "4",
3991 "--target-rank",
3992 "5"
3993 ],
3994 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]3995 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]3996 "--target-dtype",
3997 "fp16",
3998 "--fp-values-range",
3999 "-max,max",
4000 "--target-shape",
4001 "1,1,2,65550",
4002 "--target-shape",
4003 "1,4,65530,1,1"
4004 ]
4005 ]
4006 },
4007 "tosa-ext-bf16": {
4008 "supports_all": [
4009 "tosa-ext-bf16"
4010 ],
4011 "no_negative_tests": "true",
4012 "generator_args": [
4013 [
4014 "--target-dtype",
4015 "bf16",
4016 "--fp-values-range",
4017 "-max,max",
4018 "--tensor-dim-range",
4019 "18,64",
4020 "--target-rank",
4021 "1",
4022 "--target-rank",
4023 "2",
4024 "--target-rank",
4025 "3"
4026 ]
4027 ]
4028 }
4029 },
4030 "selection": {
4031 "default": {
4032 "params": {},
4033 "permutes": [
4034 "shape",
4035 "type"
4036 ]
4037 }
4038 }
4039 },
4040 "erf": {
4041 "group": "activation",
4042 "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ],
4043 "generation": {
4044 "tosa-mi": {
4045 "supports_all": [
4046 "tosa-mi"
4047 ],
4048 "generator_args": [
4049 [
4050 "--target-dtype",
4051 "fp32",
4052 "--target-dtype",
4053 "fp16",
4054 "--fp-values-range",
4055 "-4.0,4.0",
4056 "--tensor-dim-range",
4057 "18,60",
4058 "--target-rank",
4059 "0",
4060 "--target-rank",
4061 "1",
4062 "--target-rank",
4063 "2",
4064 "--target-rank",
4065 "3"
4066 ],
4067 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]4068 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]4069 "--target-dtype",
4070 "fp32",
4071 "--target-dtype",
4072 "fp16",
4073 "--fp-values-range",
4074 "-4.0,4.0",
4075 "--tensor-dim-range",
4076 "1,24",
4077 "--target-rank",
4078 "4",
4079 "--target-rank",
Jeremy Johnson4423bb72024-06-05 15:30:25 +0100[diff] [blame]4080 "5",
4081 "--target-rank",
4082 "6"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]4083 ],
4084 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]4085 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]4086 "--target-dtype",
4087 "fp32",
4088 "--fp-values-range",
4089 "-4.0,4.0",
4090 "--target-shape",
4091 "1,65535,2,1,1",
4092 "--target-shape",
4093 "1,65540,1,2"
4094 ]
4095 ]
4096 },
4097 "tosa-ext-bf16": {
4098 "supports_all": [
4099 "tosa-ext-bf16"
4100 ],
4101 "no_negative_tests": "true",
4102 "generator_args": [
4103 [
4104 "--target-dtype",
4105 "bf16",
4106 "--fp-values-range",
4107 "-4.0,4.0",
4108 "--tensor-dim-range",
4109 "18,60",
4110 "--target-rank",
4111 "0",
4112 "--target-rank",
4113 "1",
4114 "--target-rank",
4115 "2",
4116 "--target-rank",
4117 "3"
4118 ],
4119 [
4120 "--target-dtype",
4121 "bf16",
4122 "--fp-values-range",
4123 "-4.0,4.0",
4124 "--tensor-dim-range",
4125 "1,24",
4126 "--target-rank",
4127 "4",
4128 "--target-rank",
Jeremy Johnson4423bb72024-06-05 15:30:25 +0100[diff] [blame]4129 "5",
4130 "--target-rank",
4131 "6"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]4132 ]
4133 ]
4134 }
4135 },
4136 "selection": {
4137 "default": {
4138 "generator_select": "true"
4139 }
4140 }
4141 },
4142 "exp": {
4143 "group": "ew_unary",
4144 "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ],
4145 "generation": {
4146 "tosa-mi": {
4147 "supports_all": [
4148 "tosa-mi"
4149 ],
4150 "generator_args": [
4151 [
4152 "--target-dtype",
4153 "fp32",
4154 "--target-dtype",
4155 "fp16",
4156 "--fp-values-range",
4157 "-max,max",
4158 "--tensor-dim-range",
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]4159 "35,65",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]4160 "--target-rank",
4161 "0",
4162 "--target-rank",
4163 "1",
4164 "--target-rank",
4165 "2",
4166 "--target-rank",
4167 "3"
4168 ],
4169 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]4170 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]4171 "--target-dtype",
Jeremy Johnson074b98c2024-05-30 10:38:35 +0100[diff] [blame]4172 "fp32",
4173 "--target-dtype",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]4174 "fp16",
4175 "--fp-values-range",
4176 "-max,max",
4177 "--tensor-dim-range",
4178 "1,15",
4179 "--target-rank",
4180 "4",
4181 "--target-rank",
Jeremy Johnson074b98c2024-05-30 10:38:35 +0100[diff] [blame]4182 "5",
4183 "--target-rank",
4184 "6"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]4185 ],
4186 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]4187 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]4188 "--target-dtype",
4189 "fp32",
4190 "--fp-values-range",
4191 "-max,max",
4192 "--target-shape",
4193 "1,1,65535,4",
4194 "--target-shape",
4195 "2,65536,1,1,1"
4196 ]
4197 ]
4198 },
4199 "tosa-ext-bf16": {
4200 "supports_all": [
4201 "tosa-ext-bf16"
4202 ],
4203 "no_negative_tests": "true",
4204 "generator_args": [
4205 [
4206 "--target-dtype",
4207 "bf16",
4208 "--fp-values-range",
4209 "-max,max",
4210 "--tensor-dim-range",
4211 "15,64",
4212 "--target-rank",
4213 "0",
4214 "--target-rank",
4215 "1",
4216 "--target-rank",
4217 "2",
4218 "--target-rank",
4219 "3"
Jeremy Johnson074b98c2024-05-30 10:38:35 +0100[diff] [blame]4220 ],
4221 [
4222 "--no-special-tests",
4223 "--target-dtype",
4224 "bf16",
4225 "--fp-values-range",
4226 "-max,max",
4227 "--tensor-dim-range",
4228 "1,13",
4229 "--target-rank",
4230 "4",
4231 "--target-rank",
4232 "5",
4233 "--target-rank",
4234 "6"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]4235 ]
4236 ]
4237 }
4238 },
4239 "selection": {
4240 "default": {
4241 "generator_select": "true"
4242 }
4243 }
4244 },
4245 "fft2d": {
4246 "group": "tensor",
4247 "support_for": [
4248 "lazy_data_gen"
4249 ],
4250 "generation": {
4251 "tosa-ext-fft": {
4252 "supports_all": [
4253 "tosa-ext-fft"
4254 ],
4255 "generator_args": [
4256 [
4257 "--target-dtype",
4258 "fp32",
4259 "--fp-values-range",
4260 "-max,max"
4261 ],
4262 [
4263 "--target-dtype",
4264 "fp32",
4265 "--fp-values-range",
4266 "-max,max",
4267 "--target-shape",
4268 "1,256,64",
4269 "--target-shape",
4270 "1,4,4096"
4271 ]
4272 ]
4273 }
4274 },
4275 "selection": {
4276 "default": {
4277 "params": {},
4278 "permutes": [
4279 "shape",
4280 "type",
4281 "inverse"
4282 ]
4283 }
4284 }
4285 },
4286 "floor": {
4287 "group": "ew_unary",
4288 "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ],
4289 "generation": {
4290 "tosa-mi": {
4291 "supports_all": [
4292 "tosa-mi"
4293 ],
4294 "generator_args": [
4295 [
4296 "--target-dtype",
4297 "fp32",
4298 "--target-dtype",
4299 "fp16",
4300 "--fp-values-range",
4301 "-max,max",
4302 "--tensor-dim-range",
4303 "34,64",
4304 "--target-rank",
4305 "0",
4306 "--target-rank",
4307 "1",
4308 "--target-rank",
4309 "2",
4310 "--target-rank",
4311 "3"
4312 ],
4313 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]4314 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]4315 "--target-dtype",
4316 "fp32",
4317 "--target-dtype",
4318 "fp16",
4319 "--fp-values-range",
4320 "-max,max",
4321 "--tensor-dim-range",
4322 "1,16",
4323 "--target-rank",
4324 "4",
4325 "--target-rank",
Jeremy Johnson074b98c2024-05-30 10:38:35 +0100[diff] [blame]4326 "5",
4327 "--target-rank",
4328 "6"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]4329 ],
4330 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]4331 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]4332 "--target-dtype",
4333 "fp32",
4334 "--fp-values-range",
4335 "-max,max",
4336 "--target-shape",
4337 "1,1,65533,2",
4338 "--target-shape",
4339 "1,65537,1,1"
4340 ]
4341 ]
4342 },
4343 "tosa-ext-bf16": {
4344 "supports_all": [
4345 "tosa-ext-bf16"
4346 ],
4347 "no_negative_tests": "true",
4348 "generator_args": [
4349 [
4350 "--target-dtype",
4351 "bf16",
4352 "--fp-values-range",
4353 "-max,max",
4354 "--tensor-dim-range",
4355 "4,64",
4356 "--target-rank",
4357 "0",
4358 "--target-rank",
4359 "1",
4360 "--target-rank",
4361 "2",
4362 "--target-rank",
4363 "3"
4364 ],
4365 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]4366 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]4367 "--target-dtype",
4368 "bf16",
4369 "--fp-values-range",
4370 "-max,max",
4371 "--tensor-dim-range",
Jeremy Johnson074b98c2024-05-30 10:38:35 +0100[diff] [blame]4372 "1,14",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]4373 "--target-rank",
4374 "4",
4375 "--target-rank",
Jeremy Johnson074b98c2024-05-30 10:38:35 +0100[diff] [blame]4376 "5",
4377 "--target-rank",
4378 "6"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]4379 ]
4380 ]
4381 }
4382 },
4383 "selection": {
4384 "default": {
4385 "generator_select": "true"
4386 }
4387 }
4388 },
4389 "fully_connected": {
4390 "group": "tensor",
4391 "support_for": [
4392 "lazy_data_gen"
4393 ],
4394 "generation": {
4395 "tosa-bi": {
4396 "supports_all": [
4397 "tosa-bi"
4398 ],
4399 "generator_args": [
4400 [
4401 "--target-dtype",
4402 "int8"
4403 ],
4404 [
4405 "--target-dtype",
4406 "int8",
4407 "--target-shape",
4408 "444,1",
4409 "--target-shape",
4410 "3,65538",
4411 "--tensor-dim-range",
4412 "1,2"
4413 ],
4414 [
4415 "--target-dtype",
4416 "int8",
4417 "--target-shape",
4418 "3,16",
4419 "--target-shape",
4420 "1,23",
4421 "--tensor-dim-range",
4422 "100,200"
4423 ]
4424 ]
4425 },
4426 "tosa-mi": {
4427 "supports_all": [
4428 "tosa-mi"
4429 ],
4430 "negative_dim_range": "1,10",
4431 "generator_args": [
4432 [
4433 "--target-dtype",
4434 "fp32",
4435 "--target-dtype",
4436 "fp16",
4437 "--fp-values-range",
4438 "-max,max",
4439 "--tensor-dim-range",
4440 "25,64"
4441 ],
4442 [
4443 "--target-dtype",
4444 "fp32",
4445 "--fp-values-range",
4446 "-max,max",
4447 "--tensor-dim-range",
4448 "10,15",
4449 "--target-shape",
4450 "100,296",
4451 "--target-shape",
4452 "65540,2"
4453 ],
4454 [
4455 "--target-dtype",
4456 "fp32",
4457 "--target-dtype",
4458 "fp16",
4459 "--fp-values-range",
4460 "-max,max",
4461 "--tensor-dim-range",
4462 "35,64",
4463 "--target-shape",
4464 "30,16",
4465 "--target-shape",
4466 "100,23"
4467 ]
4468 ]
4469 },
4470 "tosa-ext-bf16": {
4471 "supports_all": [
4472 "tosa-ext-bf16"
4473 ],
4474 "no_negative_tests": "true",
4475 "negative_dim_range": "1,10",
4476 "generator_args": [
4477 [
4478 "--target-dtype",
4479 "bf16",
4480 "--fp-values-range",
4481 "-max,max",
4482 "--tensor-dim-range",
4483 "25,64"
4484 ],
4485 [
4486 "--target-dtype",
4487 "bf16",
4488 "--fp-values-range",
4489 "-max,max",
4490 "--tensor-dim-range",
4491 "35,64",
4492 "--target-shape",
4493 "30,16",
4494 "--target-shape",
4495 "100,23"
4496 ]
4497 ]
4498 },
4499 "tosa-ext-int16": {
4500 "supports_all": [
4501 "tosa-ext-int16"
4502 ],
4503 "no_negative_tests": "true",
4504 "generator_args": [
4505 [
4506 "--target-dtype",
4507 "int16"
4508 ]
4509 ]
4510 }
4511 },
4512 "selection": {
4513 "default": {
4514 "params": {},
4515 "permutes": [
4516 "shape",
4517 "type"
4518 ]
4519 }
4520 }
4521 },
4522 "gather": {
4523 "group": "scatter_gather",
4524 "generation": {
4525 "tosa-bi": {
4526 "supports_all": [
4527 "tosa-bi"
4528 ],
4529 "generator_args": [
4530 [
4531 "--target-dtype",
4532 "int16",
4533 "--target-dtype",
4534 "int32",
4535 "--tensor-dim-range",
4536 "4,64",
4537 "--max-batch-size",
4538 "64"
4539 ],
4540 [
4541 "--target-dtype",
4542 "int8",
4543 "--tensor-dim-range",
4544 "5,20",
4545 "--target-shape",
4546 "2,65533,1",
4547 "--target-shape",
4548 "1,2,65533",
4549 "--target-shape",
4550 "5000,1,1"
4551 ]
4552 ]
4553 },
4554 "tosa-mi": {
4555 "supports_all": [
4556 "tosa-mi"
4557 ],
4558 "negative_dim_range": "1,10",
4559 "generator_args": [
4560 [
4561 "--target-dtype",
4562 "fp32",
4563 "--target-dtype",
4564 "fp16",
4565 "--fp-values-range",
4566 "-max,max",
4567 "--tensor-dim-range",
4568 "5,72",
4569 "--max-batch-size",
4570 "72"
4571 ],
4572 [
4573 "--target-dtype",
4574 "fp32",
4575 "--fp-values-range",
4576 "-max,max",
4577 "--tensor-dim-range",
4578 "5,20",
4579 "--target-shape",
4580 "2,65536,1",
4581 "--target-shape",
4582 "1,2,65535",
4583 "--target-shape",
4584 "5000,1,1"
4585 ]
4586 ]
4587 },
4588 "tosa-ext-bf16": {
4589 "supports_all": [
4590 "tosa-ext-bf16"
4591 ],
4592 "no_negative_tests": "true",
4593 "negative_dim_range": "1,10",
4594 "generator_args": [
4595 [
4596 "--target-dtype",
4597 "bf16",
4598 "--fp-values-range",
4599 "-max,max",
4600 "--tensor-dim-range",
4601 "5,72",
4602 "--max-batch-size",
4603 "72"
4604 ]
4605 ]
4606 },
4607 "tosa-ext-fp8e4m3": {
4608 "supports_all": [
4609 "tosa-ext-fp8e4m3"
4610 ],
4611 "no_negative_tests": "true",
4612 "from_version": "v0.100.0",
4613 "generator_args": [
4614 [
4615 "--target-dtype",
4616 "fp8e4m3",
4617 "--fp-values-range",
4618 "-max,max",
4619 "--tensor-dim-range",
4620 "32,64",
4621 "--target-rank",
4622 "1",
4623 "--target-rank",
4624 "2",
4625 "--target-rank",
4626 "3",
4627 "--num-rand-permutations",
4628 "2"
4629 ]
4630 ]
4631 },
4632 "tosa-ext-fp8e5m2": {
4633 "supports_all": [
4634 "tosa-ext-fp8e5m2"
4635 ],
4636 "no_negative_tests": "true",
4637 "from_version": "v0.100.0",
4638 "generator_args": [
4639 [
4640 "--target-dtype",
4641 "fp8e5m2",
4642 "--fp-values-range",
4643 "-max,max",
4644 "--tensor-dim-range",
4645 "32,64",
4646 "--target-rank",
4647 "1",
4648 "--target-rank",
4649 "2",
4650 "--target-rank",
4651 "3",
4652 "--num-rand-permutations",
4653 "2"
4654 ]
4655 ]
4656 }
4657 },
4658 "selection": {
4659 "default": {
4660 "params": {},
4661 "permutes": [
4662 "shape",
4663 "type"
4664 ]
4665 }
4666 }
4667 },
4668 "greater": {
4669 "group": "comparison",
4670 "support_for": [
4671 "lazy_data_gen"
4672 ],
4673 "generation": {
4674 "tosa-bi": {
4675 "supports_all": [
4676 "tosa-bi"
4677 ],
4678 "generator_args": [
4679 [
4680 "--target-dtype",
4681 "int32",
4682 "--tensor-dim-range",
4683 "10,64",
4684 "--target-rank",
4685 "1",
4686 "--target-rank",
4687 "2",
4688 "--target-rank",
4689 "3"
4690 ],
4691 [
4692 "--target-dtype",
4693 "int32",
4694 "--tensor-dim-range",
4695 "1,15",
4696 "--target-rank",
4697 "4",
4698 "--target-rank",
4699 "5"
4700 ],
4701 [
4702 "--target-dtype",
4703 "int32",
4704 "--target-shape",
4705 "1,2,65530,1",
4706 "--target-shape",
4707 "1,65539,1,2,1"
4708 ]
4709 ]
4710 },
4711 "8k_level": {
4712 "supports_all": [
4713 "tosa-bi"
4714 ],
4715 "no_negative_tests": "true",
4716 "generator_args": [
4717 [
4718 "--target-dtype",
4719 "int32",
4720 "--tensor-dim-range",
4721 "1,10",
4722 "--target-rank",
4723 "6"
4724 ]
4725 ]
4726 },
4727 "tosa-mi": {
4728 "supports_all": [
4729 "tosa-mi"
4730 ],
4731 "generator_args": [
4732 [
4733 "--target-dtype",
4734 "fp32",
4735 "--target-dtype",
4736 "fp16",
4737 "--fp-values-range",
4738 "-max,max",
4739 "--tensor-dim-range",
4740 "18,64",
4741 "--target-rank",
4742 "1",
4743 "--target-rank",
4744 "2",
4745 "--target-rank",
4746 "3"
4747 ],
4748 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]4749 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]4750 "--target-dtype",
4751 "fp32",
4752 "--fp-values-range",
4753 "-max,max",
4754 "--target-shape",
4755 "3,65551,1,1",
4756 "--target-shape",
4757 "1,1,3,65531,1"
4758 ]
4759 ]
4760 },
4761 "tosa-ext-bf16": {
4762 "supports_all": [
4763 "tosa-ext-bf16"
4764 ],
4765 "no_negative_tests": "true",
4766 "generator_args": [
4767 [
4768 "--target-dtype",
4769 "bf16",
4770 "--fp-values-range",
4771 "-max,max",
4772 "--tensor-dim-range",
4773 "18,64",
4774 "--target-rank",
4775 "1",
4776 "--target-rank",
4777 "2",
4778 "--target-rank",
4779 "3"
4780 ],
4781 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]4782 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]4783 "--target-dtype",
4784 "bf16",
4785 "--fp-values-range",
4786 "-max,max",
4787 "--tensor-dim-range",
4788 "1,18",
4789 "--target-rank",
4790 "4",
4791 "--target-rank",
4792 "5"
4793 ]
4794 ]
4795 }
4796 },
4797 "selection": {
4798 "default": {
4799 "params": {},
4800 "permutes": [
4801 "shape",
4802 "type"
4803 ]
4804 }
4805 }
4806 },
4807 "greater_equal": {
4808 "group": "comparison",
4809 "support_for": [
4810 "lazy_data_gen"
4811 ],
4812 "generation": {
4813 "tosa-bi": {
4814 "supports_all": [
4815 "tosa-bi"
4816 ],
4817 "selector": "true",
4818 "generator_args": [
4819 [
4820 "--target-dtype",
4821 "int32",
4822 "--tensor-dim-range",
4823 "20,64",
4824 "--target-rank",
4825 "1",
4826 "--target-rank",
4827 "2",
4828 "--target-rank",
4829 "3"
4830 ],
4831 [
4832 "--target-dtype",
4833 "int32",
4834 "--tensor-dim-range",
4835 "1,20",
4836 "--target-rank",
4837 "4",
4838 "--target-rank",
4839 "5"
4840 ],
4841 [
4842 "--target-dtype",
4843 "int32",
4844 "--target-shape",
4845 "1,2,65533,1",
4846 "--target-shape",
4847 "1,65539,1,3,1"
4848 ]
4849 ]
4850 },
4851 "8k_level": {
4852 "supports_all": [
4853 "tosa-bi"
4854 ],
4855 "no_negative_tests": "true",
4856 "generator_args": [
4857 [
4858 "--target-dtype",
4859 "int32",
4860 "--tensor-dim-range",
4861 "1,10",
4862 "--target-rank",
4863 "6"
4864 ]
4865 ]
4866 },
4867 "tosa-mi": {
4868 "supports_all": [
4869 "tosa-mi"
4870 ],
4871 "generator_args": [
4872 [
4873 "--target-dtype",
4874 "fp32",
4875 "--target-dtype",
4876 "fp16",
4877 "--fp-values-range",
4878 "-max,max",
4879 "--tensor-dim-range",
4880 "18,64",
4881 "--target-rank",
4882 "1",
4883 "--target-rank",
4884 "2",
4885 "--target-rank",
4886 "3"
4887 ],
4888 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]4889 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]4890 "--target-dtype",
4891 "fp16",
4892 "--fp-values-range",
4893 "-max,max",
4894 "--tensor-dim-range",
4895 "1,18",
4896 "--target-rank",
4897 "4",
4898 "--target-rank",
4899 "5"
4900 ]
4901 ]
4902 },
4903 "tosa-ext-bf16": {
4904 "supports_all": [
4905 "tosa-ext-bf16"
4906 ],
4907 "no_negative_tests": "true",
4908 "generator_args": [
4909 [
4910 "--target-dtype",
4911 "bf16",
4912 "--fp-values-range",
4913 "-max,max",
4914 "--tensor-dim-range",
4915 "18,64",
4916 "--target-rank",
4917 "1",
4918 "--target-rank",
4919 "2",
4920 "--target-rank",
4921 "3"
4922 ],
4923 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]4924 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]4925 "--target-dtype",
4926 "bf16",
4927 "--fp-values-range",
4928 "-max,max",
4929 "--target-shape",
4930 "65552,1,1,4",
4931 "--target-shape",
4932 "1,1,1,3,65532"
4933 ]
4934 ]
4935 }
4936 },
4937 "selection": {
4938 "default": {
4939 "params": {},
4940 "permutes": [
4941 "shape",
4942 "type"
4943 ]
4944 }
4945 }
4946 },
4947 "identity": {
4948 "group": "data_nodes",
4949 "support_for": [
4950 "lazy_data_gen"
4951 ],
4952 "generation": {
4953 "tosa-bi": {
4954 "supports_all": [
4955 "tosa-bi"
4956 ],
4957 "no_negative_tests": "true",
4958 "generator_args": [
4959 [
4960 "--target-dtype",
4961 "int8",
4962 "--target-dtype",
4963 "int16",
4964 "--target-dtype",
4965 "int32",
4966 "--target-dtype",
4967 "int48",
4968 "--target-dtype",
4969 "bool",
4970 "--tensor-dim-range",
4971 "1,61",
4972 "--target-rank",
4973 "1",
4974 "--target-rank",
4975 "2",
4976 "--target-rank",
4977 "3"
4978 ],
4979 [
4980 "--target-dtype",
4981 "int8",
4982 "--target-dtype",
4983 "int16",
4984 "--target-dtype",
4985 "int32",
4986 "--target-dtype",
4987 "int48",
4988 "--target-dtype",
4989 "bool",
4990 "--tensor-dim-range",
4991 "1,15",
4992 "--target-rank",
4993 "4",
4994 "--target-rank",
4995 "5"
4996 ],
4997 [
4998 "--target-dtype",
4999 "int8",
5000 "--target-shape",
5001 "1,1,65536,1,1",
5002 "--target-shape",
5003 "1,65531,1,2"
5004 ]
5005 ]
5006 },
5007 "tosa-mi": {
5008 "supports_all": [
5009 "tosa-mi"
5010 ],
5011 "no_negative_tests": "true",
5012 "generator_args": [
5013 [
5014 "--target-dtype",
5015 "fp32",
5016 "--target-dtype",
5017 "fp16",
5018 "--fp-values-range",
5019 "-max,max",
5020 "--tensor-dim-range",
5021 "1,67",
5022 "--target-rank",
5023 "1",
5024 "--target-rank",
5025 "2",
5026 "--target-rank",
5027 "3"
5028 ],
5029 [
5030 "--target-dtype",
5031 "fp32",
5032 "--target-dtype",
5033 "fp16",
5034 "--fp-values-range",
5035 "-max,max",
5036 "--tensor-dim-range",
5037 "1,19",
5038 "--target-rank",
5039 "4",
5040 "--target-rank",
5041 "5"
5042 ],
5043 [
5044 "--target-dtype",
5045 "fp32",
5046 "--fp-values-range",
5047 "-max,max",
5048 "--target-shape",
5049 "1,1,65539,1,1",
5050 "--target-shape",
5051 "1,1,2,65529"
5052 ]
5053 ]
5054 },
5055 "tosa-ext-bf16": {
5056 "supports_all": [
5057 "tosa-ext-bf16"
5058 ],
5059 "no_negative_tests": "true",
5060 "generator_args": [
5061 [
5062 "--target-dtype",
5063 "bf16",
5064 "--fp-values-range",
5065 "-max,max",
5066 "--tensor-dim-range",
5067 "1,67",
5068 "--target-rank",
5069 "1",
5070 "--target-rank",
5071 "2",
5072 "--target-rank",
5073 "3"
5074 ],
5075 [
5076 "--target-dtype",
5077 "bf16",
5078 "--fp-values-range",
5079 "-max,max",
5080 "--tensor-dim-range",
5081 "1,19",
5082 "--target-rank",
5083 "4",
5084 "--target-rank",
5085 "5"
5086 ]
5087 ]
5088 },
5089 "tosa-ext-int16": {
5090 "supports_all": [
5091 "tosa-ext-int16"
5092 ],
5093 "no_negative_tests": "true",
5094 "generator_args": [
5095 [
5096 "--target-dtype",
5097 "int16",
5098 "--tensor-dim-range",
5099 "1,61",
5100 "--target-rank",
5101 "1",
5102 "--target-rank",
5103 "2",
5104 "--target-rank",
5105 "3"
5106 ],
5107 [
5108 "--target-dtype",
5109 "int16",
5110 "--tensor-dim-range",
5111 "1,15",
5112 "--target-rank",
5113 "4",
5114 "--target-rank",
5115 "5"
5116 ]
5117 ]
5118 },
5119 "tosa-ext-int4": {
5120 "supports_all": [
5121 "tosa-ext-int4"
5122 ],
5123 "no_negative_tests": "true",
5124 "generator_args": [
5125 [
5126 "--target-dtype",
5127 "int4",
5128 "--tensor-dim-range",
5129 "1,61",
5130 "--target-rank",
5131 "1",
5132 "--target-rank",
5133 "2",
5134 "--target-rank",
5135 "3"
5136 ],
5137 [
5138 "--target-dtype",
5139 "int4",
5140 "--tensor-dim-range",
5141 "1,15",
5142 "--target-rank",
5143 "4",
5144 "--target-rank",
5145 "5"
5146 ]
5147 ]
5148 }
5149 },
5150 "selection": {
5151 "default": {
5152 "params": {},
5153 "permutes": [
5154 "shape",
5155 "type"
5156 ]
5157 }
5158 }
5159 },
5160 "intdiv": {
5161 "group": "ew_binary",
5162 "generation": {
5163 "tosa-bi-mi": {
5164 "supports_any": [
5165 "tosa-bi",
5166 "tosa-mi"
5167 ],
5168 "generator_args": [
5169 [
5170 "--target-dtype",
5171 "int32",
5172 "--tensor-dim-range",
5173 "16,64",
5174 "--target-rank",
5175 "1",
5176 "--target-rank",
5177 "2",
5178 "--target-rank",
5179 "3"
5180 ],
5181 [
5182 "--target-dtype",
5183 "int32",
5184 "--tensor-dim-range",
5185 "1,16",
5186 "--target-rank",
5187 "4",
5188 "--target-rank",
5189 "5"
5190 ],
5191 [
5192 "--target-dtype",
5193 "int32",
5194 "--target-shape",
5195 "1,65533,1",
5196 "--target-shape",
5197 "1,1,65539,1,1"
5198 ]
5199 ]
5200 },
5201 "8k_level": {
5202 "supports_all": [
5203 "tosa-bi"
5204 ],
5205 "no_negative_tests": "true",
5206 "generator_args": [
5207 [
5208 "--target-dtype",
5209 "int32",
5210 "--tensor-dim-range",
5211 "1,10",
5212 "--target-rank",
5213 "6"
5214 ]
5215 ]
5216 }
5217 },
5218 "selection": {
5219 "default": {
5220 "params": {},
5221 "permutes": [
5222 "shape",
5223 "type"
5224 ]
5225 }
5226 }
5227 },
5228 "log": {
5229 "group": "ew_unary",
5230 "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ],
5231 "generation": {
5232 "tosa-mi": {
5233 "supports_all": [
5234 "tosa-mi"
5235 ],
5236 "generator_args": [
5237 [
5238 "--target-dtype",
5239 "fp32",
5240 "--target-dtype",
5241 "fp16",
5242 "--fp-values-range",
5243 "-max,max",
5244 "--tensor-dim-range",
5245 "35,64",
5246 "--target-rank",
5247 "0",
5248 "--target-rank",
5249 "1",
5250 "--target-rank",
5251 "2",
5252 "--target-rank",
5253 "3"
5254 ],
5255 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]5256 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]5257 "--target-dtype",
5258 "fp32",
Jeremy Johnson074b98c2024-05-30 10:38:35 +0100[diff] [blame]5259 "--target-dtype",
5260 "fp16",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]5261 "--fp-values-range",
5262 "-max,max",
5263 "--tensor-dim-range",
5264 "1,15",
5265 "--target-rank",
5266 "4",
5267 "--target-rank",
Jeremy Johnson074b98c2024-05-30 10:38:35 +0100[diff] [blame]5268 "5",
5269 "--target-rank",
5270 "6"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]5271 ],
5272 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]5273 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]5274 "--target-dtype",
5275 "fp16",
5276 "--fp-values-range",
5277 "-max,max",
5278 "--target-shape",
5279 "1,65530,1,2",
5280 "--target-shape",
5281 "3,1,65540,1,1"
5282 ]
5283 ]
5284 },
5285 "tosa-ext-bf16": {
5286 "supports_all": [
5287 "tosa-ext-bf16"
5288 ],
5289 "no_negative_tests": "true",
5290 "generator_args": [
5291 [
5292 "--target-dtype",
5293 "bf16",
5294 "--fp-values-range",
5295 "-max,max",
5296 "--tensor-dim-range",
5297 "15,64",
5298 "--target-rank",
5299 "0",
5300 "--target-rank",
5301 "1",
5302 "--target-rank",
5303 "2",
5304 "--target-rank",
5305 "3"
Jeremy Johnson074b98c2024-05-30 10:38:35 +0100[diff] [blame]5306 ],
5307 [
5308 "--no-special-tests",
5309 "--target-dtype",
5310 "bf16",
5311 "--fp-values-range",
5312 "-max,max",
5313 "--tensor-dim-range",
5314 "1,11",
5315 "--target-rank",
5316 "4",
5317 "--target-rank",
5318 "5",
5319 "--target-rank",
5320 "6"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]5321 ]
5322 ]
5323 }
5324 },
5325 "selection": {
5326 "default": {
5327 "generator_select": "true"
5328 }
5329 }
5330 },
5331 "logical_and": {
5332 "group": "ew_binary",
5333 "generation": {
5334 "tosa-bi-mi": {
5335 "supports_any": [
5336 "tosa-bi",
5337 "tosa-mi"
5338 ],
5339 "generator_args": [
5340 [
5341 "--target-dtype",
5342 "bool",
5343 "--tensor-dim-range",
5344 "1,64",
5345 "--target-rank",
5346 "1",
5347 "--target-rank",
5348 "2",
5349 "--target-rank",
5350 "3"
5351 ],
5352 [
5353 "--target-dtype",
5354 "bool",
5355 "--tensor-dim-range",
5356 "1,16",
5357 "--target-rank",
5358 "4",
5359 "--target-rank",
5360 "5"
5361 ],
5362 [
5363 "--target-dtype",
5364 "bool",
5365 "--target-shape",
5366 "1,65538,1",
5367 "--target-shape",
5368 "1,2,1,65536"
5369 ]
5370 ]
5371 },
5372 "8k_level": {
5373 "supports_all": [
5374 "tosa-bi"
5375 ],
5376 "no_negative_tests": "true",
5377 "generator_args": [
5378 [
5379 "--target-dtype",
5380 "bool",
5381 "--tensor-dim-range",
5382 "1,10",
5383 "--target-rank",
5384 "6"
5385 ]
5386 ]
5387 }
5388 },
5389 "selection": {
5390 "default": {
5391 "params": {},
5392 "permutes": [
5393 "shape",
5394 "type"
5395 ]
5396 }
5397 }
5398 },
5399 "logical_left_shift": {
5400 "group": "ew_binary",
5401 "generation": {
5402 "tosa-bi-mi": {
5403 "supports_any": [
5404 "tosa-bi",
5405 "tosa-mi"
5406 ],
5407 "generator_args": [
5408 [
5409 "--target-dtype",
5410 "int8",
5411 "--target-dtype",
5412 "int16",
5413 "--target-dtype",
5414 "int32",
5415 "--tensor-dim-range",
5416 "16,64",
5417 "--target-rank",
5418 "1",
5419 "--target-rank",
5420 "2",
5421 "--target-rank",
5422 "3"
5423 ],
5424 [
5425 "--target-dtype",
5426 "int8",
5427 "--target-dtype",
5428 "int16",
5429 "--target-dtype",
5430 "int32",
5431 "--tensor-dim-range",
5432 "1,16",
5433 "--target-rank",
5434 "4",
5435 "--target-rank",
5436 "5"
5437 ],
5438 [
5439 "--target-dtype",
5440 "int8",
5441 "--target-shape",
5442 "1,65532,1,1",
5443 "--target-shape",
5444 "1,2,1,65538"
5445 ]
5446 ]
5447 },
5448 "8k_level": {
5449 "supports_all": [
5450 "tosa-bi"
5451 ],
5452 "no_negative_tests": "true",
5453 "generator_args": [
5454 [
5455 "--target-dtype",
5456 "int8",
5457 "--target-dtype",
5458 "int16",
5459 "--target-dtype",
5460 "int32",
5461 "--tensor-dim-range",
5462 "1,10",
5463 "--target-rank",
5464 "6"
5465 ]
5466 ]
5467 }
5468 },
5469 "selection": {
5470 "default": {
5471 "params": {},
5472 "permutes": [
5473 "shape",
5474 "type"
5475 ]
5476 }
5477 }
5478 },
5479 "logical_not": {
5480 "group": "ew_unary",
5481 "generation": {
5482 "tosa-bi-mi": {
5483 "supports_any": [
5484 "tosa-bi",
5485 "tosa-mi"
5486 ],
5487 "generator_args": [
5488 [
5489 "--target-dtype",
5490 "bool",
5491 "--tensor-dim-range",
5492 "2,64",
5493 "--target-rank",
5494 "1",
5495 "--target-rank",
5496 "2",
5497 "--target-rank",
5498 "3"
5499 ],
5500 [
5501 "--target-dtype",
5502 "bool",
5503 "--tensor-dim-range",
5504 "1,15",
5505 "--target-rank",
5506 "4",
5507 "--target-rank",
5508 "5"
5509 ],
5510 [
5511 "--target-dtype",
5512 "bool",
5513 "--target-shape",
5514 "1,65537,1",
5515 "--target-shape",
5516 "1,2,1,65535"
5517 ]
5518 ]
5519 },
5520 "8k_level": {
5521 "supports_all": [
5522 "tosa-bi"
5523 ],
5524 "no_negative_tests": "true",
5525 "generator_args": [
5526 [
5527 "--target-dtype",
5528 "bool",
5529 "--tensor-dim-range",
5530 "1,10",
5531 "--target-rank",
5532 "6"
5533 ]
5534 ]
5535 }
5536 },
5537 "selection": {
5538 "default": {
5539 "params": {},
5540 "permutes": [
5541 "shape",
5542 "type"
5543 ]
5544 }
5545 }
5546 },
5547 "logical_or": {
5548 "group": "ew_binary",
5549 "generation": {
5550 "tosa-bi-mi": {
5551 "supports_any": [
5552 "tosa-bi",
5553 "tosa-mi"
5554 ],
5555 "generator_args": [
5556 [
5557 "--target-dtype",
5558 "bool",
5559 "--tensor-dim-range",
5560 "3,64",
5561 "--target-rank",
5562 "1",
5563 "--target-rank",
5564 "2",
5565 "--target-rank",
5566 "3"
5567 ],
5568 [
5569 "--target-dtype",
5570 "bool",
5571 "--tensor-dim-range",
5572 "1,14",
5573 "--target-rank",
5574 "4",
5575 "--target-rank",
5576 "5"
5577 ],
5578 [
5579 "--target-dtype",
5580 "bool",
5581 "--target-shape",
5582 "1,65536,1",
5583 "--target-shape",
5584 "1,2,1,65537"
5585 ]
5586 ]
5587 },
5588 "8k_level": {
5589 "supports_all": [
5590 "tosa-bi"
5591 ],
5592 "no_negative_tests": "true",
5593 "generator_args": [
5594 [
5595 "--target-dtype",
5596 "bool",
5597 "--tensor-dim-range",
5598 "1,10",
5599 "--target-rank",
5600 "6"
5601 ]
5602 ]
5603 }
5604 },
5605 "selection": {
5606 "default": {
5607 "params": {},
5608 "permutes": [
5609 "shape",
5610 "type"
5611 ]
5612 }
5613 }
5614 },
5615 "logical_right_shift": {
5616 "group": "ew_binary",
5617 "generation": {
5618 "tosa-bi-mi": {
5619 "supports_any": [
5620 "tosa-bi",
5621 "tosa-mi"
5622 ],
5623 "generator_args": [
5624 [
5625 "--target-dtype",
5626 "int8",
5627 "--target-dtype",
5628 "int16",
5629 "--target-dtype",
5630 "int32",
5631 "--tensor-dim-range",
5632 "25,64",
5633 "--target-rank",
5634 "1",
5635 "--target-rank",
5636 "2",
5637 "--target-rank",
5638 "3"
5639 ],
5640 [
5641 "--target-dtype",
5642 "int8",
5643 "--target-dtype",
5644 "int16",
5645 "--target-dtype",
5646 "int32",
5647 "--tensor-dim-range",
5648 "1,25",
5649 "--target-rank",
5650 "4",
5651 "--target-rank",
5652 "5"
5653 ],
5654 [
5655 "--target-dtype",
5656 "int8",
5657 "--target-shape",
5658 "1,1,65540,3",
5659 "--target-shape",
5660 "2,1,65530,1,1"
5661 ]
5662 ]
5663 },
5664 "8k_level": {
5665 "supports_all": [
5666 "tosa-bi"
5667 ],
5668 "no_negative_tests": "true",
5669 "generator_args": [
5670 [
5671 "--target-dtype",
5672 "int8",
5673 "--target-dtype",
5674 "int16",
5675 "--target-dtype",
5676 "int32",
5677 "--tensor-dim-range",
5678 "1,10",
5679 "--target-rank",
5680 "6"
5681 ]
5682 ]
5683 }
5684 },
5685 "selection": {
5686 "default": {
5687 "params": {},
5688 "permutes": [
5689 "shape",
5690 "type"
5691 ]
5692 }
5693 }
5694 },
5695 "logical_xor": {
5696 "group": "ew_binary",
5697 "generation": {
5698 "tosa-bi-mi": {
5699 "supports_any": [
5700 "tosa-bi",
5701 "tosa-mi"
5702 ],
5703 "generator_args": [
5704 [
5705 "--target-dtype",
5706 "bool",
5707 "--tensor-dim-range",
5708 "4,64",
5709 "--target-rank",
5710 "1",
5711 "--target-rank",
5712 "2",
5713 "--target-rank",
5714 "3"
5715 ],
5716 [
5717 "--target-dtype",
5718 "bool",
5719 "--tensor-dim-range",
5720 "3,16",
5721 "--target-rank",
5722 "4",
5723 "--target-rank",
5724 "5"
5725 ],
5726 [
5727 "--target-dtype",
5728 "bool",
5729 "--target-shape",
5730 "1,65539,1",
5731 "--target-shape",
5732 "1,2,1,65534"
5733 ]
5734 ]
5735 },
5736 "8k_level": {
5737 "supports_all": [
5738 "tosa-bi"
5739 ],
5740 "no_negative_tests": "true",
5741 "generator_args": [
5742 [
5743 "--target-dtype",
5744 "bool",
5745 "--tensor-dim-range",
5746 "1,10",
5747 "--target-rank",
5748 "6"
5749 ]
5750 ]
5751 }
5752 },
5753 "selection": {
5754 "default": {
5755 "params": {},
5756 "permutes": [
5757 "shape",
5758 "type"
5759 ]
5760 }
5761 }
5762 },
5763 "matmul": {
5764 "group": "tensor",
5765 "support_for": [
5766 "lazy_data_gen"
5767 ],
5768 "generation": {
5769 "tosa-bi": {
5770 "supports_all": [
5771 "tosa-bi"
5772 ],
5773 "selector": "tosa-bi",
5774 "generator_args": [
5775 [
5776 "--target-dtype",
5777 "int8",
5778 "--tensor-dim-range",
5779 "16,64"
5780 ],
5781 [
5782 "--target-dtype",
5783 "int8",
5784 "--tensor-dim-range",
5785 "1,16"
5786 ],
5787 [
5788 "--target-dtype",
5789 "int8",
5790 "--target-shape",
5791 "1,65532,1",
5792 "--target-shape",
5793 "1,2,65538",
5794 "--target-shape",
5795 "2000,1,1"
5796 ]
5797 ]
5798 },
5799 "tosa-mi": {
5800 "supports_all": [
5801 "tosa-mi"
5802 ],
5803 "selector": "tosa-mi",
5804 "negative_dim_range": "1,10",
5805 "generator_args": [
5806 [
5807 "--target-dtype",
5808 "fp32",
5809 "--target-dtype",
5810 "fp16",
5811 "--fp-values-range",
5812 "-2.0,2.0",
5813 "--tensor-dim-range",
5814 "13,70"
5815 ],
5816 [
5817 "--target-dtype",
5818 "fp32",
5819 "--target-dtype",
5820 "fp16",
5821 "--fp-values-range",
5822 "-2.0,2.0",
5823 "--tensor-dim-range",
5824 "10,19"
5825 ],
5826 [
5827 "--target-dtype",
5828 "fp32",
5829 "--fp-values-range",
5830 "-2.0,2.0",
5831 "--target-shape",
5832 "4,65530,1",
5833 "--target-shape",
5834 "1,65538,3"
5835 ]
5836 ]
5837 },
5838 "tosa-ext-bf16": {
5839 "supports_all": [
5840 "tosa-ext-bf16"
5841 ],
5842 "selector": "tosa-mi",
5843 "no_negative_tests": "true",
5844 "negative_dim_range": "1,10",
5845 "generator_args": [
5846 [
5847 "--target-dtype",
5848 "bf16",
5849 "--fp-values-range",
5850 "-2.0,2.0",
5851 "--tensor-dim-range",
5852 "13,70"
5853 ],
5854 [
5855 "--target-dtype",
5856 "bf16",
5857 "--fp-values-range",
5858 "-2.0,2.0",
5859 "--tensor-dim-range",
5860 "10,19"
5861 ]
5862 ]
5863 },
5864 "tosa-ext-fp8e4m3": {
5865 "supports_all": [
5866 "tosa-ext-fp8e4m3"
5867 ],
5868 "selector": "tosa-mi",
5869 "no_negative_tests": "true",
5870 "from_version": "v0.100.0",
5871 "generator_args": [
5872 [
5873 "--target-dtype",
5874 "fp8e4m3",
5875 "--fp-values-range",
5876 "-max,max",
5877 "--tensor-dim-range",
5878 "32,64",
5879 "--target-rank",
5880 "1",
5881 "--target-rank",
5882 "2",
5883 "--target-rank",
5884 "3",
5885 "--num-rand-permutations",
5886 "2"
5887 ]
5888 ]
5889 },
5890 "tosa-ext-fp8e5m2": {
5891 "supports_all": [
5892 "tosa-ext-fp8e5m2"
5893 ],
5894 "selector": "tosa-mi",
5895 "no_negative_tests": "true",
5896 "from_version": "v0.100.0",
5897 "generator_args": [
5898 [
5899 "--target-dtype",
5900 "fp8e5m2",
5901 "--fp-values-range",
5902 "-max,max",
5903 "--tensor-dim-range",
5904 "32,64",
5905 "--target-rank",
5906 "1",
5907 "--target-rank",
5908 "2",
5909 "--target-rank",
5910 "3",
5911 "--num-rand-permutations",
5912 "2"
5913 ]
5914 ]
5915 },
5916 "tosa-ext-int16": {
5917 "supports_all": [
5918 "tosa-ext-int16"
5919 ],
5920 "selector": "tosa-bi",
5921 "no_negative_tests": "true",
5922 "generator_args": [
5923 [
5924 "--target-dtype",
5925 "int16",
5926 "--tensor-dim-range",
5927 "16,64"
5928 ],
5929 [
5930 "--target-dtype",
5931 "int16",
5932 "--tensor-dim-range",
5933 "1,16"
5934 ]
5935 ]
5936 }
5937 },
5938 "selection": {
5939 "tosa-bi": {
5940 "default": {
5941 "params": {},
5942 "permutes": [
5943 "shape",
5944 "type"
5945 ]
5946 }
5947 },
5948 "tosa-mi": {
5949 "all": "true"
5950 }
5951 }
5952 },
5953 "maximum": {
5954 "group": "ew_binary",
5955 "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ],
5956 "generation": {
5957 "tosa-bi": {
5958 "supports_all": [
5959 "tosa-bi"
5960 ],
5961 "generator_args": [
5962 [
5963 "--target-dtype",
5964 "int32",
5965 "--tensor-dim-range",
5966 "16,64",
5967 "--target-rank",
5968 "1",
5969 "--target-rank",
5970 "2",
5971 "--target-rank",
5972 "3"
5973 ],
5974 [
5975 "--target-dtype",
5976 "int32",
5977 "--tensor-dim-range",
5978 "1,16",
5979 "--target-rank",
5980 "4",
5981 "--target-rank",
5982 "5"
5983 ],
5984 [
5985 "--target-dtype",
5986 "int32",
5987 "--target-shape",
5988 "1,2,65533,1",
5989 "--target-shape",
5990 "1,65539,1,1,1"
5991 ]
5992 ]
5993 },
5994 "8k_level": {
5995 "supports_all": [
5996 "tosa-bi"
5997 ],
5998 "no_negative_tests": "true",
5999 "generator_args": [
6000 [
6001 "--target-dtype",
6002 "int32",
6003 "--tensor-dim-range",
6004 "1,10",
6005 "--target-rank",
6006 "6"
6007 ]
6008 ]
6009 },
6010 "tosa-mi": {
6011 "supports_all": [
6012 "tosa-mi"
6013 ],
6014 "selector": "tosa-mi",
6015 "generator_args": [
6016 [
6017 "--target-dtype",
6018 "fp32",
6019 "--target-dtype",
6020 "fp16",
6021 "--fp-values-range",
6022 "-max,max",
6023 "--tensor-dim-range",
6024 "16,64",
6025 "--target-rank",
6026 "0",
6027 "--target-rank",
6028 "1",
6029 "--target-rank",
6030 "2",
6031 "--target-rank",
6032 "3"
6033 ],
6034 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]6035 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]6036 "--target-dtype",
6037 "fp32",
6038 "--target-dtype",
6039 "fp16",
6040 "--fp-values-range",
6041 "-max,max",
6042 "--tensor-dim-range",
6043 "1,16",
6044 "--target-rank",
6045 "4",
6046 "--target-rank",
6047 "5"
6048 ],
6049 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]6050 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]6051 "--target-dtype",
6052 "fp32",
6053 "--fp-values-range",
6054 "-max,max",
6055 "--target-shape",
6056 "65531,1,2,1",
6057 "--target-shape",
6058 "1,1,65541,1"
6059 ]
6060 ]
6061 },
6062 "tosa-ext-bf16": {
6063 "supports_all": [
6064 "tosa-ext-bf16"
6065 ],
6066 "selector": "tosa-mi",
6067 "no_negative_tests": "true",
6068 "generator_args": [
6069 [
6070 "--target-dtype",
6071 "bf16",
6072 "--fp-values-range",
6073 "-max,max",
6074 "--tensor-dim-range",
6075 "16,64",
6076 "--target-rank",
6077 "0",
6078 "--target-rank",
6079 "1",
6080 "--target-rank",
6081 "2",
6082 "--target-rank",
6083 "3"
6084 ],
6085 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]6086 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]6087 "--target-dtype",
6088 "bf16",
6089 "--fp-values-range",
6090 "-max,max",
6091 "--tensor-dim-range",
6092 "1,16",
6093 "--target-rank",
6094 "4",
6095 "--target-rank",
6096 "5"
6097 ]
6098 ]
6099 }
6100 },
6101 "selection": {
6102 "default": {
6103 "params": {},
6104 "permutes": [
6105 "shape",
6106 "type"
6107 ]
6108 },
6109 "tosa-mi": {
6110 "generator_select": "true"
6111 }
6112 }
6113 },
6114 "max_pool2d": {
6115 "group": "tensor",
6116 "support_for": [
6117 "lazy_data_gen"
6118 ],
6119 "generation": {
6120 "tosa-bi": {
6121 "supports_all": [
6122 "tosa-bi"
6123 ],
6124 "selector": "tosa-bi",
6125 "generator_args": [
6126 [
6127 "--target-dtype",
6128 "int8",
6129 "--target-shape",
6130 "1,4,75,3",
6131 "--target-shape",
6132 "1,11,44,1",
6133 "--allow-pooling-and-conv-oversizes"
6134 ],
6135 [
6136 "--target-dtype",
6137 "int8",
6138 "--target-shape",
6139 "1,65535,5,1",
6140 "--target-shape",
6141 "1,3,65537,1",
6142 "--target-shape",
6143 "1,3,2,1",
6144 "--allow-pooling-and-conv-oversizes"
6145 ]
6146 ]
6147 },
6148 "8k_level": {
6149 "supports_all": [
6150 "tosa-bi"
6151 ],
6152 "no_negative_tests": "true",
6153 "selector": "8k_level",
6154 "generator_args": [
6155 [
6156 "--target-dtype",
6157 "int8",
6158 "--tensor-dim-range",
6159 "3,10",
6160 "--level-8k-sizes"
6161 ]
6162 ]
6163 },
6164 "tosa-mi": {
6165 "supports_all": [
6166 "tosa-mi"
6167 ],
6168 "selector": "tosa-mi",
6169 "generator_args": [
6170 [
6171 "--target-dtype",
6172 "fp32",
6173 "--target-dtype",
6174 "fp16",
6175 "--fp-values-range",
6176 "-max,max",
6177 "--target-shape",
6178 "1,47,37,25",
6179 "--target-shape",
6180 "1,15,13,38",
6181 "--allow-pooling-and-conv-oversizes"
6182 ],
6183 [
6184 "--target-dtype",
6185 "fp32",
6186 "--fp-values-range",
6187 "-max,max",
6188 "--target-shape",
6189 "1,65534,4,1",
6190 "--target-shape",
6191 "1,6,65536,1",
6192 "--max-conv-dilation",
6193 "1",
6194 "--allow-pooling-and-conv-oversizes"
6195 ]
6196 ]
6197 },
6198 "tosa-ext-bf16": {
6199 "supports_all": [
6200 "tosa-ext-bf16"
6201 ],
6202 "selector": "tosa-mi",
6203 "no_negative_tests": "true",
6204 "generator_args": [
6205 [
6206 "--target-dtype",
6207 "bf16",
6208 "--fp-values-range",
6209 "-max,max",
6210 "--target-shape",
6211 "1,47,37,25",
6212 "--target-shape",
6213 "1,15,13,38",
6214 "--allow-pooling-and-conv-oversizes"
6215 ]
6216 ]
6217 },
6218 "tosa-ext-fp8e4m3": {
6219 "supports_all": [
6220 "tosa-ext-fp8e4m3"
6221 ],
6222 "selector": "tosa-mi",
6223 "no_negative_tests": "true",
6224 "from_version": "v0.100.0",
6225 "generator_args": [
6226 [
6227 "--target-dtype",
6228 "fp8e4m3",
6229 "--fp-values-range",
6230 "-max,max",
6231 "--tensor-dim-range",
6232 "32,64",
6233 "--num-rand-permutations",
6234 "2"
6235 ]
6236 ]
6237 },
6238 "tosa-ext-fp8e5m2": {
6239 "supports_all": [
6240 "tosa-ext-fp8e5m2"
6241 ],
6242 "selector": "tosa-mi",
6243 "no_negative_tests": "true",
6244 "from_version": "v0.100.0",
6245 "generator_args": [
6246 [
6247 "--target-dtype",
6248 "fp8e5m2",
6249 "--fp-values-range",
6250 "-max,max",
6251 "--tensor-dim-range",
6252 "32,64",
6253 "--num-rand-permutations",
6254 "2"
6255 ]
6256 ]
6257 },
6258 "tosa-ext-int16": {
6259 "supports_all": [
6260 "tosa-ext-int16"
6261 ],
6262 "selector": "tosa-bi",
6263 "no_negative_tests": "true",
6264 "generator_args": [
6265 [
6266 "--target-dtype",
6267 "int16",
6268 "--target-shape",
6269 "1,4,75,3",
6270 "--target-shape",
6271 "1,11,44,1",
6272 "--allow-pooling-and-conv-oversizes"
6273 ]
6274 ]
6275 },
6276 "8k_level_ext_int16": {
6277 "supports_all": [
6278 "tosa-ext-int16"
6279 ],
6280 "no_negative_tests": "true",
6281 "selector": "8k_level",
6282 "generator_args": [
6283 [
6284 "--target-dtype",
6285 "int16",
6286 "--tensor-dim-range",
6287 "3,10",
6288 "--level-8k-sizes"
6289 ]
6290 ]
6291 }
6292 },
6293 "selection": {
6294 "tosa-bi": {
6295 "params": {
6296 "shape": [],
6297 "type": [
6298 "i8",
6299 "i16"
6300 ],
6301 "kernel": [],
6302 "stride": [],
6303 "pad": []
6304 },
6305 "permutes": [
6306 "shape",
6307 "type",
6308 "pad"
6309 ]
6310 },
6311 "8k_level": {
6312 "all": "true"
6313 },
6314 "tosa-mi": {
6315 "params": {
6316 "shape": [],
6317 "type": [
6318 "f32",
6319 "f16",
6320 "bf16"
6321 ],
6322 "stride": [],
6323 "pad": []
6324 },
6325 "permutes": [
6326 "shape",
6327 "type",
6328 "pad"
6329 ],
6330 "sparsity": {
6331 "pad": 11
6332 }
6333 }
6334 }
6335 },
6336 "minimum": {
6337 "group": "ew_binary",
6338 "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ],
6339 "generation": {
6340 "tosa-bi": {
6341 "supports_all": [
6342 "tosa-bi"
6343 ],
6344 "generator_args": [
6345 [
6346 "--target-dtype",
6347 "int32",
6348 "--tensor-dim-range",
6349 "16,64",
6350 "--target-rank",
6351 "1",
6352 "--target-rank",
6353 "2",
6354 "--target-rank",
6355 "3"
6356 ],
6357 [
6358 "--target-dtype",
6359 "int32",
6360 "--tensor-dim-range",
6361 "1,16",
6362 "--target-rank",
6363 "4",
6364 "--target-rank",
6365 "5"
6366 ],
6367 [
6368 "--target-dtype",
6369 "int32",
6370 "--target-shape",
6371 "1,2,65533,1",
6372 "--target-shape",
6373 "1,65539,1,1,1"
6374 ]
6375 ]
6376 },
6377 "8k_level": {
6378 "supports_all": [
6379 "tosa-bi"
6380 ],
6381 "no_negative_tests": "true",
6382 "generator_args": [
6383 [
6384 "--target-dtype",
6385 "int32",
6386 "--tensor-dim-range",
6387 "1,10",
6388 "--target-rank",
6389 "6"
6390 ]
6391 ]
6392 },
6393 "tosa-mi": {
6394 "supports_all": [
6395 "tosa-mi"
6396 ],
6397 "selector": "tosa-mi",
6398 "generator_args": [
6399 [
6400 "--target-dtype",
6401 "fp32",
6402 "--target-dtype",
6403 "fp16",
6404 "--fp-values-range",
6405 "-max,max",
6406 "--tensor-dim-range",
6407 "16,64",
6408 "--target-rank",
6409 "0",
6410 "--target-rank",
6411 "1",
6412 "--target-rank",
6413 "2",
6414 "--target-rank",
6415 "3"
6416 ],
6417 [
6418 "--target-dtype",
6419 "fp32",
6420 "--target-dtype",
6421 "fp16",
6422 "--fp-values-range",
6423 "-max,max",
6424 "--tensor-dim-range",
6425 "1,16",
6426 "--target-rank",
6427 "4",
6428 "--target-rank",
6429 "5"
6430 ],
6431 [
6432 "--target-dtype",
6433 "fp32",
6434 "--fp-values-range",
6435 "-max,max",
6436 "--target-shape",
6437 "1,65531,2,1",
6438 "--target-shape",
6439 "1,1,1,65538"
6440 ]
6441 ]
6442 },
6443 "tosa-ext-bf16": {
6444 "supports_all": [
6445 "tosa-ext-bf16"
6446 ],
6447 "selector": "tosa-mi",
6448 "no_negative_tests": "true",
6449 "generator_args": [
6450 [
6451 "--target-dtype",
6452 "bf16",
6453 "--fp-values-range",
6454 "-max,max",
6455 "--tensor-dim-range",
6456 "16,64",
6457 "--target-rank",
6458 "0",
6459 "--target-rank",
6460 "1",
6461 "--target-rank",
6462 "2",
6463 "--target-rank",
6464 "3"
6465 ],
6466 [
6467 "--target-dtype",
6468 "bf16",
6469 "--fp-values-range",
6470 "-max,max",
6471 "--tensor-dim-range",
6472 "1,16",
6473 "--target-rank",
6474 "4",
6475 "--target-rank",
6476 "5"
6477 ]
6478 ]
6479 }
6480 },
6481 "selection": {
6482 "default": {
6483 "params": {},
6484 "permutes": [
6485 "shape",
6486 "type"
6487 ]
6488 },
6489 "tosa-mi": {
6490 "generator_select": "true"
6491 }
6492 }
6493 },
6494 "mul": {
6495 "group": "ew_binary",
6496 "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ],
6497 "generation": {
6498 "tosa-bi-mi": {
6499 "supports_any": [
6500 "tosa-bi",
6501 "tosa-mi"
6502 ],
6503 "selector": "tosa-bi",
6504 "no_negative_tests": "true",
6505 "generator_args": [
6506 [
6507 "--target-dtype",
6508 "int32",
6509 "--tensor-dim-range",
6510 "16,64",
6511 "--target-rank",
6512 "1",
6513 "--target-rank",
6514 "2",
6515 "--target-rank",
6516 "3"
6517 ],
6518 [
6519 "--target-dtype",
6520 "int32",
6521 "--tensor-dim-range",
6522 "1,16",
6523 "--target-rank",
6524 "4",
6525 "--target-rank",
6526 "5"
6527 ]
6528 ]
6529 },
6530 "tosa-bi": {
6531 "supports_all": [
6532 "tosa-bi"
6533 ],
6534 "selector": "tosa-bi",
6535 "generator_args": [
6536 [
6537 "--target-dtype",
6538 "int8",
6539 "--target-dtype",
6540 "int16",
6541 "--tensor-dim-range",
6542 "16,64",
6543 "--target-rank",
6544 "1",
6545 "--target-rank",
6546 "2",
6547 "--target-rank",
6548 "3"
6549 ],
6550 [
6551 "--target-dtype",
6552 "int8",
6553 "--target-dtype",
6554 "int16",
6555 "--tensor-dim-range",
6556 "1,16",
6557 "--target-rank",
6558 "4",
6559 "--target-rank",
6560 "5"
6561 ],
6562 [
6563 "--target-dtype",
6564 "int8",
6565 "--target-shape",
6566 "1,2,65500,1",
6567 "--target-shape",
6568 "1,68539,1,1,1"
6569 ]
6570 ]
6571 },
6572 "8k_level": {
6573 "supports_all": [
6574 "tosa-bi"
6575 ],
6576 "selector": "tosa-bi",
6577 "no_negative_tests": "true",
6578 "generator_args": [
6579 [
6580 "--target-dtype",
6581 "int8",
6582 "--target-dtype",
6583 "int16",
6584 "--tensor-dim-range",
6585 "1,10",
6586 "--target-rank",
6587 "6"
6588 ]
6589 ]
6590 },
6591 "8k_level_bi_mi": {
6592 "supports_any": [
6593 "tosa-bi",
6594 "tosa-mi"
6595 ],
6596 "selector": "tosa-bi",
6597 "no_negative_tests": "true",
6598 "generator_args": [
6599 [
6600 "--target-dtype",
6601 "int32",
6602 "--tensor-dim-range",
6603 "1,10",
6604 "--target-rank",
6605 "6"
6606 ]
6607 ]
6608 },
6609 "tosa-mi": {
6610 "supports_all": [
6611 "tosa-mi"
6612 ],
6613 "selector": "tosa-mi",
6614 "negative_dim_range": "1,10",
6615 "generator_args": [
6616 [
6617 "--target-dtype",
6618 "fp32",
6619 "--target-dtype",
6620 "fp16",
6621 "--fp-values-range",
6622 "-max,max",
6623 "--tensor-dim-range",
6624 "16,64",
6625 "--target-rank",
6626 "0",
6627 "--target-rank",
6628 "1",
6629 "--target-rank",
6630 "2",
6631 "--target-rank",
6632 "3"
6633 ],
6634 [
6635 "--target-dtype",
6636 "fp16",
6637 "--fp-values-range",
6638 "-max,max",
6639 "--tensor-dim-range",
6640 "1,16",
6641 "--target-rank",
6642 "4",
6643 "--target-rank",
6644 "5"
6645 ]
6646 ]
6647 },
6648 "tosa-ext-bf16": {
6649 "supports_all": [
6650 "tosa-ext-bf16"
6651 ],
6652 "selector": "tosa-mi",
6653 "no_negative_tests": "true",
6654 "negative_dim_range": "1,10",
6655 "generator_args": [
6656 [
6657 "--target-dtype",
6658 "bf16",
6659 "--fp-values-range",
6660 "-max,max",
6661 "--tensor-dim-range",
6662 "16,64",
6663 "--target-rank",
6664 "0",
6665 "--target-rank",
6666 "1",
6667 "--target-rank",
6668 "2",
6669 "--target-rank",
6670 "3"
6671 ],
6672 [
6673 "--target-dtype",
6674 "bf16",
6675 "--fp-values-range",
6676 "-max,max",
6677 "--target-shape",
6678 "1,1,3,65534",
6679 "--target-shape",
6680 "1,2,1,65541,1"
6681 ]
6682 ]
6683 }
6684 },
6685 "selection": {
6686 "tosa-bi": {
6687 "params": {},
6688 "permutes": [
6689 "shape",
6690 "type",
6691 "perm",
6692 "shift"
6693 ]
6694 },
6695 "tosa-mi": {
6696 "generator_select": "true"
6697 }
6698 }
6699 },
6700 "mul_shape": {
6701 "group": "shape",
6702 "generation": {
6703 "standard": {
6704 "supports_any": [
6705 "tosa-bi",
6706 "tosa-mi"
6707 ],
6708 "generator_args": [
6709 [
6710 "--target-dtype",
6711 "shape",
6712 "--tensor-dim-range",
6713 "1,16",
6714 "--target-rank",
6715 "1"
6716 ]
6717 ]
6718 }
6719 },
6720 "selection": {
6721 "default": {
6722 "params": {},
6723 "permutes": [
6724 "shape"
6725 ]
6726 }
6727 }
6728 },
6729 "negate": {
6730 "group": "ew_unary",
6731 "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ],
6732 "generation": {
6733 "tosa-bi": {
6734 "supports_all": [
6735 "tosa-bi"
6736 ],
6737 "generator_args": [
6738 [
6739 "--target-dtype",
6740 "int8",
6741 "--target-dtype",
6742 "int16",
6743 "--target-dtype",
6744 "int32",
6745 "--tensor-dim-range",
6746 "32,64",
6747 "--target-rank",
6748 "1",
6749 "--target-rank",
6750 "2",
6751 "--target-rank",
6752 "3"
6753 ],
6754 [
6755 "--target-dtype",
6756 "int8",
6757 "--target-dtype",
6758 "int16",
6759 "--target-dtype",
6760 "int32",
6761 "--tensor-dim-range",
6762 "1,22",
6763 "--target-rank",
6764 "4",
6765 "--target-rank",
6766 "5"
6767 ],
6768 [
6769 "--target-dtype",
6770 "int8",
6771 "--target-shape",
6772 "3,1,65540,1",
6773 "--target-shape",
6774 "1,1,2,1,65531"
6775 ]
6776 ]
6777 },
6778 "8k_level": {
6779 "supports_all": [
6780 "tosa-bi"
6781 ],
6782 "no_negative_tests": "true",
6783 "generator_args": [
6784 [
6785 "--target-dtype",
6786 "int8",
6787 "--target-dtype",
6788 "int16",
6789 "--target-dtype",
6790 "int32",
6791 "--tensor-dim-range",
6792 "1,10",
6793 "--target-rank",
6794 "6"
6795 ]
6796 ]
6797 },
6798 "tosa-mi": {
6799 "supports_all": [
6800 "tosa-mi"
6801 ],
6802 "selector": "tosa-mi",
6803 "negative_dim_range": "1,10",
6804 "generator_args": [
6805 [
6806 "--target-dtype",
6807 "fp32",
6808 "--target-dtype",
6809 "fp16",
6810 "--fp-values-range",
6811 "-max,max",
6812 "--tensor-dim-range",
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]6813 "37,64",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]6814 "--target-rank",
6815 "0",
6816 "--target-rank",
6817 "1",
6818 "--target-rank",
6819 "2",
6820 "--target-rank",
6821 "3"
6822 ],
6823 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]6824 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]6825 "--target-dtype",
6826 "fp32",
6827 "--target-dtype",
6828 "fp16",
6829 "--fp-values-range",
6830 "-max,max",
6831 "--tensor-dim-range",
Jeremy Johnson074b98c2024-05-30 10:38:35 +0100[diff] [blame]6832 "1,16",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]6833 "--target-rank",
6834 "4",
6835 "--target-rank",
Jeremy Johnson074b98c2024-05-30 10:38:35 +0100[diff] [blame]6836 "5",
6837 "--target-rank",
6838 "6"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]6839 ],
6840 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]6841 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]6842 "--target-dtype",
6843 "fp32",
6844 "--fp-values-range",
6845 "-max,max",
6846 "--target-shape",
6847 "1,65541,3,1",
6848 "--target-shape",
6849 "1,2,1,65532,1"
6850 ]
6851 ]
6852 },
6853 "tosa-ext-bf16": {
6854 "supports_all": [
6855 "tosa-ext-bf16"
6856 ],
6857 "selector": "tosa-mi",
6858 "no_negative_tests": "true",
6859 "generator_args": [
6860 [
6861 "--target-dtype",
6862 "bf16",
6863 "--fp-values-range",
6864 "-max,max",
6865 "--tensor-dim-range",
6866 "17,64",
6867 "--target-rank",
6868 "0",
6869 "--target-rank",
6870 "1",
6871 "--target-rank",
6872 "2",
6873 "--target-rank",
6874 "3"
6875 ],
6876 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]6877 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]6878 "--target-dtype",
6879 "bf16",
6880 "--fp-values-range",
6881 "-max,max",
6882 "--tensor-dim-range",
Jeremy Johnson074b98c2024-05-30 10:38:35 +0100[diff] [blame]6883 "1,15",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]6884 "--target-rank",
6885 "4",
6886 "--target-rank",
Jeremy Johnson074b98c2024-05-30 10:38:35 +0100[diff] [blame]6887 "5",
6888 "--target-rank",
6889 "6"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]6890 ]
6891 ]
6892 }
6893 },
6894 "selection": {
6895 "default": {
6896 "params": {},
6897 "permutes": [
6898 "shape",
6899 "type"
6900 ]
6901 },
6902 "tosa-mi": {
6903 "generator_select": "true"
6904 }
6905 }
6906 },
6907 "pad": {
6908 "group": "data_layout",
6909 "support_for": [
6910 "lazy_data_gen",
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]6911 "random_const_inputs",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]6912 "stable_random_gen"
6913 ],
6914 "generation": {
6915 "tosa-bi-mi": {
6916 "supports_any": [
6917 "tosa-bi",
6918 "tosa-mi"
6919 ],
6920 "selector": "tosa-bi",
6921 "generator_args": [
6922 [
6923 "--target-dtype",
6924 "bool",
6925 "--tensor-dim-range",
6926 "1,64",
6927 "--target-rank",
6928 "1",
6929 "--target-rank",
6930 "2",
6931 "--target-rank",
6932 "3"
6933 ],
6934 [
6935 "--target-dtype",
6936 "bool",
6937 "--tensor-dim-range",
6938 "1,16",
6939 "--target-rank",
6940 "4"
6941 ]
6942 ]
6943 },
6944 "tosa-bi": {
6945 "supports_all": [
6946 "tosa-bi"
6947 ],
6948 "selector": "tosa-bi",
6949 "no_negative_tests": "true",
6950 "generator_args": [
6951 [
6952 "--target-dtype",
6953 "int8",
6954 "--target-dtype",
6955 "int16",
6956 "--target-dtype",
6957 "int32",
6958 "--tensor-dim-range",
6959 "1,64",
6960 "--target-rank",
6961 "1",
6962 "--target-rank",
6963 "2",
6964 "--target-rank",
6965 "3"
6966 ],
6967 [
6968 "--target-dtype",
6969 "int8",
6970 "--target-dtype",
6971 "int16",
6972 "--target-dtype",
6973 "int32",
6974 "--tensor-dim-range",
6975 "1,16",
6976 "--target-rank",
6977 "4"
6978 ],
6979 [
6980 "--target-dtype",
6981 "int16",
6982 "--tensor-dim-range",
6983 "1,16",
6984 "--target-rank",
6985 "5"
6986 ],
6987 [
6988 "--target-dtype",
6989 "int8",
6990 "--target-shape",
6991 "1,1,65537,1"
6992 ]
6993 ]
6994 },
6995 "8k_level": {
6996 "supports_all": [
6997 "tosa-bi"
6998 ],
6999 "no_negative_tests": "true",
7000 "selector": "8k_level",
7001 "generator_args": [
7002 [
7003 "--target-dtype",
7004 "int8",
7005 "--tensor-dim-range",
7006 "1,10",
7007 "--target-rank",
7008 "6"
7009 ]
7010 ]
7011 },
7012 "tosa-mi": {
7013 "supports_all": [
7014 "tosa-mi"
7015 ],
7016 "selector": "tosa-mi",
7017 "no_negative_tests": "true",
7018 "generator_args": [
7019 [
7020 "--target-dtype",
7021 "fp32",
7022 "--target-dtype",
7023 "fp16",
7024 "--fp-values-range",
7025 "-max,max",
7026 "--tensor-dim-range",
7027 "1,65",
7028 "--target-rank",
7029 "1",
7030 "--target-rank",
7031 "2",
7032 "--target-rank",
7033 "3"
7034 ],
7035 [
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]7036 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]7037 "--target-dtype",
7038 "fp32",
7039 "--target-dtype",
7040 "fp16",
7041 "--fp-values-range",
7042 "-max,max",
7043 "--tensor-dim-range",
7044 "1,11",
7045 "--target-rank",
7046 "4",
7047 "--target-rank",
7048 "5",
7049 "--target-rank",
7050 "6"
7051 ]
7052 ]
7053 },
7054 "tosa-ext-bf16": {
7055 "supports_all": [
7056 "tosa-ext-bf16"
7057 ],
7058 "selector": "tosa-mi",
7059 "no_negative_tests": "true",
7060 "generator_args": [
7061 [
7062 "--target-dtype",
7063 "bf16",
7064 "--fp-values-range",
7065 "-max,max",
7066 "--tensor-dim-range",
7067 "1,65",
7068 "--target-rank",
7069 "1",
7070 "--target-rank",
7071 "2",
7072 "--target-rank",
7073 "3"
7074 ],
7075 [
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]7076 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]7077 "--target-dtype",
7078 "bf16",
7079 "--fp-values-range",
7080 "-max,max",
7081 "--tensor-dim-range",
7082 "1,11",
7083 "--target-rank",
7084 "4",
7085 "--target-rank",
7086 "5",
7087 "--target-rank",
7088 "6"
7089 ]
7090 ]
7091 },
7092 "tosa-ext-fp8e4m3": {
7093 "supports_all": [
7094 "tosa-ext-fp8e4m3"
7095 ],
7096 "selector": "tosa-mi",
7097 "no_negative_tests": "true",
7098 "from_version": "v0.100.0",
7099 "generator_args": [
7100 [
7101 "--target-dtype",
7102 "fp8e4m3",
7103 "--fp-values-range",
7104 "-max,max",
7105 "--tensor-dim-range",
7106 "32,64",
7107 "--target-rank",
7108 "1",
7109 "--target-rank",
7110 "2",
7111 "--target-rank",
7112 "3",
7113 "--num-rand-permutations",
7114 "2"
7115 ]
7116 ]
7117 },
7118 "tosa-ext-fp8e5m2": {
7119 "supports_all": [
7120 "tosa-ext-fp8e5m2"
7121 ],
7122 "selector": "tosa-mi",
7123 "no_negative_tests": "true",
7124 "from_version": "v0.100.0",
7125 "generator_args": [
7126 [
7127 "--target-dtype",
7128 "fp8e5m2",
7129 "--fp-values-range",
7130 "-max,max",
7131 "--tensor-dim-range",
7132 "32,64",
7133 "--target-rank",
7134 "1",
7135 "--target-rank",
7136 "2",
7137 "--target-rank",
7138 "3",
7139 "--num-rand-permutations",
7140 "2"
7141 ]
7142 ]
7143 },
7144 "extra": {
7145 "supports_all": [
7146 "tosa-mi"
7147 ],
7148 "selector": "minimal",
7149 "no_negative_tests": "true",
7150 "generator_args": [
7151 [
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]7152 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]7153 "--target-dtype",
7154 "fp32",
7155 "--target-dtype",
7156 "fp16",
7157 "--fp-values-range",
7158 "-max,max",
7159 "--target-shape",
7160 "1,1,65539,1"
7161 ],
7162 [
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]7163 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]7164 "--target-dtype",
7165 "fp32",
7166 "--target-dtype",
7167 "fp16",
7168 "--fp-values-range",
7169 "-max,max",
7170 "--target-rank",
7171 "1",
7172 "--target-rank",
7173 "2",
7174 "--target-rank",
7175 "3",
7176 "--target-rank",
7177 "4",
7178 "--oversize"
7179 ]
7180 ]
7181 },
7182 "extra-ext-bf16": {
7183 "supports_all": [
7184 "tosa-ext-bf16"
7185 ],
7186 "selector": "minimal",
7187 "no_negative_tests": "true",
7188 "generator_args": [
7189 [
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]7190 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]7191 "--target-dtype",
7192 "bf16",
7193 "--fp-values-range",
7194 "-max,max",
7195 "--target-shape",
7196 "1,1,65539,1"
7197 ],
7198 [
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]7199 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]7200 "--target-dtype",
7201 "bf16",
7202 "--fp-values-range",
7203 "-max,max",
7204 "--target-rank",
7205 "1",
7206 "--target-rank",
7207 "2",
7208 "--target-rank",
7209 "3",
7210 "--target-rank",
7211 "4",
7212 "--oversize"
7213 ]
7214 ]
7215 }
7216 },
7217 "selection": {
7218 "tosa-bi": {
7219 "params": {},
7220 "permutes": [
7221 "shape",
7222 "type"
7223 ],
7224 "sparsity": {
7225 "pad": 15
7226 },
7227 "preselected": [
7228 {
7229 "shape": "6",
7230 "type": "i8",
7231 "pad": "pad00"
7232 }
7233 ]
7234 },
7235 "8k_level": {
7236 "params": {},
7237 "permutes": [],
7238 "sparsity": {
7239 "pad": 3
7240 },
7241 "full_sparsity": "true"
7242 },
7243 "tosa-mi": {
7244 "generator_select": "true",
7245 "maximum": 5,
7246 "specifics": {
7247 "pad": [
7248 "[[0 0]]",
7249 "[[0 0], [0 0]]",
7250 "[[0 0], [0 0], [0 0]]",
7251 "[[0 0], [0 0], [0 0], [0 0]]",
7252 "[[0 0], [0 0], [0 0], [0 0], [0 0]]",
7253 "[[0 0], [0 0], [0 0], [0 0], [0 0], [0 0]]"
7254 ]
7255 }
7256 },
7257 "minimal": {
7258 "generator_select": "true",
7259 "maximum": 2
7260 }
7261 }
7262 },
7263 "pow": {
7264 "group": "ew_binary",
7265 "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ],
7266 "generation": {
7267 "tosa-mi": {
7268 "supports_all": [
7269 "tosa-mi"
7270 ],
7271 "generator_args": [
7272 [
7273 "--target-dtype",
7274 "fp32",
7275 "--target-dtype",
7276 "fp16",
7277 "--fp-values-range",
7278 "-max,max",
7279 "--tensor-dim-range",
7280 "16,64",
7281 "--target-rank",
7282 "0",
7283 "--target-rank",
7284 "1",
7285 "--target-rank",
7286 "2",
7287 "--target-rank",
7288 "3"
7289 ],
7290 [
7291 "--target-dtype",
7292 "fp16",
7293 "--fp-values-range",
7294 "-max,max",
7295 "--tensor-dim-range",
7296 "1,16",
7297 "--target-rank",
7298 "4",
7299 "--target-rank",
7300 "5"
7301 ]
7302 ]
7303 },
7304 "tosa-ext-bf16": {
7305 "supports_all": [
7306 "tosa-ext-bf16"
7307 ],
7308 "no_negative_tests": "true",
7309 "generator_args": [
7310 [
7311 "--target-dtype",
7312 "bf16",
7313 "--fp-values-range",
7314 "-max,max",
7315 "--tensor-dim-range",
7316 "16,64",
7317 "--target-rank",
7318 "0",
7319 "--target-rank",
7320 "1",
7321 "--target-rank",
7322 "2",
7323 "--target-rank",
7324 "3"
7325 ],
7326 [
7327 "--target-dtype",
7328 "bf16",
7329 "--fp-values-range",
7330 "-max,max",
7331 "--target-shape",
7332 "65534,3,1,1",
7333 "--target-shape",
7334 "2,65541,1,1,1"
7335 ]
7336 ]
7337 }
7338 },
7339 "selection": {
7340 "default": {
7341 "generator_select": "true"
7342 }
7343 }
7344 },
7345 "reciprocal": {
7346 "group": "ew_unary",
7347 "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ],
7348 "generation": {
7349 "tosa-mi": {
7350 "supports_all": [
7351 "tosa-mi"
7352 ],
7353 "generator_args": [
7354 [
7355 "--target-dtype",
7356 "fp32",
7357 "--target-dtype",
7358 "fp16",
7359 "--fp-values-range",
7360 "-max,max",
7361 "--tensor-dim-range",
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]7362 "38,62",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]7363 "--target-rank",
7364 "0",
7365 "--target-rank",
7366 "1",
7367 "--target-rank",
7368 "2",
7369 "--target-rank",
7370 "3"
7371 ],
7372 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]7373 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]7374 "--target-dtype",
7375 "fp16",
Jeremy Johnson074b98c2024-05-30 10:38:35 +0100[diff] [blame]7376 "--target-dtype",
7377 "fp32",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]7378 "--fp-values-range",
7379 "-max,max",
7380 "--tensor-dim-range",
7381 "1,15",
7382 "--target-rank",
7383 "4",
7384 "--target-rank",
Jeremy Johnson074b98c2024-05-30 10:38:35 +0100[diff] [blame]7385 "5",
7386 "--target-rank",
7387 "6"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]7388 ]
7389 ]
7390 },
7391 "tosa-ext-bf16": {
7392 "supports_all": [
7393 "tosa-ext-bf16"
7394 ],
7395 "no_negative_tests": "true",
7396 "generator_args": [
7397 [
7398 "--target-dtype",
7399 "bf16",
7400 "--fp-values-range",
7401 "-max,max",
7402 "--tensor-dim-range",
7403 "15,64",
7404 "--target-rank",
7405 "0",
7406 "--target-rank",
7407 "1",
7408 "--target-rank",
7409 "2",
7410 "--target-rank",
7411 "3"
7412 ],
7413 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]7414 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]7415 "--target-dtype",
7416 "bf16",
7417 "--fp-values-range",
7418 "-max,max",
Jeremy Johnson074b98c2024-05-30 10:38:35 +0100[diff] [blame]7419 "--tensor-dim-range",
7420 "1,11",
7421 "--target-rank",
7422 "4",
7423 "--target-rank",
7424 "5",
7425 "--target-rank",
7426 "6"
7427 ],
7428 [
7429 "--no-special-tests",
7430 "--target-dtype",
7431 "bf16",
7432 "--fp-values-range",
7433 "-max,max",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]7434 "--target-shape",
7435 "1,1,65536,4",
7436 "--target-shape",
7437 "2,65535,1,1,1"
7438 ]
7439 ]
7440 }
7441 },
7442 "selection": {
7443 "default": {
7444 "generator_select": "true"
7445 }
7446 }
7447 },
7448 "reduce_all": {
7449 "group": "reduction",
7450 "generation": {
7451 "tosa-bi-mi": {
7452 "supports_any": [
7453 "tosa-bi",
7454 "tosa-mi"
7455 ],
7456 "generator_args": [
7457 [
7458 "--target-dtype",
7459 "bool",
7460 "--tensor-dim-range",
7461 "1,32",
7462 "--target-rank",
7463 "1",
7464 "--target-rank",
7465 "2",
7466 "--target-rank",
7467 "3",
7468 "--target-rank",
7469 "4"
7470 ],
7471 [
7472 "--target-dtype",
7473 "bool",
7474 "--target-shape",
7475 "65530,1,1,1",
7476 "--target-shape",
7477 "2,65538,1,1"
7478 ]
7479 ]
7480 },
7481 "8k_level": {
7482 "supports_all": [
7483 "tosa-bi"
7484 ],
7485 "from_version": "v0.80.0",
7486 "no_negative_tests": "true",
7487 "generator_args": [
7488 [
7489 "--target-dtype",
7490 "bool",
7491 "--tensor-dim-range",
7492 "1,10",
7493 "--target-rank",
7494 "5",
7495 "--target-rank",
7496 "6"
7497 ]
7498 ]
7499 }
7500 },
7501 "selection": {
7502 "default": {
7503 "params": {},
7504 "permutes": [
7505 "shape",
7506 "type",
7507 "axis"
7508 ]
7509 }
7510 }
7511 },
7512 "reduce_any": {
7513 "group": "reduction",
7514 "generation": {
7515 "tosa-bi-mi": {
7516 "supports_any": [
7517 "tosa-bi",
7518 "tosa-mi"
7519 ],
7520 "generator_args": [
7521 [
7522 "--target-dtype",
7523 "bool",
7524 "--tensor-dim-range",
7525 "1,32",
7526 "--target-rank",
7527 "1",
7528 "--target-rank",
7529 "2",
7530 "--target-rank",
7531 "3",
7532 "--target-rank",
7533 "4"
7534 ],
7535 [
7536 "--target-dtype",
7537 "bool",
7538 "--target-shape",
7539 "65530,1,1,1",
7540 "--target-shape",
7541 "2,65538,1,1"
7542 ]
7543 ]
7544 },
7545 "8k_level": {
7546 "supports_all": [
7547 "tosa-bi"
7548 ],
7549 "from_version": "v0.80.0",
7550 "no_negative_tests": "true",
7551 "generator_args": [
7552 [
7553 "--target-dtype",
7554 "bool",
7555 "--tensor-dim-range",
7556 "1,9",
7557 "--target-rank",
7558 "5",
7559 "--target-rank",
7560 "6"
7561 ]
7562 ]
7563 }
7564 },
7565 "selection": {
7566 "default": {
7567 "params": {},
7568 "permutes": [
7569 "shape",
7570 "type",
7571 "axis"
7572 ]
7573 }
7574 }
7575 },
7576 "reduce_max": {
7577 "group": "reduction",
7578 "support_for": [
7579 "lazy_data_gen"
7580 ],
7581 "generation": {
7582 "tosa-bi": {
7583 "supports_all": [
7584 "tosa-bi"
7585 ],
7586 "generator_args": [
7587 [
7588 "--target-dtype",
7589 "int8",
7590 "--target-dtype",
7591 "int16",
7592 "--target-dtype",
7593 "int32",
7594 "--tensor-dim-range",
7595 "1,32",
7596 "--target-rank",
7597 "1",
7598 "--target-rank",
7599 "2",
7600 "--target-rank",
7601 "3",
7602 "--target-rank",
7603 "4"
7604 ],
7605 [
7606 "--target-dtype",
7607 "int8",
7608 "--target-shape",
7609 "65530,1,1,1",
7610 "--target-shape",
7611 "2,65538,1,1"
7612 ]
7613 ]
7614 },
7615 "8k_level": {
7616 "supports_all": [
7617 "tosa-bi"
7618 ],
7619 "from_version": "v0.80.0",
7620 "no_negative_tests": "true",
7621 "generator_args": [
7622 [
7623 "--target-dtype",
7624 "int8",
7625 "--target-dtype",
7626 "int16",
7627 "--target-dtype",
7628 "int32",
7629 "--tensor-dim-range",
7630 "1,11",
7631 "--target-rank",
7632 "5",
7633 "--target-rank",
7634 "6"
7635 ]
7636 ]
7637 },
7638 "tosa-mi": {
7639 "supports_all": [
7640 "tosa-mi"
7641 ],
7642 "negative_dim_range": "1,10",
7643 "generator_args": [
7644 [
7645 "--target-dtype",
7646 "fp32",
7647 "--target-dtype",
7648 "fp16",
7649 "--fp-values-range",
7650 "-max,max",
7651 "--tensor-dim-range",
7652 "1,32",
7653 "--target-rank",
7654 "1",
7655 "--target-rank",
7656 "2",
7657 "--target-rank",
7658 "3",
7659 "--target-rank",
7660 "4"
7661 ],
7662 [
7663 "--target-dtype",
7664 "fp32",
7665 "--fp-values-range",
7666 "-max,max",
7667 "--target-shape",
7668 "1,1,1,65531",
7669 "--target-shape",
7670 "1,1,65539,2"
7671 ]
7672 ]
7673 },
7674 "tosa-ext-bf16": {
7675 "supports_all": [
7676 "tosa-ext-bf16"
7677 ],
7678 "no_negative_tests": "true",
7679 "negative_dim_range": "1,10",
7680 "generator_args": [
7681 [
7682 "--target-dtype",
7683 "bf16",
7684 "--fp-values-range",
7685 "-max,max",
7686 "--tensor-dim-range",
7687 "1,32",
7688 "--target-rank",
7689 "1",
7690 "--target-rank",
7691 "2",
7692 "--target-rank",
7693 "3",
7694 "--target-rank",
7695 "4"
7696 ]
7697 ]
7698 },
7699 "8k_level_mi": {
7700 "supports_all": [
7701 "tosa-mi"
7702 ],
7703 "from_version": "v0.80.0",
7704 "no_negative_tests": "true",
7705 "generator_args": [
7706 [
7707 "--target-dtype",
7708 "fp32",
7709 "--target-dtype",
7710 "fp16",
7711 "--fp-values-range",
7712 "-max,max",
7713 "--tensor-dim-range",
7714 "1,10",
7715 "--target-rank",
7716 "5",
7717 "--target-rank",
7718 "6"
7719 ]
7720 ]
7721 },
7722 "8k_level_ext_bf16": {
7723 "supports_all": [
7724 "tosa-ext-bf16"
7725 ],
7726 "from_version": "v0.80.0",
7727 "no_negative_tests": "true",
7728 "generator_args": [
7729 [
7730 "--target-dtype",
7731 "bf16",
7732 "--fp-values-range",
7733 "-max,max",
7734 "--tensor-dim-range",
7735 "1,10",
7736 "--target-rank",
7737 "5",
7738 "--target-rank",
7739 "6"
7740 ]
7741 ]
7742 }
7743 },
7744 "selection": {
7745 "default": {
7746 "params": {},
7747 "permutes": [
7748 "shape",
7749 "type",
7750 "axis"
7751 ]
7752 }
7753 }
7754 },
7755 "reduce_min": {
7756 "group": "reduction",
7757 "support_for": [
7758 "lazy_data_gen"
7759 ],
7760 "generation": {
7761 "tosa-bi": {
7762 "supports_all": [
7763 "tosa-bi"
7764 ],
7765 "generator_args": [
7766 [
7767 "--target-dtype",
7768 "int8",
7769 "--target-dtype",
7770 "int16",
7771 "--target-dtype",
7772 "int32",
7773 "--tensor-dim-range",
7774 "1,24",
7775 "--target-rank",
7776 "1",
7777 "--target-rank",
7778 "2",
7779 "--target-rank",
7780 "3",
7781 "--target-rank",
7782 "4"
7783 ],
7784 [
7785 "--target-dtype",
7786 "int8",
7787 "--target-shape",
7788 "1,1,65538,1",
7789 "--target-shape",
7790 "2,65535,1,1"
7791 ]
7792 ]
7793 },
7794 "8k_level": {
7795 "supports_all": [
7796 "tosa-bi"
7797 ],
7798 "from_version": "v0.80.0",
7799 "no_negative_tests": "true",
7800 "generator_args": [
7801 [
7802 "--target-dtype",
7803 "int8",
7804 "--target-dtype",
7805 "int16",
7806 "--target-dtype",
7807 "int32",
7808 "--tensor-dim-range",
7809 "2,10",
7810 "--target-rank",
7811 "5",
7812 "--target-rank",
7813 "6"
7814 ]
7815 ]
7816 },
7817 "tosa-mi": {
7818 "supports_all": [
7819 "tosa-mi"
7820 ],
7821 "negative_dim_range": "1,10",
7822 "generator_args": [
7823 [
7824 "--target-dtype",
7825 "fp32",
7826 "--target-dtype",
7827 "fp16",
7828 "--fp-values-range",
7829 "-max,max",
7830 "--tensor-dim-range",
7831 "1,32",
7832 "--target-rank",
7833 "1",
7834 "--target-rank",
7835 "2",
7836 "--target-rank",
7837 "3",
7838 "--target-rank",
7839 "4"
7840 ],
7841 [
7842 "--target-dtype",
7843 "fp32",
7844 "--fp-values-range",
7845 "-max,max",
7846 "--target-shape",
7847 "1,1,65531,1",
7848 "--target-shape",
7849 "1,65539,2,1"
7850 ]
7851 ]
7852 },
7853 "tosa-ext-bf16": {
7854 "supports_all": [
7855 "tosa-ext-bf16"
7856 ],
7857 "no_negative_tests": "true",
7858 "negative_dim_range": "1,10",
7859 "generator_args": [
7860 [
7861 "--target-dtype",
7862 "bf16",
7863 "--fp-values-range",
7864 "-max,max",
7865 "--tensor-dim-range",
7866 "1,32",
7867 "--target-rank",
7868 "1",
7869 "--target-rank",
7870 "2",
7871 "--target-rank",
7872 "3",
7873 "--target-rank",
7874 "4"
7875 ]
7876 ]
7877 },
7878 "8k_level_mi": {
7879 "supports_all": [
7880 "tosa-mi"
7881 ],
7882 "from_version": "v0.80.0",
7883 "no_negative_tests": "true",
7884 "generator_args": [
7885 [
7886 "--target-dtype",
7887 "fp32",
7888 "--target-dtype",
7889 "fp16",
7890 "--fp-values-range",
7891 "-max,max",
7892 "--tensor-dim-range",
7893 "1,9",
7894 "--target-rank",
7895 "5",
7896 "--target-rank",
7897 "6"
7898 ]
7899 ]
7900 },
7901 "8k_level_ext_bf16": {
7902 "supports_all": [
7903 "tosa-ext-bf16"
7904 ],
7905 "from_version": "v0.80.0",
7906 "no_negative_tests": "true",
7907 "generator_args": [
7908 [
7909 "--target-dtype",
7910 "bf16",
7911 "--fp-values-range",
7912 "-max,max",
7913 "--tensor-dim-range",
7914 "1,9",
7915 "--target-rank",
7916 "5",
7917 "--target-rank",
7918 "6"
7919 ]
7920 ]
7921 }
7922 },
7923 "selection": {
7924 "default": {
7925 "params": {},
7926 "permutes": [
7927 "shape",
7928 "type",
7929 "axis"
7930 ]
7931 }
7932 }
7933 },
7934 "reduce_product": {
7935 "group": "reduction",
7936 "support_for": [
7937 "lazy_data_gen"
7938 ],
7939 "generation": {
7940 "tosa-mi": {
7941 "supports_all": [
7942 "tosa-mi"
7943 ],
7944 "generator_args": [
7945 [
7946 "--target-dtype",
7947 "fp32",
7948 "--target-dtype",
7949 "fp16",
7950 "--fp-values-range",
7951 "-max,max",
7952 "--tensor-dim-range",
7953 "1,34",
7954 "--target-rank",
7955 "1",
7956 "--target-rank",
7957 "2",
7958 "--target-rank",
7959 "3",
7960 "--target-rank",
7961 "4"
7962 ],
7963 [
7964 "--target-dtype",
7965 "fp16",
7966 "--fp-values-range",
7967 "-max,max",
7968 "--target-shape",
7969 "2,65527,3,1",
7970 "--target-shape",
7971 "1,3,2,65541"
7972 ]
7973 ]
7974 },
7975 "tosa-ext-bf16": {
7976 "supports_all": [
7977 "tosa-ext-bf16"
7978 ],
7979 "no_negative_tests": "true",
7980 "generator_args": [
7981 [
7982 "--target-dtype",
7983 "bf16",
7984 "--fp-values-range",
7985 "-max,max",
7986 "--tensor-dim-range",
7987 "1,34",
7988 "--target-rank",
7989 "1",
7990 "--target-rank",
7991 "2",
7992 "--target-rank",
7993 "3",
7994 "--target-rank",
7995 "4"
7996 ]
7997 ]
7998 },
7999 "8k_level": {
8000 "supports_all": [
8001 "tosa-mi"
8002 ],
8003 "from_version": "v0.80.0",
8004 "no_negative_tests": "true",
8005 "generator_args": [
8006 [
8007 "--target-dtype",
8008 "fp32",
8009 "--target-dtype",
8010 "fp16",
8011 "--target-dtype",
8012 "bf16",
8013 "--fp-values-range",
8014 "-max,max",
8015 "--tensor-dim-range",
8016 "1,11",
8017 "--target-rank",
8018 "5",
8019 "--target-rank",
8020 "6"
8021 ]
8022 ]
8023 }
8024 },
8025 "selection": {
8026 "default": {
8027 "params": {},
8028 "permutes": [
8029 "shape",
8030 "type",
8031 "axis"
8032 ]
8033 }
8034 }
8035 },
8036 "reduce_sum": {
8037 "group": "reduction",
8038 "support_for": [
8039 "lazy_data_gen"
8040 ],
8041 "generation": {
8042 "tosa-bi": {
8043 "supports_all": [
8044 "tosa-bi"
8045 ],
8046 "generator_args": [
8047 [
8048 "--target-dtype",
8049 "int32",
8050 "--tensor-dim-range",
8051 "1,32",
8052 "--target-rank",
8053 "1",
8054 "--target-rank",
8055 "2",
8056 "--target-rank",
8057 "3",
8058 "--target-rank",
8059 "4"
8060 ],
8061 [
8062 "--target-dtype",
8063 "int32",
8064 "--target-shape",
8065 "3,65532,1,1",
8066 "--target-shape",
8067 "65536,1,1,2"
8068 ]
8069 ]
8070 },
8071 "8k_level": {
8072 "supports_all": [
8073 "tosa-bi"
8074 ],
8075 "from_version": "v0.80.0",
8076 "no_negative_tests": "true",
8077 "generator_args": [
8078 [
8079 "--target-dtype",
8080 "int32",
8081 "--tensor-dim-range",
8082 "1,12",
8083 "--target-rank",
8084 "5",
8085 "--target-rank",
8086 "6"
8087 ]
8088 ]
8089 },
8090 "tosa-mi": {
8091 "supports_all": [
8092 "tosa-mi"
8093 ],
8094 "generator_args": [
8095 [
8096 "--target-dtype",
8097 "fp32",
8098 "--target-dtype",
8099 "fp16",
8100 "--fp-values-range",
8101 "-max,max",
8102 "--target-rank",
8103 "3",
8104 "--target-rank",
8105 "4",
8106 "--tensor-dim-range",
8107 "10,37",
8108 "--target-rank",
8109 "1",
8110 "--target-rank",
8111 "2",
8112 "--target-rank",
8113 "3",
8114 "--target-rank",
8115 "4"
8116 ],
8117 [
8118 "--target-dtype",
8119 "fp32",
8120 "--fp-values-range",
8121 "-max,max",
8122 "--target-shape",
8123 "1001",
8124 "--target-shape",
8125 "50,200"
8126 ],
8127 [
8128 "--target-dtype",
8129 "fp16",
8130 "--fp-values-range",
8131 "-max,max",
8132 "--target-shape",
8133 "65529",
8134 "--target-shape",
8135 "65537,2"
8136 ]
8137 ]
8138 },
8139 "tosa-ext-bf16": {
8140 "supports_all": [
8141 "tosa-ext-bf16"
8142 ],
8143 "no_negative_tests": "true",
8144 "generator_args": [
8145 [
8146 "--target-dtype",
8147 "bf16",
8148 "--fp-values-range",
8149 "-max,max",
8150 "--target-rank",
8151 "3",
8152 "--target-rank",
8153 "4",
8154 "--tensor-dim-range",
8155 "10,37",
8156 "--target-rank",
8157 "1",
8158 "--target-rank",
8159 "2",
8160 "--target-rank",
8161 "3",
8162 "--target-rank",
8163 "4"
8164 ],
8165 [
8166 "--target-dtype",
8167 "bf16",
8168 "--fp-values-range",
8169 "-max,max",
8170 "--target-shape",
8171 "16001",
8172 "--target-shape",
8173 "16,1000",
8174 "--target-shape",
8175 "1,3,65529,1",
8176 "--target-shape",
8177 "1,65537,1,1"
8178 ]
8179 ]
8180 },
8181 "8k_level_mi": {
8182 "supports_all": [
8183 "tosa-mi"
8184 ],
8185 "from_version": "v0.80.0",
8186 "no_negative_tests": "true",
8187 "generator_args": [
8188 [
8189 "--target-dtype",
8190 "fp32",
8191 "--target-dtype",
8192 "fp16",
8193 "--fp-values-range",
8194 "-max,max",
8195 "--tensor-dim-range",
8196 "5,12",
8197 "--target-rank",
8198 "5",
8199 "--target-rank",
8200 "6"
8201 ]
8202 ]
8203 },
8204 "8k_level_ext_bf16": {
8205 "supports_all": [
8206 "tosa-ext-bf16"
8207 ],
8208 "from_version": "v0.80.0",
8209 "no_negative_tests": "true",
8210 "generator_args": [
8211 [
8212 "--target-dtype",
8213 "bf16",
8214 "--fp-values-range",
8215 "-max,max",
8216 "--tensor-dim-range",
8217 "5,12",
8218 "--target-rank",
8219 "5",
8220 "--target-rank",
8221 "6"
8222 ]
8223 ]
8224 }
8225 },
8226 "selection": {
8227 "default": {
8228 "params": {},
8229 "permutes": [
8230 "shape",
8231 "type",
8232 "axis"
8233 ]
8234 }
8235 }
8236 },
8237 "rescale": {
8238 "group": "type_conversion",
8239 "generation": {
8240 "tosa-bi": {
8241 "supports_all": [
8242 "tosa-bi"
8243 ],
8244 "generator_args": [
8245 [
8246 "--target-dtype",
8247 "int8",
8248 "--target-dtype",
8249 "int16",
8250 "--target-dtype",
8251 "int32",
8252 "--target-dtype",
8253 "uint8",
8254 "--target-dtype",
8255 "uint16",
8256 "--tensor-dim-range",
8257 "16,32",
8258 "--target-rank",
8259 "1",
8260 "--target-rank",
8261 "2",
8262 "--target-rank",
8263 "3"
8264 ],
8265 [
8266 "--target-dtype",
8267 "int8",
8268 "--target-dtype",
8269 "int16",
8270 "--target-dtype",
8271 "int32",
8272 "--target-dtype",
8273 "uint8",
8274 "--target-dtype",
8275 "uint16",
8276 "--tensor-dim-range",
8277 "1,16",
8278 "--target-rank",
8279 "4",
8280 "--target-rank",
8281 "5"
8282 ],
8283 [
8284 "--target-dtype",
8285 "int8",
8286 "--target-shape",
8287 "1,2,65533,1",
8288 "--target-shape",
8289 "1,65539,1,1"
8290 ]
8291 ]
8292 },
8293 "8k_level": {
8294 "supports_all": [
8295 "tosa-bi"
8296 ],
8297 "no_negative_tests": "true",
8298 "generator_args": [
8299 [
8300 "--target-dtype",
8301 "int8",
8302 "--target-dtype",
8303 "int16",
8304 "--target-dtype",
8305 "int32",
8306 "--target-dtype",
8307 "uint8",
8308 "--target-dtype",
8309 "uint16",
8310 "--tensor-dim-range",
8311 "1,10",
8312 "--target-rank",
8313 "6"
8314 ]
8315 ]
8316 },
8317 "tosa-ext-int16": {
8318 "supports_all": [
8319 "tosa-ext-int16"
8320 ],
8321 "no_negative_tests": "true",
8322 "generator_args": [
8323 [
8324 "--target-dtype",
8325 "int48",
8326 "--tensor-dim-range",
8327 "16,32",
8328 "--target-rank",
8329 "1",
8330 "--target-rank",
8331 "2",
8332 "--target-rank",
8333 "3"
8334 ],
8335 [
8336 "--target-dtype",
8337 "int48",
8338 "--tensor-dim-range",
8339 "1,16",
8340 "--target-rank",
8341 "4",
8342 "--target-rank",
8343 "5"
8344 ]
8345 ]
8346 },
8347 "8k_level_ext_int16": {
8348 "supports_all": [
8349 "tosa-ext-int16"
8350 ],
8351 "no_negative_tests": "true",
8352 "generator_args": [
8353 [
8354 "--target-dtype",
8355 "int48",
8356 "--tensor-dim-range",
8357 "1,10",
8358 "--target-rank",
8359 "6"
8360 ]
8361 ]
8362 }
8363 },
8364 "selection": {
8365 "default": {
8366 "params": {
8367 "scale": [
8368 "sc0",
8369 "sc0",
8370 "sc1",
8371 "sc1",
8372 "sc1",
8373 "sc1"
8374 ],
8375 "double_round": [
8376 "dr0",
8377 "dr0",
8378 "dr0",
8379 "dr0",
8380 "dr1",
8381 "dr1"
8382 ],
8383 "per_channel": [
8384 "pc0",
8385 "pc1",
8386 "pc0",
8387 "pc1",
8388 "pc0",
8389 "pc1",
8390 "pc1"
8391 ]
8392 },
8393 "permutes": [
8394 "shape",
8395 "type",
8396 "output_type"
8397 ]
8398 }
8399 }
8400 },
8401 "reshape": {
8402 "group": "data_layout",
8403 "support_for": [
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]8404 "lazy_data_gen",
8405 "random_const_inputs",
8406 "stable_random_gen"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]8407 ],
8408 "generation": {
8409 "tosa-bi-mi": {
8410 "supports_any": [
8411 "tosa-bi",
8412 "tosa-mi"
8413 ],
8414 "generator_args": [
8415 [
8416 "--target-dtype",
8417 "bool",
8418 "--target-rank",
8419 "1",
8420 "--target-rank",
8421 "2",
8422 "--target-rank",
8423 "3",
8424 "--target-rank",
8425 "4",
8426 "--num-rand-permutations",
8427 "2"
8428 ]
8429 ]
8430 },
8431 "tosa-bi": {
8432 "supports_all": [
8433 "tosa-bi"
8434 ],
8435 "generator_args": [
8436 [
8437 "--target-dtype",
8438 "int8",
8439 "--target-dtype",
8440 "int16",
8441 "--target-dtype",
8442 "int32",
8443 "--target-dtype",
8444 "bool",
8445 "--target-rank",
8446 "1",
8447 "--target-rank",
8448 "2",
8449 "--target-rank",
8450 "3",
8451 "--target-rank",
8452 "4",
8453 "--num-rand-permutations",
8454 "2"
8455 ],
8456 [
8457 "--target-dtype",
8458 "int8",
8459 "--target-shape",
8460 "1,1,65535,1",
8461 "--target-shape",
8462 "1,65538,1,2",
8463 "--num-rand-permutations",
8464 "2"
8465 ]
8466 ]
8467 },
8468 "8k_level": {
8469 "supports_all": [
8470 "tosa-bi"
8471 ],
8472 "no_negative_tests": "true",
8473 "generator_args": [
8474 [
8475 "--target-dtype",
8476 "int8",
8477 "--target-dtype",
8478 "int16",
8479 "--target-dtype",
8480 "int32",
8481 "--tensor-dim-range",
8482 "1,10",
8483 "--target-rank",
8484 "5",
8485 "--target-rank",
8486 "6",
8487 "--num-rand-permutations",
8488 "2"
8489 ]
8490 ]
8491 },
8492 "8k_level-bi-mi": {
8493 "supports_all": [
8494 "tosa-bi"
8495 ],
8496 "no_negative_tests": "true",
8497 "generator_args": [
8498 [
8499 "--target-dtype",
8500 "bool",
8501 "--tensor-dim-range",
8502 "1,10",
8503 "--target-rank",
8504 "5",
8505 "--target-rank",
8506 "6",
8507 "--num-rand-permutations",
8508 "2"
8509 ]
8510 ]
8511 },
8512 "tosa-mi": {
8513 "supports_all": [
8514 "tosa-mi"
8515 ],
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]8516 "selector": "tosa-mi",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]8517 "generator_args": [
8518 [
8519 "--target-dtype",
8520 "fp32",
8521 "--target-dtype",
8522 "fp16",
8523 "--fp-values-range",
8524 "-max,max",
8525 "--target-rank",
8526 "1",
8527 "--target-rank",
8528 "2",
8529 "--target-rank",
8530 "3",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]8531 "--num-rand-permutations",
8532 "2"
8533 ],
8534 [
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]8535 "--no-special-tests",
8536 "--target-dtype",
8537 "fp32",
8538 "--target-dtype",
8539 "fp16",
8540 "--fp-values-range",
8541 "-max,max",
8542 "--tensor-dim-range",
8543 "1,10",
8544 "--target-rank",
8545 "4",
8546 "--target-rank",
8547 "5",
8548 "--target-rank",
8549 "6",
8550 "--num-rand-permutations",
8551 "1"
8552 ],
8553 [
8554 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]8555 "--target-dtype",
8556 "fp32",
8557 "--fp-values-range",
8558 "-max,max",
8559 "--target-shape",
8560 "1,1,65536,1",
8561 "--target-shape",
8562 "1,65535,1,2",
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]8563 "--target-shape",
8564 "7,13,11",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]8565 "--num-rand-permutations",
8566 "2"
8567 ]
8568 ]
8569 },
8570 "tosa-ext-bf16": {
8571 "supports_all": [
8572 "tosa-ext-bf16"
8573 ],
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]8574 "selector": "tosa-mi",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]8575 "no_negative_tests": "true",
8576 "generator_args": [
8577 [
8578 "--target-dtype",
8579 "bf16",
8580 "--fp-values-range",
8581 "-max,max",
8582 "--target-rank",
8583 "1",
8584 "--target-rank",
8585 "2",
8586 "--target-rank",
8587 "3",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]8588 "--num-rand-permutations",
8589 "2"
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]8590 ],
8591 [
8592 "--no-special-tests",
8593 "--target-dtype",
8594 "bf16",
8595 "--fp-values-range",
8596 "-max,max",
8597 "--tensor-dim-range",
8598 "1,10",
8599 "--target-rank",
8600 "4",
8601 "--target-rank",
8602 "5",
8603 "--target-rank",
8604 "6",
8605 "--num-rand-permutations",
8606 "1"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]8607 ]
8608 ]
8609 },
8610 "tosa-ext-fp8e4m3": {
8611 "supports_all": [
8612 "tosa-ext-fp8e4m3"
8613 ],
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]8614 "selector": "tosa-mi",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]8615 "no_negative_tests": "true",
8616 "from_version": "v0.100.0",
8617 "generator_args": [
8618 [
8619 "--target-dtype",
8620 "fp8e4m3",
8621 "--fp-values-range",
8622 "-max,max",
8623 "--tensor-dim-range",
8624 "32,64",
8625 "--target-rank",
8626 "1",
8627 "--target-rank",
8628 "2",
8629 "--target-rank",
8630 "3",
8631 "--num-rand-permutations",
8632 "2"
8633 ]
8634 ]
8635 },
8636 "tosa-ext-fp8e5m2": {
8637 "supports_all": [
8638 "tosa-ext-fp8e5m2"
8639 ],
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]8640 "selector": "tosa-mi",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]8641 "no_negative_tests": "true",
8642 "from_version": "v0.100.0",
8643 "generator_args": [
8644 [
8645 "--target-dtype",
8646 "fp8e5m2",
8647 "--fp-values-range",
8648 "-max,max",
8649 "--tensor-dim-range",
8650 "32,64",
8651 "--target-rank",
8652 "1",
8653 "--target-rank",
8654 "2",
8655 "--target-rank",
8656 "3",
8657 "--num-rand-permutations",
8658 "2"
8659 ]
8660 ]
8661 }
8662 },
8663 "selection": {
8664 "default": {
8665 "params": {},
8666 "permutes": [
8667 "shape",
8668 "type",
8669 "rank",
8670 "perm"
8671 ]
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]8672 },
8673 "tosa-mi": {
8674 "generator_select": true
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]8675 }
8676 }
8677 },
8678 "resize": {
8679 "group": "image",
8680 "support_for": [
8681 "lazy_data_gen"
8682 ],
8683 "generation": {
8684 "tosa-bi": {
8685 "supports_all": [
8686 "tosa-bi"
8687 ],
8688 "selector": "tosa-bi",
8689 "generator_args": [
8690 [
8691 "--target-dtype",
8692 "int8"
8693 ],
8694 [
8695 "--target-dtype",
8696 "int8",
8697 "--target-shape",
8698 "1,3,16383,1",
8699 "--max-resize-output-dim",
8700 "9500"
8701 ]
8702 ]
8703 },
8704 "8k_level": {
8705 "supports_all": [
8706 "tosa-bi"
8707 ],
8708 "selector": "tosa-bi",
8709 "no_negative_tests": "true",
8710 "generator_args": [
8711 [
8712 "--target-dtype",
8713 "int8",
8714 "--level-8k-sizes",
8715 "--tensor-dim-range",
8716 "2,10",
8717 "--num-rand-permutations",
8718 "2"
8719 ]
8720 ]
8721 },
8722 "8k_level_ext_int16": {
8723 "supports_all": [
8724 "tosa-ext-int16"
8725 ],
8726 "selector": "tosa-bi",
8727 "no_negative_tests": "true",
8728 "generator_args": [
8729 [
8730 "--target-dtype",
8731 "int16",
8732 "--level-8k-sizes",
8733 "--tensor-dim-range",
8734 "2,10",
8735 "--num-rand-permutations",
8736 "2"
8737 ]
8738 ]
8739 },
8740 "tosa-mi": {
8741 "supports_all": [
8742 "tosa-mi"
8743 ],
8744 "selector": "tosa-mi",
8745 "negative_dim_range": "1,10",
8746 "generator_args": [
8747 [
8748 "--target-dtype",
8749 "fp32",
8750 "--target-dtype",
8751 "fp16",
8752 "--fp-values-range",
8753 "-max,max"
8754 ],
8755 [
8756 "--target-dtype",
8757 "fp32",
8758 "--fp-values-range",
8759 "-max,max",
8760 "--target-shape",
8761 "1,1103,1,2",
8762 "--max-resize-output-dim",
8763 "9100"
8764 ]
8765 ]
8766 },
8767 "tosa-ext-bf16": {
8768 "supports_all": [
8769 "tosa-ext-bf16"
8770 ],
8771 "selector": "tosa-mi",
8772 "no_negative_tests": "true",
8773 "negative_dim_range": "1,10",
8774 "generator_args": [
8775 [
8776 "--target-dtype",
8777 "bf16",
8778 "--fp-values-range",
8779 "-max,max"
8780 ]
8781 ]
8782 },
8783 "tosa-ext-int16": {
8784 "supports_all": [
8785 "tosa-ext-int16"
8786 ],
8787 "selector": "tosa-bi",
8788 "no_negative_tests": "true",
8789 "generator_args": [
8790 [
8791 "--target-dtype",
8792 "int16"
8793 ]
8794 ]
8795 }
8796 },
8797 "selection": {
8798 "tosa-bi": {
8799 "params": {},
8800 "permutes": [
8801 "shape",
8802 "type"
8803 ]
8804 },
8805 "tosa-mi": {
8806 "params": {
8807 "offset": [],
8808 "shape": [],
8809 "type": [],
8810 "mode": [],
8811 "output_type": [],
8812 "scale": [
8813 "sc2x1x3x1"
8814 ],
8815 "border": [
8816 "bor0x1"
8817 ]
8818 },
8819 "permutes": [
8820 "shape",
8821 "type"
8822 ],
8823 "exclude_patterns": [
8824 ".*_ERRORIF_MaxDimExceeded_1x16584x5x1_.*"
8825 ]
8826 }
8827 }
8828 },
8829 "reverse": {
8830 "group": "data_layout",
8831 "support_for": [
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]8832 "lazy_data_gen",
8833 "random_const_inputs",
8834 "stable_random_gen"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]8835 ],
8836 "generation": {
8837 "tosa-bi-mi": {
8838 "supports_any": [
8839 "tosa-bi",
8840 "tosa-mi"
8841 ],
8842 "generator_args": [
8843 [
8844 "--target-dtype",
8845 "bool",
8846 "--tensor-dim-range",
8847 "1,32",
8848 "--target-rank",
8849 "1",
8850 "--target-rank",
8851 "2",
8852 "--target-rank",
8853 "3",
8854 "--target-rank",
8855 "4"
8856 ]
8857 ]
8858 },
8859 "tosa-bi": {
8860 "supports_all": [
8861 "tosa-bi"
8862 ],
8863 "generator_args": [
8864 [
8865 "--target-dtype",
8866 "int8",
8867 "--target-dtype",
8868 "int16",
8869 "--target-dtype",
8870 "int32",
8871 "--tensor-dim-range",
8872 "1,32",
8873 "--target-rank",
8874 "1",
8875 "--target-rank",
8876 "2",
8877 "--target-rank",
8878 "3",
8879 "--target-rank",
8880 "4"
8881 ],
8882 [
8883 "--target-dtype",
8884 "int8",
8885 "--target-shape",
8886 "1,1,65535",
8887 "--target-shape",
8888 "1,65540,1,2"
8889 ]
8890 ]
8891 },
8892 "8k_level": {
8893 "supports_all": [
8894 "tosa-bi"
8895 ],
8896 "no_negative_tests": "true",
8897 "generator_args": [
8898 [
8899 "--target-dtype",
8900 "int8",
8901 "--target-dtype",
8902 "int16",
8903 "--target-dtype",
8904 "int32",
8905 "--tensor-dim-range",
8906 "1,12",
8907 "--target-rank",
8908 "5",
8909 "--target-rank",
8910 "6"
8911 ]
8912 ]
8913 },
8914 "8k_level_bi_mi": {
8915 "supports_any": [
8916 "tosa-bi",
8917 "tosa-mi"
8918 ],
8919 "no_negative_tests": "true",
8920 "generator_args": [
8921 [
8922 "--target-dtype",
8923 "bool",
8924 "--tensor-dim-range",
8925 "1,12",
8926 "--target-rank",
8927 "5",
8928 "--target-rank",
8929 "6"
8930 ]
8931 ]
8932 },
8933 "tosa-mi": {
8934 "supports_all": [
8935 "tosa-mi"
8936 ],
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]8937 "selector": "tosa-mi",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]8938 "generator_args": [
8939 [
8940 "--target-dtype",
8941 "fp32",
8942 "--target-dtype",
8943 "fp16",
8944 "--fp-values-range",
8945 "-max,max",
8946 "--tensor-dim-range",
8947 "1,48",
8948 "--target-rank",
8949 "1",
8950 "--target-rank",
8951 "2",
8952 "--target-rank",
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]8953 "3"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]8954 ],
8955 [
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]8956 "--no-special-tests",
8957 "--target-dtype",
8958 "fp32",
8959 "--target-dtype",
8960 "fp16",
8961 "--fp-values-range",
8962 "-max,max",
8963 "--tensor-dim-range",
8964 "1,11",
8965 "--target-rank",
8966 "4",
8967 "--target-rank",
8968 "5",
8969 "--target-rank",
8970 "6"
8971 ],
8972 [
8973 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]8974 "--target-dtype",
8975 "fp32",
8976 "--fp-values-range",
8977 "-max,max",
8978 "--target-shape",
8979 "1,1,65537",
8980 "--target-shape",
8981 "1,65535,1,2"
8982 ]
8983 ]
8984 },
8985 "tosa-ext-bf16": {
8986 "supports_all": [
8987 "tosa-ext-bf16"
8988 ],
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]8989 "selector": "tosa-mi",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]8990 "no_negative_tests": "true",
8991 "generator_args": [
8992 [
8993 "--target-dtype",
8994 "bf16",
8995 "--fp-values-range",
8996 "-max,max",
8997 "--tensor-dim-range",
8998 "1,48",
8999 "--target-rank",
9000 "1",
9001 "--target-rank",
9002 "2",
9003 "--target-rank",
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]9004 "3"
9005 ],
9006 [
9007 "--no-special-tests",
9008 "--target-dtype",
9009 "bf16",
9010 "--fp-values-range",
9011 "-max,max",
9012 "--tensor-dim-range",
9013 "1,11",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]9014 "--target-rank",
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]9015 "4",
9016 "--target-rank",
9017 "5",
9018 "--target-rank",
9019 "6"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]9020 ]
9021 ]
9022 },
9023 "tosa-ext-fp8e4m3": {
9024 "supports_all": [
9025 "tosa-ext-fp8e4m3"
9026 ],
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]9027 "selector": "tosa-mi",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]9028 "no_negative_tests": "true",
9029 "from_version": "v0.100.0",
9030 "generator_args": [
9031 [
9032 "--target-dtype",
9033 "fp8e4m3",
9034 "--fp-values-range",
9035 "-max,max",
9036 "--tensor-dim-range",
9037 "32,64",
9038 "--target-rank",
9039 "1",
9040 "--target-rank",
9041 "2",
9042 "--target-rank",
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]9043 "3"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]9044 ]
9045 ]
9046 },
9047 "tosa-ext-fp8e5m2": {
9048 "supports_all": [
9049 "tosa-ext-fp8e5m2"
9050 ],
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]9051 "selector": "tosa-mi",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]9052 "no_negative_tests": "true",
9053 "from_version": "v0.100.0",
9054 "generator_args": [
9055 [
9056 "--target-dtype",
9057 "fp8e5m2",
9058 "--fp-values-range",
9059 "-max,max",
9060 "--tensor-dim-range",
9061 "32,64",
9062 "--target-rank",
9063 "1",
9064 "--target-rank",
9065 "2",
9066 "--target-rank",
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]9067 "3"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]9068 ]
9069 ]
9070 }
9071 },
9072 "selection": {
9073 "default": {
9074 "params": {},
9075 "permutes": [
9076 "shape",
9077 "type"
9078 ]
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]9079 },
9080 "tosa-mi": {
9081 "generator_select": true,
9082 "maximum": 3
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]9083 }
9084 }
9085 },
9086 "rfft2d": {
9087 "group": "tensor",
9088 "support_for": [
9089 "lazy_data_gen"
9090 ],
9091 "generation": {
9092 "standard": {
9093 "supports_all": [
9094 "tosa-ext-fft"
9095 ],
9096 "generator_args": [
9097 [
9098 "--target-dtype",
9099 "fp32",
9100 "--fp-values-range",
9101 "-max,max"
9102 ],
9103 [
9104 "--target-dtype",
9105 "fp32",
9106 "--fp-values-range",
9107 "-max,max",
9108 "--target-shape",
9109 "1,16,512",
9110 "--target-shape",
9111 "1,8192,2"
9112 ]
9113 ]
9114 }
9115 },
9116 "selection": {
9117 "default": {
9118 "params": {},
9119 "permutes": [
9120 "shape",
9121 "type"
9122 ]
9123 }
9124 }
9125 },
9126 "rsqrt": {
9127 "group": "ew_unary",
9128 "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ],
9129 "generation": {
9130 "tosa-mi": {
9131 "supports_all": [
9132 "tosa-mi"
9133 ],
9134 "generator_args": [
9135 [
9136 "--target-dtype",
9137 "fp32",
9138 "--target-dtype",
9139 "fp16",
9140 "--fp-values-range",
9141 "-max,max",
9142 "--tensor-dim-range",
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]9143 "39,60",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]9144 "--target-rank",
9145 "0",
9146 "--target-rank",
9147 "1",
9148 "--target-rank",
9149 "2",
9150 "--target-rank",
9151 "3"
9152 ],
9153 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]9154 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]9155 "--target-dtype",
9156 "fp32",
Jeremy Johnson074b98c2024-05-30 10:38:35 +0100[diff] [blame]9157 "--target-dtype",
9158 "fp16",
9159 "--fp-values-range",
9160 "-max,max",
9161 "--tensor-dim-range",
9162 "1,13",
9163 "--target-rank",
9164 "4",
9165 "--target-rank",
9166 "5",
9167 "--target-rank",
9168 "6"
9169 ],
9170 [
9171 "--no-special-tests",
9172 "--target-dtype",
9173 "fp32",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]9174 "--fp-values-range",
9175 "-max,max",
9176 "--target-shape",
9177 "1,1,3,65541",
9178 "--target-shape",
9179 "65532,1,3,1,1"
9180 ]
9181 ]
9182 },
9183 "tosa-ext-bf16": {
9184 "supports_all": [
9185 "tosa-ext-bf16"
9186 ],
9187 "no_negative_tests": "true",
9188 "generator_args": [
9189 [
9190 "--target-dtype",
9191 "bf16",
9192 "--fp-values-range",
9193 "-max,max",
9194 "--tensor-dim-range",
9195 "15,64",
9196 "--target-rank",
9197 "0",
9198 "--target-rank",
9199 "1",
9200 "--target-rank",
9201 "2",
9202 "--target-rank",
9203 "3"
9204 ],
9205 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]9206 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]9207 "--target-dtype",
9208 "bf16",
9209 "--fp-values-range",
9210 "-max,max",
9211 "--tensor-dim-range",
9212 "1,15",
9213 "--target-rank",
9214 "4",
9215 "--target-rank",
Jeremy Johnson074b98c2024-05-30 10:38:35 +0100[diff] [blame]9216 "5",
9217 "--target-rank",
9218 "6"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]9219 ]
9220 ]
9221 }
9222 },
9223 "selection": {
9224 "default": {
9225 "generator_select": "true"
9226 }
9227 }
9228 },
9229 "scatter": {
9230 "group": "scatter_gather",
9231 "support_for": [
9232 "lazy_data_gen"
9233 ],
9234 "generation": {
9235 "tosa-bi": {
9236 "supports_all": [
9237 "tosa-bi"
9238 ],
9239 "generator_args": [
9240 [
9241 "--target-dtype",
9242 "int16",
9243 "--target-dtype",
9244 "int32",
9245 "--tensor-dim-range",
9246 "4,64",
9247 "--max-batch-size",
9248 "64"
9249 ],
9250 [
9251 "--target-dtype",
9252 "int8",
9253 "--target-shape",
9254 "1,65540,1",
9255 "--target-shape",
9256 "1,1,65533",
9257 "--target-shape",
9258 "3000,7,1"
9259 ]
9260 ]
9261 },
9262 "tosa-mi": {
9263 "supports_all": [
9264 "tosa-mi"
9265 ],
9266 "negative_dim_range": "1,10",
9267 "generator_args": [
9268 [
9269 "--target-dtype",
9270 "fp32",
9271 "--target-dtype",
9272 "fp16",
9273 "--fp-values-range",
9274 "-max,max",
9275 "--tensor-dim-range",
9276 "5,56",
9277 "--max-batch-size",
9278 "56"
9279 ],
9280 [
9281 "--target-dtype",
9282 "fp32",
9283 "--fp-values-range",
9284 "-max,max",
9285 "--target-shape",
9286 "1,65541,1",
9287 "--target-shape",
9288 "1,1,65531",
9289 "--target-shape",
9290 "2989,6,1"
9291 ]
9292 ]
9293 },
9294 "tosa-ext-bf16": {
9295 "supports_all": [
9296 "tosa-ext-bf16"
9297 ],
9298 "no_negative_tests": "true",
9299 "negative_dim_range": "1,10",
9300 "generator_args": [
9301 [
9302 "--target-dtype",
9303 "bf16",
9304 "--fp-values-range",
9305 "-max,max",
9306 "--tensor-dim-range",
9307 "5,56",
9308 "--max-batch-size",
9309 "56"
9310 ]
9311 ]
9312 },
9313 "tosa-ext-fp8e4m3": {
9314 "supports_all": [
9315 "tosa-ext-fp8e4m3"
9316 ],
9317 "no_negative_tests": "true",
9318 "from_version": "v0.100.0",
9319 "generator_args": [
9320 [
9321 "--target-dtype",
9322 "fp8e4m3",
9323 "--fp-values-range",
9324 "-max,max",
9325 "--tensor-dim-range",
9326 "32,64",
9327 "--num-rand-permutations",
9328 "2"
9329 ]
9330 ]
9331 },
9332 "tosa-ext-fp8e5m2": {
9333 "supports_all": [
9334 "tosa-ext-fp8e5m2"
9335 ],
9336 "no_negative_tests": "true",
9337 "from_version": "v0.100.0",
9338 "generator_args": [
9339 [
9340 "--target-dtype",
9341 "fp8e5m2",
9342 "--fp-values-range",
9343 "-max,max",
9344 "--tensor-dim-range",
9345 "32,64",
9346 "--num-rand-permutations",
9347 "2"
9348 ]
9349 ]
9350 }
9351 },
9352 "selection": {
9353 "default": {
9354 "params": {},
9355 "permutes": [
9356 "shape",
9357 "type"
9358 ]
9359 }
9360 }
9361 },
9362 "select": {
9363 "group": "ew_ternary",
9364 "support_for": [
9365 "lazy_data_gen"
9366 ],
9367 "generation": {
9368 "tosa-bi-mi": {
9369 "supports_any": [
9370 "tosa-bi",
9371 "tosa-mi"
9372 ],
9373 "generator_args": [
9374 [
9375 "--target-dtype",
9376 "bool",
9377 "--tensor-dim-range",
9378 "16,64",
9379 "--target-rank",
9380 "1",
9381 "--target-rank",
9382 "2",
9383 "--target-rank",
9384 "3"
9385 ],
9386 [
9387 "--target-dtype",
9388 "bool",
9389 "--tensor-dim-range",
9390 "1,16",
9391 "--target-rank",
9392 "4",
9393 "--target-rank",
9394 "5"
9395 ]
9396 ]
9397 },
9398 "tosa-bi": {
9399 "supports_all": [
9400 "tosa-bi"
9401 ],
9402 "generator_args": [
9403 [
9404 "--target-dtype",
9405 "int8",
9406 "--target-dtype",
9407 "int16",
9408 "--target-dtype",
9409 "int32",
9410 "--tensor-dim-range",
9411 "16,64",
9412 "--target-rank",
9413 "1",
9414 "--target-rank",
9415 "2",
9416 "--target-rank",
9417 "3"
9418 ],
9419 [
9420 "--target-dtype",
9421 "int8",
9422 "--target-dtype",
9423 "int16",
9424 "--target-dtype",
9425 "int32",
9426 "--tensor-dim-range",
9427 "1,16",
9428 "--target-rank",
9429 "4",
9430 "--target-rank",
9431 "5"
9432 ],
9433 [
9434 "--target-dtype",
9435 "int8",
9436 "--target-shape",
9437 "1,1,65535,1",
9438 "--target-shape",
9439 "1,65538,1,2",
9440 "--num-rand-permutations",
9441 "2"
9442 ]
9443 ]
9444 },
9445 "8k_level": {
9446 "supports_all": [
9447 "tosa-bi"
9448 ],
9449 "no_negative_tests": "true",
9450 "generator_args": [
9451 [
9452 "--target-dtype",
9453 "int8",
9454 "--target-dtype",
9455 "int16",
9456 "--target-dtype",
9457 "int32",
9458 "--tensor-dim-range",
9459 "1,10",
9460 "--target-rank",
9461 "6"
9462 ]
9463 ]
9464 },
9465 "8k_level-bi-mi": {
9466 "supports_any": [
9467 "tosa-bi",
9468 "tosa-mi"
9469 ],
9470 "no_negative_tests": "true",
9471 "generator_args": [
9472 [
9473 "--target-dtype",
9474 "bool",
9475 "--tensor-dim-range",
9476 "1,10",
9477 "--target-rank",
9478 "6"
9479 ]
9480 ]
9481 },
9482 "tosa-mi": {
9483 "supports_all": [
9484 "tosa-mi"
9485 ],
9486 "generator_args": [
9487 [
9488 "--target-dtype",
9489 "fp32",
9490 "--target-dtype",
9491 "fp16",
9492 "--fp-values-range",
9493 "-max,max",
9494 "--tensor-dim-range",
9495 "16,64",
9496 "--target-rank",
9497 "1",
9498 "--target-rank",
9499 "2",
9500 "--target-rank",
9501 "3"
9502 ],
9503 [
9504 "--target-dtype",
9505 "fp32",
9506 "--target-dtype",
9507 "fp16",
9508 "--fp-values-range",
9509 "-max,max",
9510 "--tensor-dim-range",
9511 "1,16",
9512 "--target-rank",
9513 "4",
9514 "--target-rank",
9515 "5"
9516 ],
9517 [
9518 "--target-dtype",
9519 "fp32",
9520 "--fp-values-range",
9521 "-max,max",
9522 "--target-shape",
9523 "1,2,65534,2,1",
9524 "--target-shape",
9525 "1,1,65539,2",
9526 "--num-rand-permutations",
9527 "2"
9528 ]
9529 ]
9530 },
9531 "tosa-ext-bf16": {
9532 "supports_all": [
9533 "tosa-ext-bf16"
9534 ],
9535 "no_negative_tests": "true",
9536 "generator_args": [
9537 [
9538 "--target-dtype",
9539 "bf16",
9540 "--fp-values-range",
9541 "-max,max",
9542 "--tensor-dim-range",
9543 "16,64",
9544 "--target-rank",
9545 "1",
9546 "--target-rank",
9547 "2",
9548 "--target-rank",
9549 "3"
9550 ],
9551 [
9552 "--target-dtype",
9553 "bf16",
9554 "--fp-values-range",
9555 "-max,max",
9556 "--tensor-dim-range",
9557 "1,16",
9558 "--target-rank",
9559 "4",
9560 "--target-rank",
9561 "5"
9562 ]
9563 ]
9564 }
9565 },
9566 "selection": {
9567 "default": {
9568 "params": {},
9569 "permutes": [
9570 "shape",
9571 "type"
9572 ]
9573 }
9574 }
9575 },
9576 "sigmoid": {
9577 "group": "activation",
9578 "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ],
9579 "generation": {
9580 "tosa-mi": {
9581 "supports_all": [
9582 "tosa-mi"
9583 ],
9584 "generator_args": [
9585 [
9586 "--target-dtype",
9587 "fp32",
9588 "--target-dtype",
9589 "fp16",
9590 "--fp-values-range",
9591 "-16.0,16.0",
9592 "--tensor-dim-range",
9593 "16,64",
9594 "--target-rank",
9595 "0",
9596 "--target-rank",
9597 "1",
9598 "--target-rank",
9599 "2",
9600 "--target-rank",
9601 "3"
9602 ],
9603 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]9604 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]9605 "--target-dtype",
9606 "fp32",
9607 "--target-dtype",
9608 "fp16",
9609 "--fp-values-range",
9610 "-16.0,16.0",
9611 "--tensor-dim-range",
9612 "1,19",
9613 "--target-rank",
9614 "4",
9615 "--target-rank",
Jeremy Johnson4423bb72024-06-05 15:30:25 +0100[diff] [blame]9616 "5",
9617 "--target-rank",
9618 "6"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]9619 ],
9620 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]9621 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]9622 "--target-dtype",
9623 "fp32",
9624 "--fp-values-range",
9625 "-16.0,16.0",
9626 "--target-shape",
9627 "1,2,65538,2,1",
9628 "--target-shape",
9629 "1,3,1,65533"
9630 ]
9631 ]
9632 },
9633 "tosa-ext-bf16": {
9634 "supports_all": [
9635 "tosa-ext-bf16"
9636 ],
9637 "no_negative_tests": "true",
9638 "generator_args": [
9639 [
9640 "--target-dtype",
9641 "bf16",
9642 "--fp-values-range",
9643 "-16.0,16.0",
9644 "--tensor-dim-range",
9645 "16,64",
9646 "--target-rank",
9647 "0",
9648 "--target-rank",
9649 "1",
9650 "--target-rank",
9651 "2",
9652 "--target-rank",
9653 "3"
9654 ],
9655 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]9656 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]9657 "--target-dtype",
9658 "bf16",
9659 "--fp-values-range",
9660 "-16.0,16.0",
9661 "--tensor-dim-range",
9662 "1,19",
9663 "--target-rank",
9664 "4",
9665 "--target-rank",
Jeremy Johnson4423bb72024-06-05 15:30:25 +0100[diff] [blame]9666 "5",
9667 "--target-rank",
9668 "6"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]9669 ]
9670 ]
9671 }
9672 },
9673 "selection": {
9674 "default": {
9675 "generator_select": "true"
9676 }
9677 }
9678 },
9679 "sin": {
9680 "group": "ew_unary",
Jeremy Johnson074b98c2024-05-30 10:38:35 +0100[diff] [blame]9681 "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ],
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]9682 "generation": {
9683 "tosa-mi": {
9684 "supports_all": [
9685 "tosa-mi"
9686 ],
9687 "generator_args": [
9688 [
9689 "--target-dtype",
9690 "fp32",
9691 "--target-dtype",
9692 "fp16",
9693 "--fp-values-range",
9694 "-max,max",
9695 "--tensor-dim-range",
Jeremy Johnson074b98c2024-05-30 10:38:35 +0100[diff] [blame]9696 "35,60",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]9697 "--target-rank",
9698 "1",
9699 "--target-rank",
9700 "2",
9701 "--target-rank",
9702 "3"
9703 ],
9704 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]9705 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]9706 "--target-dtype",
9707 "fp32",
Jeremy Johnson074b98c2024-05-30 10:38:35 +0100[diff] [blame]9708 "--target-dtype",
9709 "fp16",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]9710 "--fp-values-range",
9711 "-max,max",
9712 "--tensor-dim-range",
9713 "1,15",
9714 "--target-rank",
9715 "4",
9716 "--target-rank",
Jeremy Johnson074b98c2024-05-30 10:38:35 +0100[diff] [blame]9717 "5",
9718 "--target-rank",
9719 "6"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]9720 ],
9721 [
Jeremy Johnson074b98c2024-05-30 10:38:35 +0100[diff] [blame]9722 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]9723 "--target-dtype",
9724 "fp16",
9725 "--fp-values-range",
9726 "-max,max",
9727 "--target-shape",
9728 "3,1,65534,2",
9729 "--target-shape",
9730 "65533,1,3,2,1"
9731 ]
9732 ]
9733 },
9734 "tosa-ext-bf16": {
9735 "supports_all": [
9736 "tosa-ext-bf16"
9737 ],
9738 "no_negative_tests": "true",
9739 "generator_args": [
9740 [
9741 "--target-dtype",
9742 "bf16",
9743 "--fp-values-range",
9744 "-max,max",
9745 "--tensor-dim-range",
9746 "15,64",
9747 "--target-rank",
9748 "1",
9749 "--target-rank",
9750 "2",
9751 "--target-rank",
9752 "3"
Jeremy Johnson074b98c2024-05-30 10:38:35 +0100[diff] [blame]9753 ],
9754 [
9755 "--no-special-tests",
9756 "--target-dtype",
9757 "bf16",
9758 "--fp-values-range",
9759 "-max,max",
9760 "--tensor-dim-range",
9761 "1,10",
9762 "--target-rank",
9763 "4",
9764 "--target-rank",
9765 "5",
9766 "--target-rank",
9767 "6"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]9768 ]
9769 ]
9770 }
9771 },
9772 "selection": {
9773 "default": {
Jeremy Johnson074b98c2024-05-30 10:38:35 +0100[diff] [blame]9774 "generator_select": "true"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]9775 }
9776 }
9777 },
9778 "slice": {
9779 "group": "data_layout",
9780 "support_for": [
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]9781 "lazy_data_gen",
9782 "random_const_inputs",
9783 "stable_random_gen"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]9784 ],
9785 "generation": {
9786 "tosa-bi-mi": {
9787 "supports_any": [
9788 "tosa-bi",
9789 "tosa-mi"
9790 ],
9791 "no_negative_tests": "true",
9792 "generator_args": [
9793 [
9794 "--target-dtype",
9795 "bool",
9796 "--tensor-dim-range",
9797 "16,64",
9798 "--target-rank",
9799 "1",
9800 "--target-rank",
9801 "2",
9802 "--target-rank",
9803 "3",
9804 "--num-rand-permutations",
9805 "2"
9806 ],
9807 [
9808 "--target-dtype",
9809 "bool",
9810 "--tensor-dim-range",
9811 "1,16",
9812 "--target-rank",
9813 "4",
9814 "--num-rand-permutations",
9815 "2"
9816 ]
9817 ]
9818 },
9819 "8k_level_bi_mi": {
9820 "supports_any": [
9821 "tosa-bi",
9822 "tosa-mi"
9823 ],
9824 "no_negative_tests": "true",
9825 "generator_args": [
9826 [
9827 "--target-dtype",
9828 "bool",
9829 "--tensor-dim-range",
9830 "1,10",
9831 "--target-rank",
9832 "5",
9833 "--target-rank",
9834 "6",
9835 "--num-rand-permutations",
9836 "2"
9837 ]
9838 ]
9839 },
9840 "tosa-bi": {
9841 "supports_all": [
9842 "tosa-bi"
9843 ],
9844 "generator_args": [
9845 [
9846 "--target-dtype",
9847 "int8",
9848 "--target-dtype",
9849 "int16",
9850 "--target-dtype",
9851 "int32",
9852 "--tensor-dim-range",
9853 "16,64",
9854 "--target-rank",
9855 "1",
9856 "--target-rank",
9857 "2",
9858 "--target-rank",
9859 "3",
9860 "--num-rand-permutations",
9861 "2"
9862 ],
9863 [
9864 "--target-dtype",
9865 "int8",
9866 "--target-dtype",
9867 "int16",
9868 "--target-dtype",
9869 "int32",
9870 "--tensor-dim-range",
9871 "1,16",
9872 "--target-rank",
9873 "4",
9874 "--num-rand-permutations",
9875 "2"
9876 ],
9877 [
9878 "--target-dtype",
9879 "int8",
9880 "--target-shape",
9881 "1,1,65535,1",
9882 "--target-shape",
9883 "1,65538,1,2",
9884 "--num-rand-permutations",
9885 "2"
9886 ]
9887 ]
9888 },
9889 "8k_level": {
9890 "supports_all": [
9891 "tosa-bi"
9892 ],
9893 "no_negative_tests": "true",
9894 "generator_args": [
9895 [
9896 "--target-dtype",
9897 "int8",
9898 "--target-dtype",
9899 "int16",
9900 "--target-dtype",
9901 "int32",
9902 "--tensor-dim-range",
9903 "1,10",
9904 "--target-rank",
9905 "5",
9906 "--target-rank",
9907 "6",
9908 "--num-rand-permutations",
9909 "2"
9910 ]
9911 ]
9912 },
9913 "tosa-mi": {
9914 "supports_all": [
9915 "tosa-mi"
9916 ],
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]9917 "selector": "tosa-mi",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]9918 "no_negative_tests": "true",
9919 "generator_args": [
9920 [
9921 "--target-dtype",
9922 "fp32",
9923 "--target-dtype",
9924 "fp16",
9925 "--fp-values-range",
9926 "-max,max",
9927 "--tensor-dim-range",
9928 "16,64",
9929 "--target-rank",
9930 "1",
9931 "--target-rank",
9932 "2",
9933 "--target-rank",
9934 "3",
9935 "--num-rand-permutations",
9936 "2"
9937 ],
9938 [
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]9939 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]9940 "--target-dtype",
9941 "fp32",
9942 "--target-dtype",
9943 "fp16",
9944 "--fp-values-range",
9945 "-max,max",
9946 "--tensor-dim-range",
9947 "1,16",
9948 "--target-rank",
9949 "4",
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]9950 "--target-rank",
9951 "5",
9952 "--target-rank",
9953 "6",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]9954 "--num-rand-permutations",
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]9955 "1"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]9956 ]
9957 ]
9958 },
9959 "tosa-ext-bf16": {
9960 "supports_all": [
9961 "tosa-ext-bf16"
9962 ],
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]9963 "selector": "tosa-mi",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]9964 "no_negative_tests": "true",
9965 "generator_args": [
9966 [
9967 "--target-dtype",
9968 "bf16",
9969 "--fp-values-range",
9970 "-max,max",
9971 "--tensor-dim-range",
9972 "16,64",
9973 "--target-rank",
9974 "1",
9975 "--target-rank",
9976 "2",
9977 "--target-rank",
9978 "3",
9979 "--num-rand-permutations",
9980 "2"
9981 ],
9982 [
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]9983 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]9984 "--target-dtype",
9985 "bf16",
9986 "--fp-values-range",
9987 "-max,max",
9988 "--tensor-dim-range",
9989 "1,16",
9990 "--target-rank",
9991 "4",
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]9992 "--target-rank",
9993 "5",
9994 "--target-rank",
9995 "6",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]9996 "--num-rand-permutations",
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]9997 "1"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]9998 ],
9999 [
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]10000 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]10001 "--target-dtype",
10002 "bf16",
10003 "--fp-values-range",
10004 "-2.0,2.0",
10005 "--target-shape",
10006 "1,1,655341,1",
10007 "--target-shape",
10008 "1,65539,1,1",
10009 "--num-rand-permutations",
10010 "2"
10011 ]
10012 ]
10013 },
10014 "tosa-ext-fp8e4m3": {
10015 "supports_all": [
10016 "tosa-ext-fp8e4m3"
10017 ],
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]10018 "selector": "tosa-mi",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]10019 "no_negative_tests": "true",
10020 "from_version": "v0.100.0",
10021 "generator_args": [
10022 [
10023 "--target-dtype",
10024 "fp8e4m3",
10025 "--fp-values-range",
10026 "-max,max",
10027 "--tensor-dim-range",
10028 "32,64",
10029 "--target-rank",
10030 "1",
10031 "--target-rank",
10032 "2",
10033 "--target-rank",
10034 "3",
10035 "--num-rand-permutations",
10036 "2"
10037 ]
10038 ]
10039 },
10040 "tosa-ext-fp8e5m2": {
10041 "supports_all": [
10042 "tosa-ext-fp8e5m2"
10043 ],
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]10044 "selector": "tosa-mi",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]10045 "no_negative_tests": "true",
10046 "from_version": "v0.100.0",
10047 "generator_args": [
10048 [
10049 "--target-dtype",
10050 "fp8e5m2",
10051 "--fp-values-range",
10052 "-max,max",
10053 "--tensor-dim-range",
10054 "32,64",
10055 "--target-rank",
10056 "1",
10057 "--target-rank",
10058 "2",
10059 "--target-rank",
10060 "3",
10061 "--num-rand-permutations",
10062 "2"
10063 ]
10064 ]
10065 }
10066 },
10067 "selection": {
10068 "default": {
10069 "params": {},
10070 "permutes": [
10071 "shape",
10072 "type",
10073 "perm"
10074 ]
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]10075 },
10076 "tosa-mi": {
10077 "generator_select": true
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]10078 }
10079 }
10080 },
10081 "sub": {
10082 "group": "ew_binary",
10083 "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ],
10084 "generation": {
10085 "tosa-bi-mi": {
10086 "supports_any": [
10087 "tosa-bi",
10088 "tosa-mi"
10089 ],
10090 "no_negative_tests": "true",
10091 "generator_args": [
10092 [
10093 "--target-dtype",
10094 "int32",
10095 "--tensor-dim-range",
10096 "16,64",
10097 "--target-rank",
10098 "1",
10099 "--target-rank",
10100 "2",
10101 "--target-rank",
10102 "3"
10103 ],
10104 [
10105 "--target-dtype",
10106 "int32",
10107 "--tensor-dim-range",
10108 "1,16",
10109 "--target-rank",
10110 "4",
10111 "--target-rank",
10112 "5"
10113 ],
10114 [
10115 "--target-dtype",
10116 "int32",
10117 "--target-shape",
10118 "1,2,65533,1",
10119 "--target-shape",
10120 "1,65539,1,1,1"
10121 ]
10122 ]
10123 },
10124 "8k_level": {
10125 "supports_all": [
10126 "tosa-bi"
10127 ],
10128 "no_negative_tests": "true",
10129 "generator_args": [
10130 [
10131 "--target-dtype",
10132 "int32",
10133 "--tensor-dim-range",
10134 "1,10",
10135 "--target-rank",
10136 "6"
10137 ]
10138 ]
10139 },
10140 "tosa-mi": {
10141 "supports_all": [
10142 "tosa-mi"
10143 ],
10144 "selector": "tosa-mi",
10145 "generator_args": [
10146 [
10147 "--target-dtype",
10148 "fp32",
10149 "--target-dtype",
10150 "fp16",
10151 "--fp-values-range",
10152 "-max,max",
10153 "--tensor-dim-range",
10154 "16,64",
10155 "--target-rank",
10156 "0",
10157 "--target-rank",
10158 "1",
10159 "--target-rank",
10160 "2",
10161 "--target-rank",
10162 "3"
10163 ],
10164 [
10165 "--target-dtype",
10166 "fp32",
10167 "--fp-values-range",
10168 "-max,max",
10169 "--target-shape",
10170 "1,1,65533,3",
10171 "--target-shape",
10172 "1,1,2,1,65545"
10173 ]
10174 ]
10175 },
10176 "tosa-ext-bf16": {
10177 "supports_all": [
10178 "tosa-ext-bf16"
10179 ],
10180 "selector": "tosa-mi",
10181 "no_negative_tests": "true",
10182 "generator_args": [
10183 [
10184 "--target-dtype",
10185 "bf16",
10186 "--fp-values-range",
10187 "-max,max",
10188 "--tensor-dim-range",
10189 "16,64",
10190 "--target-rank",
10191 "0",
10192 "--target-rank",
10193 "1",
10194 "--target-rank",
10195 "2",
10196 "--target-rank",
10197 "3"
10198 ],
10199 [
10200 "--target-dtype",
10201 "bf16",
10202 "--fp-values-range",
10203 "-max,max",
10204 "--tensor-dim-range",
10205 "1,16",
10206 "--target-rank",
10207 "4",
10208 "--target-rank",
10209 "5"
10210 ]
10211 ]
10212 }
10213 },
10214 "selection": {
10215 "default": {
10216 "params": {},
10217 "permutes": [
10218 "shape",
10219 "type"
10220 ]
10221 },
10222 "tosa-mi": {
10223 "generator_select": "true"
10224 }
10225 }
10226 },
10227 "sub_shape": {
10228 "group": "shape",
10229 "generation": {
10230 "standard": {
10231 "supports_any": [
10232 "tosa-bi",
10233 "tosa-mi"
10234 ],
10235 "generator_args": [
10236 [
10237 "--target-dtype",
10238 "shape",
10239 "--tensor-dim-range",
10240 "1,16",
10241 "--target-rank",
10242 "1"
10243 ]
10244 ]
10245 }
10246 },
10247 "selection": {
10248 "default": {
10249 "params": {},
10250 "permutes": [
10251 "shape"
10252 ]
10253 }
10254 }
10255 },
10256 "table": {
10257 "group": "ew_binary",
10258 "generation": {
10259 "tosa-bi": {
10260 "supports_all": [
10261 "tosa-bi"
10262 ],
10263 "generator_args": [
10264 [
10265 "--target-dtype",
10266 "int8",
10267 "--tensor-dim-range",
10268 "16,64",
10269 "--target-rank",
10270 "1",
10271 "--target-rank",
10272 "2",
10273 "--target-rank",
10274 "3"
10275 ],
10276 [
10277 "--target-dtype",
10278 "int8",
10279 "--tensor-dim-range",
10280 "1,16",
10281 "--target-rank",
10282 "4",
10283 "--target-rank",
10284 "5"
10285 ],
10286 [
10287 "--target-dtype",
10288 "int8",
10289 "--target-shape",
10290 "1,2,65536,1",
10291 "--target-shape",
10292 "1,65537,1,1,3"
10293 ]
10294 ]
10295 },
10296 "8k_level": {
10297 "supports_all": [
10298 "tosa-bi"
10299 ],
10300 "no_negative_tests": "true",
10301 "generator_args": [
10302 [
10303 "--target-dtype",
10304 "int8",
10305 "--tensor-dim-range",
10306 "1,10",
10307 "--target-rank",
10308 "6"
10309 ]
10310 ]
10311 },
10312 "8k_level_ext_int16": {
10313 "supports_all": [
10314 "tosa-bi"
10315 ],
10316 "no_negative_tests": "true",
10317 "generator_args": [
10318 [
10319 "--target-dtype",
10320 "int16",
10321 "--tensor-dim-range",
10322 "1,10",
10323 "--target-rank",
10324 "6"
10325 ]
10326 ]
10327 },
10328 "tosa-ext-int16": {
10329 "supports_all": [
10330 "tosa-ext-int16"
10331 ],
10332 "no_negative_tests": "true",
10333 "generator_args": [
10334 [
10335 "--target-dtype",
10336 "int16",
10337 "--tensor-dim-range",
10338 "16,64",
10339 "--target-rank",
10340 "1",
10341 "--target-rank",
10342 "2",
10343 "--target-rank",
10344 "3"
10345 ],
10346 [
10347 "--target-dtype",
10348 "int16",
10349 "--tensor-dim-range",
10350 "1,16",
10351 "--target-rank",
10352 "4",
10353 "--target-rank",
10354 "5"
10355 ]
10356 ]
10357 }
10358 },
10359 "selection": {
10360 "default": {
10361 "params": {},
10362 "permutes": [
10363 "shape",
10364 "type"
10365 ]
10366 }
10367 }
10368 },
10369 "tanh": {
10370 "group": "activation",
10371 "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ],
10372 "generation": {
10373 "tosa-mi": {
10374 "supports_all": [
10375 "tosa-mi"
10376 ],
10377 "generator_args": [
10378 [
10379 "--target-dtype",
10380 "fp32",
10381 "--target-dtype",
10382 "fp16",
10383 "--fp-values-range",
10384 "-8.0,8.0",
10385 "--tensor-dim-range",
10386 "18,60",
10387 "--target-rank",
10388 "0",
10389 "--target-rank",
10390 "1",
10391 "--target-rank",
10392 "2",
10393 "--target-rank",
10394 "3"
10395 ],
10396 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]10397 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]10398 "--target-dtype",
10399 "fp32",
10400 "--target-dtype",
10401 "fp16",
10402 "--fp-values-range",
10403 "-8.0,8.0",
10404 "--tensor-dim-range",
10405 "1,24",
10406 "--target-rank",
10407 "4",
10408 "--target-rank",
Jeremy Johnson4423bb72024-06-05 15:30:25 +0100[diff] [blame]10409 "5",
10410 "--target-rank",
10411 "6"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]10412 ],
10413 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]10414 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]10415 "--target-dtype",
10416 "fp32",
10417 "--fp-values-range",
10418 "-8.0,8.0",
10419 "--target-shape",
10420 "1,65535,2,1,1",
10421 "--target-shape",
10422 "1,65540,1,2"
10423 ]
10424 ]
10425 },
10426 "tosa-ext-bf16": {
10427 "supports_all": [
10428 "tosa-ext-bf16"
10429 ],
10430 "no_negative_tests": "true",
10431 "generator_args": [
10432 [
10433 "--target-dtype",
10434 "bf16",
10435 "--fp-values-range",
10436 "-8.0,8.0",
10437 "--tensor-dim-range",
10438 "18,60",
10439 "--target-rank",
10440 "0",
10441 "--target-rank",
10442 "1",
10443 "--target-rank",
10444 "2",
10445 "--target-rank",
10446 "3"
10447 ],
10448 [
Jeremy Johnsona143d882024-05-28 17:53:25 +0100[diff] [blame]10449 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]10450 "--target-dtype",
10451 "bf16",
10452 "--fp-values-range",
10453 "-8.0,8.0",
10454 "--tensor-dim-range",
10455 "1,24",
10456 "--target-rank",
10457 "4",
10458 "--target-rank",
Jeremy Johnson4423bb72024-06-05 15:30:25 +0100[diff] [blame]10459 "5",
10460 "--target-rank",
10461 "6"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]10462 ]
10463 ]
10464 }
10465 },
10466 "selection": {
10467 "default": {
10468 "generator_select": "true"
10469 }
10470 }
10471 },
10472 "tile": {
10473 "group": "data_layout",
10474 "support_for": [
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]10475 "lazy_data_gen",
10476 "random_const_inputs",
10477 "stable_random_gen"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]10478 ],
10479 "generation": {
10480 "tosa-bi-mi": {
10481 "supports_any": [
10482 "tosa-bi",
10483 "tosa-mi"
10484 ],
10485 "generator_args": [
10486 [
10487 "--target-dtype",
10488 "bool",
10489 "--tensor-dim-range",
10490 "1,16",
10491 "--target-rank",
10492 "1",
10493 "--target-rank",
10494 "2",
10495 "--target-rank",
10496 "3",
10497 "--target-rank",
10498 "4"
10499 ]
10500 ]
10501 },
10502 "8k_level_bi_mi": {
10503 "supports_all": [
10504 "tosa-bi"
10505 ],
10506 "no_negative_tests": "true",
10507 "generator_args": [
10508 [
10509 "--target-dtype",
10510 "bool",
10511 "--tensor-dim-range",
10512 "1,10",
10513 "--target-rank",
10514 "5",
10515 "--target-rank",
10516 "6"
10517 ]
10518 ]
10519 },
10520 "tosa-bi": {
10521 "supports_all": [
10522 "tosa-bi"
10523 ],
10524 "no_negative_tests": "true",
10525 "generator_args": [
10526 [
10527 "--target-dtype",
10528 "int16",
10529 "--target-dtype",
10530 "int32",
10531 "--tensor-dim-range",
10532 "4,32",
10533 "--target-rank",
10534 "1",
10535 "--target-rank",
10536 "2",
10537 "--target-rank",
10538 "3",
10539 "--target-rank",
10540 "4"
10541 ],
10542 [
10543 "--target-dtype",
10544 "int8",
10545 "--target-shape",
10546 "65533",
10547 "--target-shape",
10548 "1,2,65540,1"
10549 ]
10550 ]
10551 },
10552 "8k_level": {
10553 "supports_all": [
10554 "tosa-bi"
10555 ],
10556 "no_negative_tests": "true",
10557 "generator_args": [
10558 [
10559 "--target-dtype",
10560 "int16",
10561 "--target-dtype",
10562 "int32",
10563 "--tensor-dim-range",
10564 "1,10",
10565 "--target-rank",
10566 "5",
10567 "--target-rank",
10568 "6"
10569 ]
10570 ]
10571 },
10572 "tosa-mi": {
10573 "supports_all": [
10574 "tosa-mi"
10575 ],
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]10576 "selector": "tosa-mi",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]10577 "negative_dim_range": "1,10",
10578 "generator_args": [
10579 [
10580 "--target-dtype",
10581 "fp32",
10582 "--target-dtype",
10583 "fp16",
10584 "--fp-values-range",
10585 "-max,max",
10586 "--tensor-dim-range",
10587 "4,32",
10588 "--target-rank",
10589 "1",
10590 "--target-rank",
10591 "2",
10592 "--target-rank",
10593 "3",
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]10594 "--num-rand-permutations",
10595 "2"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]10596 ],
10597 [
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]10598 "--no-special-tests",
10599 "--target-dtype",
10600 "fp32",
10601 "--target-dtype",
10602 "fp16",
10603 "--fp-values-range",
10604 "-max,max",
10605 "--tensor-dim-range",
10606 "1,13",
10607 "--target-rank",
10608 "4",
10609 "--target-rank",
10610 "5",
10611 "--target-rank",
10612 "6",
10613 "--num-rand-permutations",
10614 "1"
10615 ],
10616 [
10617 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]10618 "--target-dtype",
10619 "fp32",
10620 "--fp-values-range",
10621 "-max,max",
10622 "--target-shape",
10623 "65535",
10624 "--target-shape",
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]10625 "1,65543,2,1",
10626 "--num-rand-permutations",
10627 "2"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]10628 ]
10629 ]
10630 },
10631 "tosa-ext-bf16": {
10632 "supports_all": [
10633 "tosa-ext-bf16"
10634 ],
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]10635 "selector": "tosa-mi",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]10636 "no_negative_tests": "true",
10637 "negative_dim_range": "1,10",
10638 "generator_args": [
10639 [
10640 "--target-dtype",
10641 "bf16",
10642 "--fp-values-range",
10643 "-max,max",
10644 "--tensor-dim-range",
10645 "4,32",
10646 "--target-rank",
10647 "1",
10648 "--target-rank",
10649 "2",
10650 "--target-rank",
10651 "3",
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]10652 "--num-rand-permutations",
10653 "2"
10654 ],
10655 [
10656 "--no-special-tests",
10657 "--target-dtype",
10658 "bf16",
10659 "--fp-values-range",
10660 "-max,max",
10661 "--tensor-dim-range",
10662 "1,9",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]10663 "--target-rank",
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]10664 "4",
10665 "--target-rank",
10666 "5",
10667 "--target-rank",
10668 "6",
10669 "--num-rand-permutations",
10670 "1"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]10671 ]
10672 ]
10673 },
10674 "tosa-ext-fp8e4m3": {
10675 "supports_all": [
10676 "tosa-ext-fp8e4m3"
10677 ],
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]10678 "selector": "tosa-mi",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]10679 "no_negative_tests": "true",
10680 "from_version": "v0.100.0",
10681 "generator_args": [
10682 [
10683 "--target-dtype",
10684 "fp8e4m3",
10685 "--fp-values-range",
10686 "-max,max",
10687 "--tensor-dim-range",
10688 "32,64",
10689 "--target-rank",
10690 "1",
10691 "--target-rank",
10692 "2",
10693 "--target-rank",
10694 "3",
10695 "--num-rand-permutations",
10696 "2"
10697 ]
10698 ]
10699 },
10700 "tosa-ext-fp8e5m2": {
10701 "supports_all": [
10702 "tosa-ext-fp8e5m2"
10703 ],
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]10704 "selector": "tosa-mi",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]10705 "no_negative_tests": "true",
10706 "from_version": "v0.100.0",
10707 "generator_args": [
10708 [
10709 "--target-dtype",
10710 "fp8e5m2",
10711 "--fp-values-range",
10712 "-max,max",
10713 "--tensor-dim-range",
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]10714 "32,60",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]10715 "--target-rank",
10716 "1",
10717 "--target-rank",
10718 "2",
10719 "--target-rank",
10720 "3",
10721 "--num-rand-permutations",
10722 "2"
10723 ]
10724 ]
10725 }
10726 },
10727 "selection": {
10728 "default": {
10729 "params": {},
10730 "permutes": [
10731 "shape",
10732 "type"
10733 ]
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]10734 },
10735 "tosa-mi": {
10736 "generator_select": true
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]10737 }
10738 }
10739 },
10740 "transpose": {
10741 "group": "data_layout",
10742 "support_for": [
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]10743 "lazy_data_gen",
10744 "random_const_inputs",
10745 "stable_random_gen"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]10746 ],
10747 "generation": {
10748 "tosa-bi-mi": {
10749 "supports_any": [
10750 "tosa-bi",
10751 "tosa-mi"
10752 ],
10753 "generator_args": [
10754 [
10755 "--target-dtype",
10756 "bool",
10757 "--target-rank",
10758 "3",
10759 "--target-rank",
10760 "4",
10761 "--num-rand-permutations",
10762 "2"
10763 ]
10764 ]
10765 },
10766 "tosa-bi": {
10767 "supports_all": [
10768 "tosa-bi"
10769 ],
10770 "generator_args": [
10771 [
10772 "--target-dtype",
10773 "int8",
10774 "--target-rank",
10775 "1",
10776 "--target-rank",
10777 "2",
10778 "--num-rand-permutations",
10779 "2"
10780 ],
10781 [
10782 "--target-dtype",
10783 "int8",
10784 "--target-dtype",
10785 "int16",
10786 "--target-dtype",
10787 "int32",
10788 "--target-rank",
10789 "3",
10790 "--target-rank",
10791 "4",
10792 "--num-rand-permutations",
10793 "2"
10794 ],
10795 [
10796 "--target-dtype",
10797 "int8",
10798 "--target-shape",
10799 "1,65537,1,2",
10800 "--target-shape",
10801 "65535,1,1,1",
10802 "--num-rand-permutations",
10803 "1"
10804 ]
10805 ]
10806 },
10807 "8k_level": {
10808 "supports_all": [
10809 "tosa-bi"
10810 ],
10811 "no_negative_tests": "true",
10812 "generator_args": [
10813 [
10814 "--target-dtype",
10815 "int8",
10816 "--target-dtype",
10817 "int16",
10818 "--target-dtype",
10819 "int32",
10820 "--tensor-dim-range",
10821 "1,10",
10822 "--target-rank",
10823 "5",
10824 "--target-rank",
10825 "6",
10826 "--num-rand-permutations",
10827 "2"
10828 ]
10829 ]
10830 },
10831 "tosa-mi": {
10832 "supports_all": [
10833 "tosa-mi"
10834 ],
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]10835 "selector": "tosa-mi",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]10836 "generator_args": [
10837 [
10838 "--target-dtype",
10839 "fp32",
10840 "--target-dtype",
10841 "fp16",
10842 "--fp-values-range",
10843 "-max,max",
10844 "--target-rank",
10845 "1",
10846 "--target-rank",
10847 "2",
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]10848 "--target-rank",
10849 "3",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]10850 "--num-rand-permutations",
10851 "2"
10852 ],
10853 [
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]10854 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]10855 "--target-dtype",
10856 "fp32",
10857 "--target-dtype",
10858 "fp16",
10859 "--fp-values-range",
10860 "-max,max",
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]10861 "--tensor-dim-range",
10862 "1,12",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]10863 "--target-rank",
10864 "4",
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]10865 "--target-rank",
10866 "5",
10867 "--target-rank",
10868 "6",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]10869 "--num-rand-permutations",
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]10870 "1"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]10871 ],
10872 [
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]10873 "--no-special-tests",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]10874 "--target-dtype",
10875 "fp32",
10876 "--fp-values-range",
10877 "-max,max",
10878 "--target-shape",
10879 "1,1,65539,2",
10880 "--target-shape",
10881 "1,1,1,65535",
10882 "--num-rand-permutations",
10883 "1"
10884 ]
10885 ]
10886 },
10887 "tosa-ext-bf16": {
10888 "supports_all": [
10889 "tosa-ext-bf16"
10890 ],
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]10891 "selector": "tosa-mi",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]10892 "no_negative_tests": "true",
10893 "generator_args": [
10894 [
10895 "--target-dtype",
10896 "bf16",
10897 "--fp-values-range",
10898 "-max,max",
10899 "--target-rank",
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]10900 "1",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]10901 "--target-rank",
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]10902 "2",
10903 "--target-rank",
10904 "3",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]10905 "--num-rand-permutations",
10906 "2"
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]10907 ],
10908 [
10909 "--no-special-tests",
10910 "--target-dtype",
10911 "bf16",
10912 "--fp-values-range",
10913 "-max,max",
10914 "--tensor-dim-range",
10915 "1,12",
10916 "--target-rank",
10917 "4",
10918 "--target-rank",
10919 "5",
10920 "--target-rank",
10921 "6",
10922 "--num-rand-permutations",
10923 "1"
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]10924 ]
10925 ]
10926 },
10927 "tosa-ext-fp8e4m3": {
10928 "supports_all": [
10929 "tosa-ext-fp8e4m3"
10930 ],
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]10931 "selector": "tosa-mi",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]10932 "no_negative_tests": "true",
10933 "from_version": "v0.100.0",
10934 "generator_args": [
10935 [
10936 "--target-dtype",
10937 "fp8e4m3",
10938 "--fp-values-range",
10939 "-max,max",
10940 "--tensor-dim-range",
10941 "32,64",
10942 "--target-rank",
10943 "1",
10944 "--target-rank",
10945 "2",
10946 "--target-rank",
10947 "3",
10948 "--num-rand-permutations",
10949 "2"
10950 ]
10951 ]
10952 },
10953 "tosa-ext-fp8e5m2": {
10954 "supports_all": [
10955 "tosa-ext-fp8e4m3"
10956 ],
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]10957 "selector": "tosa-mi",
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]10958 "no_negative_tests": "true",
10959 "from_version": "v0.100.0",
10960 "generator_args": [
10961 [
10962 "--target-dtype",
10963 "fp8e5m2",
10964 "--fp-values-range",
10965 "-max,max",
10966 "--tensor-dim-range",
10967 "32,64",
10968 "--target-rank",
10969 "1",
10970 "--target-rank",
10971 "2",
10972 "--target-rank",
10973 "3",
10974 "--num-rand-permutations",
10975 "2"
10976 ]
10977 ]
10978 }
10979 },
10980 "selection": {
10981 "default": {
10982 "params": {},
10983 "permutes": [
10984 "shape",
10985 "type",
10986 "perm"
10987 ]
Jeremy Johnson32e7f442024-05-30 17:56:02 +0100[diff] [blame]10988 },
10989 "tosa-mi": {
10990 "generator_select": true
Won Jeon3eec59b2024-03-11 22:17:13 +0000[diff] [blame]10991 }
10992 }
10993 },
10994 "transpose_conv2d": {
10995 "group": "tensor",
10996 "support_for": [
10997 "lazy_data_gen"
10998 ],
10999 "gen_filter": "^transpose_conv2d",
11000 "generation": {
11001 "tosa-bi": {
11002 "supports_all": [
11003 "tosa-bi"
11004 ],
11005 "selector": "tosa-bi",
11006 "generator_args": [
11007 [
11008 "--target-dtype",
11009 "int8",
11010 "--target-shape",
11011 "1,49,33,1",
11012 "--target-shape",
11013 "1,11,33,3",
11014 "--tensor-dim-range",
11015 "1,18",
11016 "--allow-pooling-and-conv-oversizes"
11017 ],
11018 [
11019 "--target-dtype",
11020 "int8",
11021 "--target-shape",
11022 "1,65536,1,1",
11023 "--tensor-dim-range",
11024 "1,12",
11025 "--max-conv-dilation",
11026 "1"
11027 ]
11028 ]
11029 },
11030 "8k_level": {
11031 "supports_all": [
11032 "tosa-bi"
11033 ],
11034 "no_negative_tests": "true",
11035 "selector": "8k_level",
11036 "generator_args": [
11037 [
11038 "--target-dtype",
11039 "int8",
11040 "--tensor-dim-range",
11041 "2,5",
11042 "--level-8k-sizes"
11043 ]
11044 ]
11045 },
11046 "8k_level_ext_int16": {
11047 "supports_all": [
11048 "tosa-ext-int16"
11049 ],
11050 "no_negative_tests": "true",
11051 "selector": "8k_level",
11052 "generator_args": [
11053 [
11054 "--target-dtype",
11055 "int16",
11056 "--tensor-dim-range",
11057 "2,5",
11058 "--level-8k-sizes"
11059 ]
11060 ]
11061 },
11062 "tosa-mi": {
11063 "supports_all": [
11064 "tosa-mi"
11065 ],
11066 "selector": "tosa-mi",
11067 "negative_dim_range": "1,10",
11068 "generator_args": [
11069 [
11070 "--target-dtype",
11071 "fp32",
11072 "--target-dtype",
11073 "fp16",
11074 "--fp-values-range",
11075 "-max,max",
11076 "--target-shape",
11077 "10,24,9,13",
11078 "--target-shape",
11079 "8,14,20,5",
11080 "--tensor-dim-range",
11081 "1,16",
11082 "--allow-pooling-and-conv-oversizes"
11083 ],
11084 [
11085 "--target-dtype",
11086 "fp32",
11087 "--fp-values-range",
11088 "-max,max",
11089 "--target-shape",
11090 "1,65536,1,2",
11091 "--tensor-dim-range",
11092 "1,17",
11093 "--max-conv-dilation",
11094 "1"
11095 ]
11096 ]
11097 },
11098 "tosa-ext-bf16": {
11099 "supports_all": [
11100 "tosa-ext-bf16"
11101 ],
11102 "selector": "tosa-mi",
11103 "no_negative_tests": "true",
11104 "negative_dim_range": "1,10",
11105 "generator_args": [
11106 [
11107 "--target-dtype",
11108 "bf16",
11109 "--fp-values-range",
11110 "-max,max",
11111 "--target-shape",
11112 "10,24,9,13",
11113 "--target-shape",
11114 "8,14,20,5",
11115 "--tensor-dim-range",
11116 "1,16",
11117 "--allow-pooling-and-conv-oversizes"
11118 ]
11119 ]
11120 },
11121 "tosa-ext-fp8e4m3": {
11122 "supports_all": [
11123 "tosa-ext-fp8e4m3"
11124 ],
11125 "selector": "tosa-mi",
11126 "no_negative_tests": "true",
11127 "from_version": "v0.100.0",
11128 "generator_args": [
11129 [
11130 "--target-dtype",
11131 "fp8e4m3",
11132 "--fp-values-range",
11133 "-max,max",
11134 "--tensor-dim-range",
11135 "32,64",
11136 "--target-shape",
11137 "10,24,9,13",
11138 "--target-shape",
11139 "8,14,20,5",
11140 "--tensor-dim-range",
11141 "1,16",
11142 "--num-rand-permutations",
11143 "2"
11144 ]
11145 ]
11146 },
11147 "tosa-ext-fp8e5m2": {
11148 "supports_all": [
11149 "tosa-ext-fp8e5m2"
11150 ],
11151 "selector": "tosa-mi",
11152 "no_negative_tests": "true",
11153 "from_version": "v0.100.0",
11154 "generator_args": [
11155 [
11156 "--target-dtype",
11157 "fp8e5m2",
11158 "--fp-values-range",
11159 "-max,max",
11160 "--tensor-dim-range",
11161 "32,64",
11162 "--target-shape",
11163 "10,24,9,13",
11164 "--target-shape",
11165 "8,14,20,5",
11166 "--tensor-dim-range",
11167 "1,16",
11168 "--num-rand-permutations",
11169 "2"
11170 ]
11171 ]
11172 },
11173 "tosa-ext-int16": {
11174 "supports_all": [
11175 "tosa-ext-int16"
11176 ],
11177 "selector": "tosa-bi",
11178 "no_negative_tests": "true",
11179 "generator_args": [
11180 [
11181 "--target-dtype",
11182 "int16",
11183 "--target-shape",
11184 "1,49,33,1",
11185 "--target-shape",
11186 "1,11,33,3",
11187 "--tensor-dim-range",
11188 "1,18",
11189 "--allow-pooling-and-conv-oversizes"
11190 ]
11191 ]
11192 }
11193 },
11194 "selection": {
11195 "tosa-bi": {
11196 "params": {
11197 "shape": [],
11198 "type": [
11199 "i8xi8",
11200 "i16xi8",
11201 "i8xi4"
11202 ],
11203 "kernel": [],
11204 "stride": [],
11205 "pad": [],
11206 "out_shape": []
11207 },
11208 "permutes": [
11209 "shape",
11210 "type",
11211 "stride",
11212 "pad"
11213 ]
11214 },
11215 "8k_level": {
11216 "all": "true"
11217 },
11218 "tosa-mi": {
11219 "params": {
11220 "shape": [],
11221 "type": [],
11222 "kernel": [],
11223 "stride": [],
11224 "pad": [],
11225 "out_shape": []
11226 },
11227 "permutes": [
11228 "shape",
11229 "type",
11230 "stride",
11231 "pad"
11232 ]
11233 }
11234 }
11235 },
11236 "while_loop": {
11237 "group": "control_flow",
11238 "generation": {
11239 "standard": {
11240 "supports_any": [
11241 "tosa-bi",
11242 "tosa-mi"
11243 ],
11244 "generator_args": [
11245 [
11246 "--target-dtype",
11247 "int32"
11248 ],
11249 [
11250 "--target-dtype",
11251 "int32",
11252 "--target-shape",
11253 "3,22,17,1,2",
11254 "--target-shape",
11255 "1,65537,2,1"
11256 ]
11257 ]
11258 }
11259 },
11260 "selection": {
11261 "default": {
11262 "params": {},
11263 "permutes": [
11264 "shape",
11265 "type"
11266 ]
11267 }
11268 }
11269 }
IanTayler1a083b52024-05-10 09:35:24 +0100[diff] [blame^]11270}