diff options
Diffstat (limited to 'verif/conformance/tosa_ext_profile_ops_info.json')
| -rw-r--r-- | verif/conformance/tosa_ext_profile_ops_info.json | 10880 |
1 files changed, 10880 insertions, 0 deletions
diff --git a/verif/conformance/tosa_ext_profile_ops_info.json b/verif/conformance/tosa_ext_profile_ops_info.json new file mode 100644 index 0000000..9ca19db --- /dev/null +++ b/verif/conformance/tosa_ext_profile_ops_info.json @@ -0,0 +1,10880 @@ +{ + "abs": { + "group": "ew_unary", + "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ], + "generation": { + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "selector": "tosa-bi", + "generator_args": [ + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "int32", + "--target-shape", + "1,2,65533,1", + "--target-shape", + "1,65539,1,1,1" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "selector": "tosa-bi", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "selector": "tosa-mi", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "30,64", + "--target-rank", + "0", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "selector": "tosa-mi", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "16,64", + "--target-rank", + "0", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,65531,2,1", + "--target-shape", + "65538,1,1,1" + ] + ] + } + }, + "selection": { + "tosa-bi": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "tosa-mi": { + "generator_select": "true" + } + } + }, + "add": { + "group": "ew_binary", + "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ], + "generation": { + "tosa-bi-mi": { + "supports_any": [ + "tosa-bi", + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "int32", + "--target-shape", + "1,2,65533,1", + "--target-shape", + "1,65539,1,1,1" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "selector": "tosa-mi", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "16,64", + "--target-rank", + "0", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--target-shape", + "2,1,1,65532", + "--target-shape", + "65542,2,1,1,1" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "selector": "tosa-mi", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "16,64", + "--target-rank", + "0", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "tosa-mi": { + "generator_select": "true" + } + } + }, + "add_shape": { + "group": "shape", + "generation": { + "tosa-bi-mi": { + "supports_any": [ + "tosa-bi", + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "shape", + "--tensor-dim-range", + "1,16", + "--target-rank", + "1" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape" + ] + } + } + }, + "argmax": { + "group": "tensor", + "support_for": [ + "lazy_data_gen" + ], + "generation": { + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "generator_args": [ + [ + "--target-dtype", + "int8", + "--tensor-dim-range", + "32,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ], + [ + "--target-dtype", + "int8", + "--tensor-dim-range", + "1,32", + "--target-rank", + "4", + "--num-rand-permutations", + "2" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "from_version": "v0.70.0", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--tensor-dim-range", + "1,10", + "--target-rank", + "5", + "--target-rank", + "6", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "32,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ], + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,32", + "--target-rank", + "4", + "--num-rand-permutations", + "2" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,3,65535,1", + "--target-shape", + "1,65537,1,2", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ], + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,32", + "--target-rank", + "4", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-ext-fp8e4m3": { + "supports_all": [ + "tosa-ext-fp8e4m3" + ], + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e4m3", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-ext-fp8e5m2": { + "supports_all": [ + "tosa-ext-fp8e5m2" + ], + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e5m2", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-ext-int16": { + "supports_all": [ + "tosa-ext-int16" + ], + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "int16", + "--tensor-dim-range", + "32,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ], + [ + "--target-dtype", + "int16", + "--tensor-dim-range", + "1,32", + "--target-rank", + "4", + "--num-rand-permutations", + "2" + ] + ] + }, + "8k_level_ext-int16": { + "supports_all": [ + "tosa-ext-int16" + ], + "from_version": "v0.70.0", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int16", + "--tensor-dim-range", + "1,10", + "--target-rank", + "5", + "--target-rank", + "6", + "--num-rand-permutations", + "2" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + } + } + }, + "arithmetic_right_shift": { + "group": "ew_binary", + "generation": { + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "int8", + "--target-shape", + "1,2,65533,1", + "--target-shape", + "1,65539,1,1,1" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type", + "rounding" + ] + } + } + }, + "avg_pool2d": { + "group": "tensor", + "support_for": [ + "lazy_data_gen" + ], + "generation": { + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "selector": "tosa-bi", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-shape", + "1,49,42,28", + "--target-shape", + "1,11,44,3", + "--allow-pooling-and-conv-oversizes" + ], + [ + "--target-dtype", + "int8", + "--target-shape", + "1,65535,5,1", + "--target-shape", + "1,3,65537,1", + "--allow-pooling-and-conv-oversizes" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "selector": "8k_level", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--tensor-dim-range", + "3,10", + "--level-8k-sizes" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "selector": "tosa-mi", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--target-shape", + "4,11,7,39", + "--target-shape", + "5,21,36,17", + "--allow-pooling-and-conv-oversizes" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,2,65533,2", + "--target-shape", + "1,65539,2,1", + "--allow-pooling-and-conv-oversizes" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "selector": "tosa-mi", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--target-shape", + "4,11,7,39", + "--target-shape", + "5,21,36,17", + "--allow-pooling-and-conv-oversizes" + ] + ] + }, + "tosa-ext-fp8e4m3": { + "supports_all": [ + "tosa-ext-fp8e4m3" + ], + "selector": "tosa-mi", + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e4m3", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-ext-fp8e5m2": { + "supports_all": [ + "tosa-ext-fp8e5m2" + ], + "selector": "tosa-mi", + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e5m2", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-ext-int16": { + "supports_all": [ + "tosa-ext-int16" + ], + "selector": "tosa-bi", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int16", + "--target-shape", + "1,49,42,28", + "--target-shape", + "1,11,44,3", + "--allow-pooling-and-conv-oversizes" + ] + ] + }, + "8k_level_ext_int16": { + "supports_all": [ + "tosa-ext-int16" + ], + "no_negative_tests": "true", + "selector": "8k_level", + "generator_args": [ + [ + "--target-dtype", + "int16", + "--tensor-dim-range", + "3,10", + "--level-8k-sizes" + ] + ] + } + }, + "selection": { + "tosa-bi": { + "params": { + "shape": [], + "type": [ + "i8", + "i16" + ], + "kernel": [], + "stride": [], + "pad": [] + }, + "permutes": [ + "shape", + "type", + "pad" + ] + }, + "tosa-mi": { + "params": {}, + "permutes": [ + "shape", + "type" + ], + "sparsity": { + "pad": 17 + } + }, + "8k_level": { + "all": "true" + } + } + }, + "bitwise_and": { + "group": "ew_binary", + "generation": { + "standard": { + "supports_all": [ + "tosa-bi" + ], + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int8", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "int8", + "--target-shape", + "1,2,65536,1", + "--target-shape", + "1,65539,1,1,1" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + } + } + }, + "bitwise_not": { + "group": "ew_unary", + "generation": { + "standard": { + "supports_all": [ + "tosa-bi" + ], + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "2,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int8", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "int8", + "--target-shape", + "1,2,65537,1", + "--target-shape", + "1,65538,1,1,1" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + } + } + }, + "bitwise_or": { + "group": "ew_binary", + "generation": { + "standard": { + "supports_all": [ + "tosa-bi" + ], + "selector": "tosa-bi", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "3,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int8", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "int8", + "--target-shape", + "1,2,65538,1", + "--target-shape", + "1,65537,1,1,1" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + } + } + }, + "bitwise_xor": { + "group": "ew_binary", + "generation": { + "standard": { + "supports_all": [ + "tosa-bi" + ], + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "4,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int8", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "int8", + "--target-shape", + "1,2,65536,1", + "--target-shape", + "1,65536,1,1,1" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + } + } + }, + "cast": { + "group": "type_conversion", + "support_for": [ + "lazy_data_gen" + ], + "generation": { + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "selector": "tosa-bi", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--target-dtype", + "bool", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "int8", + "--target-shape", + "1,1,65535,1,1", + "--target-shape", + "1,65538,1,2" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "selector": "tosa-bi", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "selector": "tosa-mi", + "negative_dim_range": "1,10", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,1,1,65533,1", + "--target-shape", + "2,65538,1,1" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "selector": "tosa-ext-bf16", + "no_negative_tests": "true", + "negative_dim_range": "1,10", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "bf16", + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "fp32", + "--target-dtype", + "bf16", + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ] + ] + }, + "tosa-ext-bf16-fp8e4m3": { + "supports_all": [ + "tosa-ext-bf16", + "tosa-ext-fp8e4m3" + ], + "selector": "tosa-ext-bf16-fp8e4m3", + "no_negative_tests": "true", + "negative_dim_range": "1,10", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--target-dtype", + "fp8e4m3", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "bf16", + "--target-dtype", + "fp8e4m3", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ] + ] + }, + "tosa-ext-bf16-fp8e5m2": { + "supports_all": [ + "tosa-ext-bf16", + "tosa-ext-fp8e5m2" + ], + "selector": "tosa-ext-bf16-fp8e5m2", + "no_negative_tests": "true", + "negative_dim_range": "1,10", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--target-dtype", + "fp8e5m2", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "bf16", + "--target-dtype", + "fp8e5m2", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ] + ] + }, + "tosa-ext-fp8e4m3": { + "supports_all": [ + "tosa-ext-fp8e4m3" + ], + "no_negative_tests": "true", + "selector": "tosa-ext-fp8e4m3", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e4m3", + "--target-dtype", + "fp16", + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-ext-fp8e5m2": { + "supports_all": [ + "tosa-ext-fp8e5m2" + ], + "selector": "tosa-ext-fp8e5m2", + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e5m2", + "--target-dtype", + "fp16", + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ] + ] + } + }, + "selection": { + "tosa-bi": { + "params": { + "shape": [], + "type": [], + "output_type": [ + "outi8", + "outi16", + "outi32", + "outb" + ] + }, + "permutes": [ + "shape", + "type", + "output_type" + ], + "exclude_patterns": [ + "^((?!ERRORIF).)*_out(f16|f32)" + ] + }, + "tosa-mi": { + "params": { + "shape": [], + "type": [], + "output_type": [ + "outi8", + "outi16", + "outi32", + "outf16", + "outf32" + ] + }, + "permutes": [ + "shape", + "type", + "output_type" + ], + "exclude_patterns": [ + "^((?!ERRORIF).)*_(b|i8|i16|i32|bf16|f8e4m3|f8e5m2)_out(b|i8|i16|i32|bf16|f8e4m3|f8e5m2)" + ] + }, + "tosa-ext-bf16": { + "params": { + "shape": [], + "type": [], + "output_type": [ + "outi8", + "outi16", + "outi32", + "outf32", + "outbf16" + ] + }, + "permutes": [ + "shape", + "type", + "output_type" + ], + "exclude_patterns": [ + "^((?!ERRORIF).)*_(b|i8|i16|i32|f16|f32)_out(b|i8|i16|i32|f16|f32)", + "^((?!ERRORIF).)*_bf16_out(b|f16|f8e4m3|f8e5m2)", + "^((?!ERRORIF).)*_(b|f16|f8e4m3|f8e5m2)_outbf16" + ] + }, + "tosa-ext-fp8e4m3": { + "params": { + "shape": [], + "type": [], + "output_type": [ + "outf16", + "outf32", + "outf8e4m3" + ] + }, + "permutes": [ + "shape", + "type", + "output_type" + ], + "exclude_patterns": [ + "^((?!ERRORIF).)*_(b|i8|i16|i32|f16|bf16|f32)_out(b|i8|i16|i32|f16|bf16|f32|f8e5m2)", + "^((?!ERRORIF).)*_f8e4m3_out(b|i8|i16|i32|bf16|f8e4m3|f8e5m2)", + "^((?!ERRORIF).)*_(b|i8|i16|i32|bf16|f8e4m3|f8e5m2)_outf8e4m3" + ] + }, + "tosa-ext-fp8e5m2": { + "params": { + "shape": [], + "type": [], + "output_type": [ + "outf16", + "outf32", + "outf8e5m2" + ] + }, + "permutes": [ + "shape", + "type", + "output_type" + ], + "exclude_patterns": [ + "^((?!ERRORIF).)*_(b|i8|i16|i32|f16|bf16|f32)_out(b|i8|i16|i32|f16|bf16|f32|f8e4m3)", + "^((?!ERRORIF).)*_f8e5m2_out(b|i8|i16|i32|bf16|f8e4m3|f8e5m2)", + "^((?!ERRORIF).)*_(b|i8|i16|i32|bf16|f8e4m3|f8e5m2)_outf8e5m2" + ] + }, + "tosa-ext-bf16-fp8e4m3": { + "params": { + "shape": [], + "type": [], + "output_type": [ + "outf8e4m3", + "outbf16" + ] + }, + "permutes": [ + "shape", + "type", + "output_type" + ], + "exclude_patterns": [ + "^((?!ERRORIF).)*_(b|i8|i16|f16|f32|f8e5m2)_out(b|i8|i16|i32|f16|f32|f8e5m2)", + "^((?!ERRORIF).)*_(bf16|f8e4m3)_out(b|i8|i16|i32|f16|f32|f8e5m2)", + "^((?!ERRORIF).)*_(b|i8 d|i16|i32|f16|f32|f8e5m2)_out(bf16|f8e4m3)" + ] + }, + "tosa-ext-bf16-fp8e5m2": { + "params": { + "shape": [], + "type": [], + "output_type": [ + "outf8e5m2", + "outbf16" + ] + }, + "permutes": [ + "shape", + "type", + "output_type" + ], + "exclude_patterns": [ + "^((?!ERRORIF).)*_(b|i8|i16|f16|f32|f8e4m3)_out(b|i8|i16|i32|f16|f32|f8e4m3)", + "^((?!ERRORIF).)*_(bf16|f8e5m2)_out(b|i8|i16|i32|f16|f32|f8e4m3)", + "^((?!ERRORIF).)*_(b|i8 d|i16|i32|f16|f32|f8e4m3)_out(bf16|f8e5m2)" + ] + } + } + }, + "ceil": { + "group": "ew_unary", + "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ], + "generation": { + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "24,64", + "--target-rank", + "0", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--target-shape", + "2,1,65530,1", + "--target-shape", + "1,1,65539,1" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "4,64", + "--target-rank", + "0", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ] + ] + } + }, + "selection": { + "default": { + "generator_select": "true" + } + } + }, + "clamp": { + "group": "activation", + "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ], + "generation": { + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "generator_args": [ + [ + "--target-dtype", + "int8", + "--tensor-dim-range", + "20,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int8", + "--tensor-dim-range", + "1,20", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "int8", + "--target-shape", + "1,2,1,65536", + "--target-shape", + "1,3,65537,1,1" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "selector": "tosa-mi", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "20,64", + "--target-rank", + "0", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,20", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,2,1,65536", + "--target-shape", + "1,3,65537,1,1" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "selector": "tosa-mi", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "20,64", + "--target-rank", + "0", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,20", + "--target-rank", + "4", + "--target-rank", + "5" + ] + ] + }, + "tosa-ext-int16": { + "supports_all": [ + "tosa-ext-int16" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int16", + "--tensor-dim-range", + "20,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int16", + "--tensor-dim-range", + "1,20", + "--target-rank", + "4", + "--target-rank", + "5" + ] + ] + }, + "8k_level_ext_int16": { + "supports_all": [ + "tosa-ext-int16" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int16", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "tosa-mi": { + "generator_select": "true" + } + } + }, + "clz": { + "group": "ew_unary", + "generation": { + "standard": { + "supports_all": [ + "tosa-bi" + ], + "selector": "tosa-bi", + "generator_args": [ + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "int32", + "--target-shape", + "1,2,65533,1", + "--target-shape", + "1,65539,1,1,1" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + } + } + }, + "concat": { + "group": "data_layout", + "support_for": [ + "lazy_data_gen" + ], + "generation": { + "tosa-bi-mi": { + "supports_any": [ + "tosa-bi", + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "bool", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-const-inputs-concat", + "3" + ], + [ + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5", + "--num-const-inputs-concat", + "1" + ] + ] + }, + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int32", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-const-inputs-concat", + "3" + ], + [ + "--target-dtype", + "int8", + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5", + "--num-const-inputs-concat", + "1" + ], + [ + "--target-dtype", + "int8", + "--target-shape", + "1,1,65535,1,1", + "--target-shape", + "1,65538,1,2", + "--num-const-inputs-concat", + "2" + ] + ] + }, + "tosa-ext-int16": { + "supports_all": [ + "tosa-ext-int16" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int16", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-const-inputs-concat", + "3" + ], + [ + "--target-dtype", + "int16", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5", + "--num-const-inputs-concat", + "1" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6", + "--num-const-inputs-concat", + "2" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-const-inputs-concat", + "3" + ], + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5", + "--num-const-inputs-concat", + "1" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,1,65534,1,1", + "--target-shape", + "1,65537,1,2", + "--num-const-inputs-concat", + "2" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-const-inputs-concat", + "3" + ], + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5", + "--num-const-inputs-concat", + "1" + ] + ] + }, + "tosa-ext-fp8e4m3": { + "supports_all": [ + "tosa-ext-fp8e4m3" + ], + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e4m3", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-ext-fp8e5m2": { + "supports_all": [ + "tosa-ext-fp8e5m2" + ], + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e5m2", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + } + } + }, + "concat_shape": { + "group": "shape", + "generation": { + "tosa-bi-mi": { + "supports_any": [ + "tosa-bi", + "tosa-mi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "shape", + "--target-rank", + "1", + "--target-shape", + "1", + "--num-const-inputs-concat", + "2" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [] + } + } + }, + "cond_if": { + "group": "control_flow", + "gen_filter": "^cond_if", + "generation": { + "standard": { + "supports_any": [ + "tosa-bi", + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "bool", + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--target-rank", + "4" + ], + [ + "--target-dtype", + "int32", + "--target-shape", + "3,22,17,1,2" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "variant", + "shape", + "cond" + ] + } + } + }, + "const": { + "group": "data_nodes", + "support_for": [ + "lazy_data_gen" + ], + "generation": { + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--target-dtype", + "bool", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "int8", + "--target-shape", + "1,1,65537,1,1", + "--target-shape", + "1,65530,1,2" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,65540,1,1,1", + "--target-shape", + "1,2,1,65529" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ] + ] + }, + "tosa-ext-fp8e4m3": { + "supports_all": [ + "tosa-ext-fp8e4m3" + ], + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e4m3", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-ext-fp8e5m2": { + "supports_all": [ + "tosa-ext-fp8e5m2" + ], + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e5m2", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-ext-int16": { + "supports_all": [ + "tosa-ext-int16" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int48", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int48", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + } + } + }, + "const_shape": { + "group": "shape", + "generation": { + "standard": { + "supports_any": [ + "tosa-bi", + "tosa-mi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "shape", + "--target-rank", + "1", + "--target-shape", + "1" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [] + } + } + }, + "conv2d": { + "group": "tensor", + "support_for": [ + "lazy_data_gen", + "stable_random_gen" + ], + "gen_filter": "^conv2d", + "generation": { + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "selector": "tosa-bi", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-shape", + "1,49,42,28", + "--target-shape", + "1,11,44,13", + "--tensor-dim-range", + "1,16", + "--allow-pooling-and-conv-oversizes" + ], + [ + "--target-dtype", + "int8", + "--target-shape", + "1,65535,3,1", + "--target-shape", + "1,2,65536,1", + "--max-conv-dilation", + "1", + "--tensor-dim-range", + "1,8", + "--allow-pooling-and-conv-oversizes" + ] + ] + }, + "tosa-ext-int16": { + "supports_all": [ + "tosa-ext-int16" + ], + "selector": "tosa-ext-int16", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int16", + "--target-shape", + "1,49,42,28", + "--target-shape", + "1,11,44,13", + "--tensor-dim-range", + "1,16", + "--allow-pooling-and-conv-oversizes" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "selector": "8k_level", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--tensor-dim-range", + "2,5", + "--level-8k-sizes" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "selector": "tosa-mi", + "negative_dim_range": "1,10", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,34,19,27", + "--target-shape", + "1,16,51,7", + "--tensor-dim-range", + "1,16", + "--allow-pooling-and-conv-oversizes" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,65537,1,3", + "--target-shape", + "1,2,65530,2", + "--tensor-dim-range", + "1,16", + "--max-conv-dilation", + "1", + "--allow-pooling-and-conv-oversizes" + ], + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--conv-kernel", + "7,7", + "--conv-kernel", + "9,9", + "--target-shape", + "1,127,127,1", + "--target-shape", + "1,256,256,1", + "--target-shape", + "1,1080,720,1", + "--tensor-dim-range", + "1,8" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "selector": "tosa-mi", + "no_negative_tests": "true", + "negative_dim_range": "1,10", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,34,19,27", + "--target-shape", + "1,16,51,7", + "--tensor-dim-range", + "1,16", + "--allow-pooling-and-conv-oversizes" + ], + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--conv-kernel", + "7,7", + "--conv-kernel", + "9,9", + "--target-shape", + "1,127,127,1", + "--target-shape", + "1,256,256,1", + "--target-shape", + "1,1080,720,1", + "--tensor-dim-range", + "1,8" + ] + ] + }, + "tosa-ext-fp8e4m3": { + "supports_all": [ + "tosa-ext-fp8e4m3" + ], + "selector": "tosa-mi", + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e4m3", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-ext-fp8e5m2": { + "supports_all": [ + "tosa-ext-fp8e5m2" + ], + "selector": "tosa-mi", + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e5m2", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--num-rand-permutations", + "2" + ] + ] + }, + "8k_level_ext-int16": { + "supports_all": [ + "tosa-ext-int16" + ], + "no_negative_tests": "true", + "selector": "8k_level", + "generator_args": [ + [ + "--target-dtype", + "int16", + "--tensor-dim-range", + "2,5", + "--level-8k-sizes" + ] + ] + } + }, + "selection": { + "tosa-bi": { + "params": { + "shape": [], + "type": [ + "i8xi8" + ], + "kernel": [], + "stride": [], + "pad": [], + "dilation": [] + }, + "permutes": [ + "kernel", + "shape", + "type", + "pad" + ] + }, + "tosa-ext-int16": { + "params": { + "shape": [], + "type": [ + "i16xi8" + ], + "kernel": [], + "stride": [], + "pad": [], + "dilation": [] + }, + "permutes": [ + "kernel", + "shape", + "type", + "pad" + ] + }, + "tosa-ext-int4": { + "params": { + "shape": [], + "type": [ + "i8xi4" + ], + "kernel": [], + "stride": [], + "pad": [], + "dilation": [] + }, + "permutes": [ + "kernel", + "shape", + "type", + "pad" + ] + }, + "8k_level": { + "all": "true" + }, + "tosa-mi": { + "generator_select": "true", + "permutes": [ + "shape", + "type", + "kernel", + "acc_type" + ], + "full_params": [ + "stride", + "dilation" + ], + "specifics": { + "pad": [ + "(0, 0, 0, 0)" + ] + }, + "groups": [ + "s" + ], + "maximum": 3 + } + } + }, + "conv3d": { + "group": "tensor", + "support_for": [ + "lazy_data_gen" + ], + "gen_filter": "^conv3d", + "generation": { + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "selector": "tosa-bi", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-shape", + "1,9,21,14,1", + "--target-shape", + "1,9,11,12,3", + "--tensor-dim-range", + "1,4", + "--allow-pooling-and-conv-oversizes" + ], + [ + "--target-dtype", + "int8", + "--target-shape", + "1,1,65535,2,1", + "--target-shape", + "1,1,2,65536,1", + "--max-conv-dilation", + "1", + "--tensor-dim-range", + "1,2", + "--allow-pooling-and-conv-oversizes" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "selector": "8k_level", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--tensor-dim-range", + "2,5", + "--level-8k-sizes" + ] + ] + }, + "8k_level_ext_int16": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "selector": "8k_level", + "generator_args": [ + [ + "--target-dtype", + "int16", + "--tensor-dim-range", + "2,5", + "--level-8k-sizes" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "selector": "tosa-mi", + "negative_dim_range": "1,10", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,7,18,5,4", + "--target-shape", + "1,6,12,17,3", + "--tensor-dim-range", + "1,4", + "--allow-pooling-and-conv-oversizes" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,2,65539,1,2", + "--target-shape", + "1,65532,1,1,3", + "--tensor-dim-range", + "1,2", + "--max-conv-dilation", + "1", + "--allow-pooling-and-conv-oversizes" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "selector": "tosa-mi", + "no_negative_tests": "true", + "negative_dim_range": "1,10", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,7,18,5,4", + "--target-shape", + "1,6,12,17,3", + "--tensor-dim-range", + "1,4", + "--allow-pooling-and-conv-oversizes" + ] + ] + }, + "tosa-ext-fp8e4m3": { + "supports_all": [ + "tosa-ext-fp8e4m3" + ], + "selector": "tosa-mi", + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e4m3", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,7,18,5,4", + "--target-shape", + "1,6,12,17,3", + "--tensor-dim-range", + "1,4", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-ext-fp8e5m2": { + "supports_all": [ + "tosa-ext-fp8e5m2" + ], + "selector": "tosa-mi", + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e5m2", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,7,18,5,4", + "--target-shape", + "1,6,12,17,3", + "--tensor-dim-range", + "1,4", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-ext-int16": { + "supports_all": [ + "tosa-ext-int16" + ], + "selector": "tosa-ext-int16", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int16", + "--target-shape", + "1,9,21,14,1", + "--target-shape", + "1,9,11,12,3", + "--tensor-dim-range", + "1,4", + "--allow-pooling-and-conv-oversizes" + ] + ] + } + }, + "selection": { + "tosa-bi": { + "params": { + "shape": [], + "type": [ + "i8xi8" + ], + "kernel": [], + "stride": [], + "pad": [], + "dilation": [] + }, + "permutes": [ + "kernel", + "shape", + "type", + "pad", + "stride" + ], + "exclude_patterns": [ + "^((?!ERRORIF).)*_(i16xi8|i8xi4)_.*" + ] + }, + "tosa-ext-int16": { + "params": { + "shape": [], + "type": [ + "i16xi8" + ], + "kernel": [], + "stride": [], + "pad": [], + "dilation": [] + }, + "permutes": [ + "kernel", + "shape", + "type", + "pad", + "stride" + ], + "exclude_patterns": [ + "^((?!ERRORIF).)*_(i8xi8|i8xi4)_.*" + ] + }, + "tosa-ext-int4": { + "params": { + "shape": [], + "type": [ + "i8xi4" + ], + "kernel": [], + "stride": [], + "pad": [], + "dilation": [] + }, + "permutes": [ + "kernel", + "shape", + "type", + "pad", + "stride" + ], + "exclude_patterns": [ + "^((?!ERRORIF).)*_(i8xi8|i16xi8)_.*" + ] + }, + "8k_level": { + "all": "true" + }, + "tosa-mi": { + "params": { + "shape": [], + "type": [], + "kernel": [], + "stride": [], + "pad": [], + "dilation": [] + }, + "permutes": [ + "kernel", + "shape", + "type", + "pad", + "stride" + ] + } + } + }, + "cos": { + "group": "ew_unary", + "support_for": [ + "lazy_data_gen" + ], + "generation": { + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "selector": "tosa-mi", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "15,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,15", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--target-shape", + "2,1,65537,1", + "--target-shape", + "3,1,65539,2,1" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "selector": "tosa-mi", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "15,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ] + ] + } + }, + "selection": { + "tosa-mi": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + } + } + }, + "depthwise_conv2d": { + "group": "tensor", + "support_for": [ + "lazy_data_gen" + ], + "gen_filter": "^depthwise_conv2d", + "generation": { + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "selector": "tosa-bi", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-shape", + "1,9,44,3", + "--target-shape", + "1,11,33,4", + "--tensor-dim-range", + "1,16", + "--allow-pooling-and-conv-oversizes" + ], + [ + "--target-dtype", + "int8", + "--target-shape", + "1,65532,2,1", + "--target-shape", + "1,4,65537,1", + "--max-conv-dilation", + "1", + "--tensor-dim-range", + "1,16", + "--allow-pooling-and-conv-oversizes" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "selector": "8k_level", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--tensor-dim-range", + "2,5", + "--level-8k-sizes" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "selector": "tosa-mi", + "negative_dim_range": "1,10", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,17,31,4", + "--target-shape", + "1,37,11,5", + "--tensor-dim-range", + "1,32", + "--allow-pooling-and-conv-oversizes" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,1,65531,2", + "--target-shape", + "1,65540,1,3", + "--tensor-dim-range", + "1,16", + "--max-conv-dilation", + "1", + "--allow-pooling-and-conv-oversizes" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "selector": "tosa-ext-bf16", + "no_negative_tests": "true", + "negative_dim_range": "1,10", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,17,31,4", + "--target-shape", + "1,37,11,5", + "--tensor-dim-range", + "1,32", + "--allow-pooling-and-conv-oversizes" + ] + ] + }, + "tosa-ext-fp8e4m3": { + "supports_all": [ + "tosa-ext-fp8e4m3" + ], + "selector": "tosa-ext-fp8e4m3", + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e4m3", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-ext-fp8e5m2": { + "supports_all": [ + "tosa-ext-fp8e5m2" + ], + "selector": "tosa-ext-fp8e5m2", + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e5m2", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-ext-int16": { + "supports_all": [ + "tosa-ext-int16" + ], + "selector": "tosa-bi", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int16", + "--target-shape", + "1,9,44,3", + "--target-shape", + "1,11,33,4", + "--tensor-dim-range", + "1,16", + "--allow-pooling-and-conv-oversizes" + ] + ] + } + }, + "selection": { + "tosa-bi": { + "params": { + "shape": [], + "type": [ + "i8xi8", + "i16xi8", + "i8xi4" + ], + "kernel": [], + "stride": [], + "pad": [], + "dilation": [] + }, + "permutes": [ + "kernel", + "shape", + "type", + "pad" + ] + }, + "8k_level": { + "all": "true" + }, + "tosa-mi": { + "params": { + "shape": [], + "type": [ + "f16xf16", + "f32xf32" + ], + "kernel": [], + "stride": [], + "pad": [], + "dilation": [] + }, + "permutes": [ + "kernel", + "shape", + "type", + "pad" + ] + }, + "tosa-ext-bf16": { + "params": { + "shape": [], + "type": [ + "bf16xbf16" + ], + "kernel": [], + "stride": [], + "pad": [], + "dilation": [] + }, + "permutes": [ + "kernel", + "shape", + "type", + "pad" + ] + }, + "tosa-ext-fp8e4m3": { + "params": { + "shape": [], + "type": [ + "f8e4m3xf8e4m3" + ], + "kernel": [], + "stride": [], + "pad": [], + "dilation": [] + }, + "permutes": [ + "kernel", + "shape", + "type", + "pad" + ] + }, + "tosa-ext-fp8e5m2": { + "params": { + "shape": [], + "type": [ + "f8e5m2xf8e5m2" + ], + "kernel": [], + "stride": [], + "pad": [], + "dilation": [] + }, + "permutes": [ + "kernel", + "shape", + "type", + "pad" + ] + } + } + }, + "dim": { + "group": "data_layout", + "generation": { + "tosa-bi-mi": { + "supports_any": [ + "tosa-bi", + "tosa-mi" + ], + "from_version": "v0.80.0", + "generator_args": [ + [ + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ] + ] + }, + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "from_version": "v0.80.0", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "from_version": "v0.80.0", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "from_version": "v0.80.0", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-2.0,2.0", + "--tensor-dim-range", + "1,65", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "no_negative_tests": "true", + "from_version": "v0.80.0", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-2.0,2.0", + "--tensor-dim-range", + "1,65", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ] + ] + }, + "tosa-ext-fp8e4m3": { + "supports_all": [ + "tosa-ext-fp8e4m3" + ], + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e4m3", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-ext-fp8e5m2": { + "supports_all": [ + "tosa-ext-fp8e5m2" + ], + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e5m2", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + } + } + }, + "div_shape": { + "group": "shape", + "generation": { + "tosa-bi-mi": { + "supports_all": [ + "tosa-bi", + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "shape", + "--tensor-dim-range", + "1,16", + "--target-rank", + "1" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape" + ] + } + } + }, + "equal": { + "group": "comparison", + "support_for": [ + "lazy_data_gen" + ], + "generation": { + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "generator_args": [ + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "int32", + "--target-shape", + "1,2,65501,1", + "--target-shape", + "1,65541,1,1,1" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "18,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,18", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,1,2,65550", + "--target-shape", + "1,4,65530,1,1" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "18,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + } + } + }, + "erf": { + "group": "activation", + "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ], + "generation": { + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-4.0,4.0", + "--tensor-dim-range", + "18,60", + "--target-rank", + "0", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-4.0,4.0", + "--tensor-dim-range", + "1,24", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-4.0,4.0", + "--target-shape", + "1,65535,2,1,1", + "--target-shape", + "1,65540,1,2" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-4.0,4.0", + "--tensor-dim-range", + "18,60", + "--target-rank", + "0", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-4.0,4.0", + "--tensor-dim-range", + "1,24", + "--target-rank", + "4", + "--target-rank", + "5" + ] + ] + } + }, + "selection": { + "default": { + "generator_select": "true" + } + } + }, + "exp": { + "group": "ew_unary", + "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ], + "generation": { + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "25,64", + "--target-rank", + "0", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,15", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,1,65535,4", + "--target-shape", + "2,65536,1,1,1" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "15,64", + "--target-rank", + "0", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ] + ] + } + }, + "selection": { + "default": { + "generator_select": "true" + } + } + }, + "fft2d": { + "group": "tensor", + "support_for": [ + "lazy_data_gen" + ], + "generation": { + "tosa-ext-fft": { + "supports_all": [ + "tosa-ext-fft" + ], + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,256,64", + "--target-shape", + "1,4,4096" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type", + "inverse" + ] + } + } + }, + "floor": { + "group": "ew_unary", + "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ], + "generation": { + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "34,64", + "--target-rank", + "0", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,1,65533,2", + "--target-shape", + "1,65537,1,1" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "4,64", + "--target-rank", + "0", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ] + ] + } + }, + "selection": { + "default": { + "generator_select": "true" + } + } + }, + "fully_connected": { + "group": "tensor", + "support_for": [ + "lazy_data_gen" + ], + "generation": { + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "generator_args": [ + [ + "--target-dtype", + "int8" + ], + [ + "--target-dtype", + "int8", + "--target-shape", + "444,1", + "--target-shape", + "3,65538", + "--tensor-dim-range", + "1,2" + ], + [ + "--target-dtype", + "int8", + "--target-shape", + "3,16", + "--target-shape", + "1,23", + "--tensor-dim-range", + "100,200" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "negative_dim_range": "1,10", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "25,64" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "10,15", + "--target-shape", + "100,296", + "--target-shape", + "65540,2" + ], + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "35,64", + "--target-shape", + "30,16", + "--target-shape", + "100,23" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "no_negative_tests": "true", + "negative_dim_range": "1,10", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "25,64" + ], + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "35,64", + "--target-shape", + "30,16", + "--target-shape", + "100,23" + ] + ] + }, + "tosa-ext-int16": { + "supports_all": [ + "tosa-ext-int16" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int16" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + } + } + }, + "gather": { + "group": "scatter_gather", + "generation": { + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "generator_args": [ + [ + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "4,64", + "--max-batch-size", + "64" + ], + [ + "--target-dtype", + "int8", + "--tensor-dim-range", + "5,20", + "--target-shape", + "2,65533,1", + "--target-shape", + "1,2,65533", + "--target-shape", + "5000,1,1" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "negative_dim_range": "1,10", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "5,72", + "--max-batch-size", + "72" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "5,20", + "--target-shape", + "2,65536,1", + "--target-shape", + "1,2,65535", + "--target-shape", + "5000,1,1" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "no_negative_tests": "true", + "negative_dim_range": "1,10", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "5,72", + "--max-batch-size", + "72" + ] + ] + }, + "tosa-ext-fp8e4m3": { + "supports_all": [ + "tosa-ext-fp8e4m3" + ], + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e4m3", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-ext-fp8e5m2": { + "supports_all": [ + "tosa-ext-fp8e5m2" + ], + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e5m2", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + } + } + }, + "greater": { + "group": "comparison", + "support_for": [ + "lazy_data_gen" + ], + "generation": { + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "generator_args": [ + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "10,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,15", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "int32", + "--target-shape", + "1,2,65530,1", + "--target-shape", + "1,65539,1,2,1" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "18,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--target-shape", + "3,65551,1,1", + "--target-shape", + "1,1,3,65531,1" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "18,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,18", + "--target-rank", + "4", + "--target-rank", + "5" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + } + } + }, + "greater_equal": { + "group": "comparison", + "support_for": [ + "lazy_data_gen" + ], + "generation": { + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "selector": "true", + "generator_args": [ + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "20,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,20", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "int32", + "--target-shape", + "1,2,65533,1", + "--target-shape", + "1,65539,1,3,1" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "18,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,18", + "--target-rank", + "4", + "--target-rank", + "5" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "18,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--target-shape", + "65552,1,1,4", + "--target-shape", + "1,1,1,3,65532" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + } + } + }, + "identity": { + "group": "data_nodes", + "support_for": [ + "lazy_data_gen" + ], + "generation": { + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--target-dtype", + "int48", + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,61", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--target-dtype", + "int48", + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,15", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "int8", + "--target-shape", + "1,1,65536,1,1", + "--target-shape", + "1,65531,1,2" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,67", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,19", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,1,65539,1,1", + "--target-shape", + "1,1,2,65529" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,67", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,19", + "--target-rank", + "4", + "--target-rank", + "5" + ] + ] + }, + "tosa-ext-int16": { + "supports_all": [ + "tosa-ext-int16" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int16", + "--tensor-dim-range", + "1,61", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int16", + "--tensor-dim-range", + "1,15", + "--target-rank", + "4", + "--target-rank", + "5" + ] + ] + }, + "tosa-ext-int4": { + "supports_all": [ + "tosa-ext-int4" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int4", + "--tensor-dim-range", + "1,61", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int4", + "--tensor-dim-range", + "1,15", + "--target-rank", + "4", + "--target-rank", + "5" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + } + } + }, + "intdiv": { + "group": "ew_binary", + "generation": { + "tosa-bi-mi": { + "supports_any": [ + "tosa-bi", + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "int32", + "--target-shape", + "1,65533,1", + "--target-shape", + "1,1,65539,1,1" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + } + } + }, + "log": { + "group": "ew_unary", + "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ], + "generation": { + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "35,64", + "--target-rank", + "0", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,15", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,65530,1,2", + "--target-shape", + "3,1,65540,1,1" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "15,64", + "--target-rank", + "0", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ] + ] + } + }, + "selection": { + "default": { + "generator_select": "true" + } + } + }, + "logical_and": { + "group": "ew_binary", + "generation": { + "tosa-bi-mi": { + "supports_any": [ + "tosa-bi", + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "bool", + "--target-shape", + "1,65538,1", + "--target-shape", + "1,2,1,65536" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + } + } + }, + "logical_left_shift": { + "group": "ew_binary", + "generation": { + "tosa-bi-mi": { + "supports_any": [ + "tosa-bi", + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "int8", + "--target-shape", + "1,65532,1,1", + "--target-shape", + "1,2,1,65538" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + } + } + }, + "logical_not": { + "group": "ew_unary", + "generation": { + "tosa-bi-mi": { + "supports_any": [ + "tosa-bi", + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "bool", + "--tensor-dim-range", + "2,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,15", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "bool", + "--target-shape", + "1,65537,1", + "--target-shape", + "1,2,1,65535" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + } + } + }, + "logical_or": { + "group": "ew_binary", + "generation": { + "tosa-bi-mi": { + "supports_any": [ + "tosa-bi", + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "bool", + "--tensor-dim-range", + "3,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,14", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "bool", + "--target-shape", + "1,65536,1", + "--target-shape", + "1,2,1,65537" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + } + } + }, + "logical_right_shift": { + "group": "ew_binary", + "generation": { + "tosa-bi-mi": { + "supports_any": [ + "tosa-bi", + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "25,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,25", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "int8", + "--target-shape", + "1,1,65540,3", + "--target-shape", + "2,1,65530,1,1" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + } + } + }, + "logical_xor": { + "group": "ew_binary", + "generation": { + "tosa-bi-mi": { + "supports_any": [ + "tosa-bi", + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "bool", + "--tensor-dim-range", + "4,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "bool", + "--tensor-dim-range", + "3,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "bool", + "--target-shape", + "1,65539,1", + "--target-shape", + "1,2,1,65534" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + } + } + }, + "matmul": { + "group": "tensor", + "support_for": [ + "lazy_data_gen" + ], + "generation": { + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "selector": "tosa-bi", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--tensor-dim-range", + "16,64" + ], + [ + "--target-dtype", + "int8", + "--tensor-dim-range", + "1,16" + ], + [ + "--target-dtype", + "int8", + "--target-shape", + "1,65532,1", + "--target-shape", + "1,2,65538", + "--target-shape", + "2000,1,1" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "selector": "tosa-mi", + "negative_dim_range": "1,10", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-2.0,2.0", + "--tensor-dim-range", + "13,70" + ], + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-2.0,2.0", + "--tensor-dim-range", + "10,19" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-2.0,2.0", + "--target-shape", + "4,65530,1", + "--target-shape", + "1,65538,3" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "selector": "tosa-mi", + "no_negative_tests": "true", + "negative_dim_range": "1,10", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-2.0,2.0", + "--tensor-dim-range", + "13,70" + ], + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-2.0,2.0", + "--tensor-dim-range", + "10,19" + ] + ] + }, + "tosa-ext-fp8e4m3": { + "supports_all": [ + "tosa-ext-fp8e4m3" + ], + "selector": "tosa-mi", + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e4m3", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-ext-fp8e5m2": { + "supports_all": [ + "tosa-ext-fp8e5m2" + ], + "selector": "tosa-mi", + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e5m2", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-ext-int16": { + "supports_all": [ + "tosa-ext-int16" + ], + "selector": "tosa-bi", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int16", + "--tensor-dim-range", + "16,64" + ], + [ + "--target-dtype", + "int16", + "--tensor-dim-range", + "1,16" + ] + ] + } + }, + "selection": { + "tosa-bi": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + } + }, + "tosa-mi": { + "all": "true" + } + } + }, + "maximum": { + "group": "ew_binary", + "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ], + "generation": { + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "generator_args": [ + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "int32", + "--target-shape", + "1,2,65533,1", + "--target-shape", + "1,65539,1,1,1" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "selector": "tosa-mi", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "16,64", + "--target-rank", + "0", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--target-shape", + "65531,1,2,1", + "--target-shape", + "1,1,65541,1" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "selector": "tosa-mi", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "16,64", + "--target-rank", + "0", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "tosa-mi": { + "generator_select": "true" + } + } + }, + "max_pool2d": { + "group": "tensor", + "support_for": [ + "lazy_data_gen" + ], + "generation": { + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "selector": "tosa-bi", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-shape", + "1,4,75,3", + "--target-shape", + "1,11,44,1", + "--allow-pooling-and-conv-oversizes" + ], + [ + "--target-dtype", + "int8", + "--target-shape", + "1,65535,5,1", + "--target-shape", + "1,3,65537,1", + "--target-shape", + "1,3,2,1", + "--allow-pooling-and-conv-oversizes" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "selector": "8k_level", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--tensor-dim-range", + "3,10", + "--level-8k-sizes" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "selector": "tosa-mi", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,47,37,25", + "--target-shape", + "1,15,13,38", + "--allow-pooling-and-conv-oversizes" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,65534,4,1", + "--target-shape", + "1,6,65536,1", + "--max-conv-dilation", + "1", + "--allow-pooling-and-conv-oversizes" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "selector": "tosa-mi", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,47,37,25", + "--target-shape", + "1,15,13,38", + "--allow-pooling-and-conv-oversizes" + ] + ] + }, + "tosa-ext-fp8e4m3": { + "supports_all": [ + "tosa-ext-fp8e4m3" + ], + "selector": "tosa-mi", + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e4m3", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-ext-fp8e5m2": { + "supports_all": [ + "tosa-ext-fp8e5m2" + ], + "selector": "tosa-mi", + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e5m2", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-ext-int16": { + "supports_all": [ + "tosa-ext-int16" + ], + "selector": "tosa-bi", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int16", + "--target-shape", + "1,4,75,3", + "--target-shape", + "1,11,44,1", + "--allow-pooling-and-conv-oversizes" + ] + ] + }, + "8k_level_ext_int16": { + "supports_all": [ + "tosa-ext-int16" + ], + "no_negative_tests": "true", + "selector": "8k_level", + "generator_args": [ + [ + "--target-dtype", + "int16", + "--tensor-dim-range", + "3,10", + "--level-8k-sizes" + ] + ] + } + }, + "selection": { + "tosa-bi": { + "params": { + "shape": [], + "type": [ + "i8", + "i16" + ], + "kernel": [], + "stride": [], + "pad": [] + }, + "permutes": [ + "shape", + "type", + "pad" + ] + }, + "8k_level": { + "all": "true" + }, + "tosa-mi": { + "params": { + "shape": [], + "type": [ + "f32", + "f16", + "bf16" + ], + "stride": [], + "pad": [] + }, + "permutes": [ + "shape", + "type", + "pad" + ], + "sparsity": { + "pad": 11 + } + } + } + }, + "minimum": { + "group": "ew_binary", + "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ], + "generation": { + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "generator_args": [ + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "int32", + "--target-shape", + "1,2,65533,1", + "--target-shape", + "1,65539,1,1,1" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "selector": "tosa-mi", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "16,64", + "--target-rank", + "0", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,65531,2,1", + "--target-shape", + "1,1,1,65538" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "selector": "tosa-mi", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "16,64", + "--target-rank", + "0", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "tosa-mi": { + "generator_select": "true" + } + } + }, + "mul": { + "group": "ew_binary", + "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ], + "generation": { + "tosa-bi-mi": { + "supports_any": [ + "tosa-bi", + "tosa-mi" + ], + "selector": "tosa-bi", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ] + ] + }, + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "selector": "tosa-bi", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "int8", + "--target-shape", + "1,2,65500,1", + "--target-shape", + "1,68539,1,1,1" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "selector": "tosa-bi", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6" + ] + ] + }, + "8k_level_bi_mi": { + "supports_any": [ + "tosa-bi", + "tosa-mi" + ], + "selector": "tosa-bi", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "selector": "tosa-mi", + "negative_dim_range": "1,10", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "16,64", + "--target-rank", + "0", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "selector": "tosa-mi", + "no_negative_tests": "true", + "negative_dim_range": "1,10", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "16,64", + "--target-rank", + "0", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,1,3,65534", + "--target-shape", + "1,2,1,65541,1" + ] + ] + } + }, + "selection": { + "tosa-bi": { + "params": {}, + "permutes": [ + "shape", + "type", + "perm", + "shift" + ] + }, + "tosa-mi": { + "generator_select": "true" + } + } + }, + "mul_shape": { + "group": "shape", + "generation": { + "standard": { + "supports_any": [ + "tosa-bi", + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "shape", + "--tensor-dim-range", + "1,16", + "--target-rank", + "1" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape" + ] + } + } + }, + "negate": { + "group": "ew_unary", + "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ], + "generation": { + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "32,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,22", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "int8", + "--target-shape", + "3,1,65540,1", + "--target-shape", + "1,1,2,1,65531" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "selector": "tosa-mi", + "negative_dim_range": "1,10", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "17,64", + "--target-rank", + "0", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,22", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,65541,3,1", + "--target-shape", + "1,2,1,65532,1" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "selector": "tosa-mi", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "17,64", + "--target-rank", + "0", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,22", + "--target-rank", + "4", + "--target-rank", + "5" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "tosa-mi": { + "generator_select": "true" + } + } + }, + "pad": { + "group": "data_layout", + "support_for": [ + "lazy_data_gen", + "stable_random_gen" + ], + "generation": { + "tosa-bi-mi": { + "supports_any": [ + "tosa-bi", + "tosa-mi" + ], + "selector": "tosa-bi", + "generator_args": [ + [ + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4" + ] + ] + }, + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "selector": "tosa-bi", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4" + ], + [ + "--target-dtype", + "int16", + "--tensor-dim-range", + "1,16", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "int8", + "--target-shape", + "1,1,65537,1" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "selector": "8k_level", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "selector": "tosa-mi", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,65", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,11", + "--target-rank", + "4", + "--target-rank", + "5", + "--target-rank", + "6" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "selector": "tosa-mi", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,65", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,11", + "--target-rank", + "4", + "--target-rank", + "5", + "--target-rank", + "6" + ] + ] + }, + "tosa-ext-fp8e4m3": { + "supports_all": [ + "tosa-ext-fp8e4m3" + ], + "selector": "tosa-mi", + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e4m3", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-ext-fp8e5m2": { + "supports_all": [ + "tosa-ext-fp8e5m2" + ], + "selector": "tosa-mi", + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e5m2", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ] + ] + }, + "extra": { + "supports_all": [ + "tosa-mi" + ], + "selector": "minimal", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,1,65539,1" + ], + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--target-rank", + "4", + "--oversize" + ] + ] + }, + "extra-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "selector": "minimal", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,1,65539,1" + ], + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--target-rank", + "4", + "--oversize" + ] + ] + } + }, + "selection": { + "tosa-bi": { + "params": {}, + "permutes": [ + "shape", + "type" + ], + "sparsity": { + "pad": 15 + }, + "preselected": [ + { + "shape": "6", + "type": "i8", + "pad": "pad00" + } + ] + }, + "8k_level": { + "params": {}, + "permutes": [], + "sparsity": { + "pad": 3 + }, + "full_sparsity": "true" + }, + "tosa-mi": { + "generator_select": "true", + "maximum": 5, + "specifics": { + "pad": [ + "[[0 0]]", + "[[0 0], [0 0]]", + "[[0 0], [0 0], [0 0]]", + "[[0 0], [0 0], [0 0], [0 0]]", + "[[0 0], [0 0], [0 0], [0 0], [0 0]]", + "[[0 0], [0 0], [0 0], [0 0], [0 0], [0 0]]" + ] + } + }, + "minimal": { + "generator_select": "true", + "maximum": 2 + } + } + }, + "pow": { + "group": "ew_binary", + "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ], + "generation": { + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "16,64", + "--target-rank", + "0", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "16,64", + "--target-rank", + "0", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--target-shape", + "65534,3,1,1", + "--target-shape", + "2,65541,1,1,1" + ] + ] + } + }, + "selection": { + "default": { + "generator_select": "true" + } + } + }, + "reciprocal": { + "group": "ew_unary", + "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ], + "generation": { + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "28,64", + "--target-rank", + "0", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,15", + "--target-rank", + "4", + "--target-rank", + "5" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "15,64", + "--target-rank", + "0", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,1,65536,4", + "--target-shape", + "2,65535,1,1,1" + ] + ] + } + }, + "selection": { + "default": { + "generator_select": "true" + } + } + }, + "reduce_all": { + "group": "reduction", + "generation": { + "tosa-bi-mi": { + "supports_any": [ + "tosa-bi", + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,32", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--target-rank", + "4" + ], + [ + "--target-dtype", + "bool", + "--target-shape", + "65530,1,1,1", + "--target-shape", + "2,65538,1,1" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "from_version": "v0.80.0", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,10", + "--target-rank", + "5", + "--target-rank", + "6" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type", + "axis" + ] + } + } + }, + "reduce_any": { + "group": "reduction", + "generation": { + "tosa-bi-mi": { + "supports_any": [ + "tosa-bi", + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,32", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--target-rank", + "4" + ], + [ + "--target-dtype", + "bool", + "--target-shape", + "65530,1,1,1", + "--target-shape", + "2,65538,1,1" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "from_version": "v0.80.0", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,9", + "--target-rank", + "5", + "--target-rank", + "6" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type", + "axis" + ] + } + } + }, + "reduce_max": { + "group": "reduction", + "support_for": [ + "lazy_data_gen" + ], + "generation": { + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,32", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--target-rank", + "4" + ], + [ + "--target-dtype", + "int8", + "--target-shape", + "65530,1,1,1", + "--target-shape", + "2,65538,1,1" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "from_version": "v0.80.0", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,11", + "--target-rank", + "5", + "--target-rank", + "6" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "negative_dim_range": "1,10", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,32", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--target-rank", + "4" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,1,1,65531", + "--target-shape", + "1,1,65539,2" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "no_negative_tests": "true", + "negative_dim_range": "1,10", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,32", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--target-rank", + "4" + ] + ] + }, + "8k_level_mi": { + "supports_all": [ + "tosa-mi" + ], + "from_version": "v0.80.0", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,10", + "--target-rank", + "5", + "--target-rank", + "6" + ] + ] + }, + "8k_level_ext_bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "from_version": "v0.80.0", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,10", + "--target-rank", + "5", + "--target-rank", + "6" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type", + "axis" + ] + } + } + }, + "reduce_min": { + "group": "reduction", + "support_for": [ + "lazy_data_gen" + ], + "generation": { + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,24", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--target-rank", + "4" + ], + [ + "--target-dtype", + "int8", + "--target-shape", + "1,1,65538,1", + "--target-shape", + "2,65535,1,1" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "from_version": "v0.80.0", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "2,10", + "--target-rank", + "5", + "--target-rank", + "6" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "negative_dim_range": "1,10", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,32", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--target-rank", + "4" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,1,65531,1", + "--target-shape", + "1,65539,2,1" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "no_negative_tests": "true", + "negative_dim_range": "1,10", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,32", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--target-rank", + "4" + ] + ] + }, + "8k_level_mi": { + "supports_all": [ + "tosa-mi" + ], + "from_version": "v0.80.0", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,9", + "--target-rank", + "5", + "--target-rank", + "6" + ] + ] + }, + "8k_level_ext_bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "from_version": "v0.80.0", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,9", + "--target-rank", + "5", + "--target-rank", + "6" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type", + "axis" + ] + } + } + }, + "reduce_product": { + "group": "reduction", + "support_for": [ + "lazy_data_gen" + ], + "generation": { + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,34", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--target-rank", + "4" + ], + [ + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--target-shape", + "2,65527,3,1", + "--target-shape", + "1,3,2,65541" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,34", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--target-rank", + "4" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-mi" + ], + "from_version": "v0.80.0", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,11", + "--target-rank", + "5", + "--target-rank", + "6" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type", + "axis" + ] + } + } + }, + "reduce_sum": { + "group": "reduction", + "support_for": [ + "lazy_data_gen" + ], + "generation": { + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "generator_args": [ + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,32", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--target-rank", + "4" + ], + [ + "--target-dtype", + "int32", + "--target-shape", + "3,65532,1,1", + "--target-shape", + "65536,1,1,2" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "from_version": "v0.80.0", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,12", + "--target-rank", + "5", + "--target-rank", + "6" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--target-rank", + "3", + "--target-rank", + "4", + "--tensor-dim-range", + "10,37", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--target-rank", + "4" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--target-shape", + "1001", + "--target-shape", + "50,200" + ], + [ + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--target-shape", + "65529", + "--target-shape", + "65537,2" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--target-rank", + "3", + "--target-rank", + "4", + "--tensor-dim-range", + "10,37", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--target-rank", + "4" + ], + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--target-shape", + "16001", + "--target-shape", + "16,1000", + "--target-shape", + "1,3,65529,1", + "--target-shape", + "1,65537,1,1" + ] + ] + }, + "8k_level_mi": { + "supports_all": [ + "tosa-mi" + ], + "from_version": "v0.80.0", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "5,12", + "--target-rank", + "5", + "--target-rank", + "6" + ] + ] + }, + "8k_level_ext_bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "from_version": "v0.80.0", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "5,12", + "--target-rank", + "5", + "--target-rank", + "6" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type", + "axis" + ] + } + } + }, + "rescale": { + "group": "type_conversion", + "generation": { + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--target-dtype", + "uint8", + "--target-dtype", + "uint16", + "--tensor-dim-range", + "16,32", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--target-dtype", + "uint8", + "--target-dtype", + "uint16", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "int8", + "--target-shape", + "1,2,65533,1", + "--target-shape", + "1,65539,1,1" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--target-dtype", + "uint8", + "--target-dtype", + "uint16", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6" + ] + ] + }, + "tosa-ext-int16": { + "supports_all": [ + "tosa-ext-int16" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int48", + "--tensor-dim-range", + "16,32", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int48", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ] + ] + }, + "8k_level_ext_int16": { + "supports_all": [ + "tosa-ext-int16" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int48", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6" + ] + ] + } + }, + "selection": { + "default": { + "params": { + "scale": [ + "sc0", + "sc0", + "sc1", + "sc1", + "sc1", + "sc1" + ], + "double_round": [ + "dr0", + "dr0", + "dr0", + "dr0", + "dr1", + "dr1" + ], + "per_channel": [ + "pc0", + "pc1", + "pc0", + "pc1", + "pc0", + "pc1", + "pc1" + ] + }, + "permutes": [ + "shape", + "type", + "output_type" + ] + } + } + }, + "reshape": { + "group": "data_layout", + "support_for": [ + "lazy_data_gen" + ], + "generation": { + "tosa-bi-mi": { + "supports_any": [ + "tosa-bi", + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "bool", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--target-rank", + "4", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--target-dtype", + "bool", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--target-rank", + "4", + "--num-rand-permutations", + "2" + ], + [ + "--target-dtype", + "int8", + "--target-shape", + "1,1,65535,1", + "--target-shape", + "1,65538,1,2", + "--num-rand-permutations", + "2" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,10", + "--target-rank", + "5", + "--target-rank", + "6", + "--num-rand-permutations", + "2" + ] + ] + }, + "8k_level-bi-mi": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,10", + "--target-rank", + "5", + "--target-rank", + "6", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--target-rank", + "4", + "--num-rand-permutations", + "2" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,1,65536,1", + "--target-shape", + "1,65535,1,2", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--target-rank", + "4", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-ext-fp8e4m3": { + "supports_all": [ + "tosa-ext-fp8e4m3" + ], + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e4m3", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-ext-fp8e5m2": { + "supports_all": [ + "tosa-ext-fp8e5m2" + ], + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e5m2", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type", + "rank", + "perm" + ] + } + } + }, + "resize": { + "group": "image", + "support_for": [ + "lazy_data_gen" + ], + "generation": { + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "selector": "tosa-bi", + "generator_args": [ + [ + "--target-dtype", + "int8" + ], + [ + "--target-dtype", + "int8", + "--target-shape", + "1,3,16383,1", + "--max-resize-output-dim", + "9500" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "selector": "tosa-bi", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--level-8k-sizes", + "--tensor-dim-range", + "2,10", + "--num-rand-permutations", + "2" + ] + ] + }, + "8k_level_ext_int16": { + "supports_all": [ + "tosa-ext-int16" + ], + "selector": "tosa-bi", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int16", + "--level-8k-sizes", + "--tensor-dim-range", + "2,10", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "selector": "tosa-mi", + "negative_dim_range": "1,10", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,1103,1,2", + "--max-resize-output-dim", + "9100" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "selector": "tosa-mi", + "no_negative_tests": "true", + "negative_dim_range": "1,10", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max" + ] + ] + }, + "tosa-ext-int16": { + "supports_all": [ + "tosa-ext-int16" + ], + "selector": "tosa-bi", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int16" + ] + ] + } + }, + "selection": { + "tosa-bi": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "tosa-mi": { + "params": { + "offset": [], + "shape": [], + "type": [], + "mode": [], + "output_type": [], + "scale": [ + "sc2x1x3x1" + ], + "border": [ + "bor0x1" + ] + }, + "permutes": [ + "shape", + "type" + ], + "exclude_patterns": [ + ".*_ERRORIF_MaxDimExceeded_1x16584x5x1_.*" + ] + } + } + }, + "reverse": { + "group": "data_layout", + "support_for": [ + "lazy_data_gen" + ], + "generation": { + "tosa-bi-mi": { + "supports_any": [ + "tosa-bi", + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,32", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--target-rank", + "4" + ] + ] + }, + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,32", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--target-rank", + "4" + ], + [ + "--target-dtype", + "int8", + "--target-shape", + "1,1,65535", + "--target-shape", + "1,65540,1,2" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,12", + "--target-rank", + "5", + "--target-rank", + "6" + ] + ] + }, + "8k_level_bi_mi": { + "supports_any": [ + "tosa-bi", + "tosa-mi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,12", + "--target-rank", + "5", + "--target-rank", + "6" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,48", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--target-rank", + "4" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,1,65537", + "--target-shape", + "1,65535,1,2" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,48", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--target-rank", + "4" + ] + ] + }, + "tosa-ext-fp8e4m3": { + "supports_all": [ + "tosa-ext-fp8e4m3" + ], + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e4m3", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-ext-fp8e5m2": { + "supports_all": [ + "tosa-ext-fp8e5m2" + ], + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e5m2", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + } + } + }, + "rfft2d": { + "group": "tensor", + "support_for": [ + "lazy_data_gen" + ], + "generation": { + "standard": { + "supports_all": [ + "tosa-ext-fft" + ], + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,16,512", + "--target-shape", + "1,8192,2" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + } + } + }, + "rsqrt": { + "group": "ew_unary", + "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ], + "generation": { + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "29,64", + "--target-rank", + "0", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,1,3,65541", + "--target-shape", + "65532,1,3,1,1" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "15,64", + "--target-rank", + "0", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,15", + "--target-rank", + "4", + "--target-rank", + "5" + ] + ] + } + }, + "selection": { + "default": { + "generator_select": "true" + } + } + }, + "scatter": { + "group": "scatter_gather", + "support_for": [ + "lazy_data_gen" + ], + "generation": { + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "generator_args": [ + [ + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "4,64", + "--max-batch-size", + "64" + ], + [ + "--target-dtype", + "int8", + "--target-shape", + "1,65540,1", + "--target-shape", + "1,1,65533", + "--target-shape", + "3000,7,1" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "negative_dim_range": "1,10", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "5,56", + "--max-batch-size", + "56" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,65541,1", + "--target-shape", + "1,1,65531", + "--target-shape", + "2989,6,1" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "no_negative_tests": "true", + "negative_dim_range": "1,10", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "5,56", + "--max-batch-size", + "56" + ] + ] + }, + "tosa-ext-fp8e4m3": { + "supports_all": [ + "tosa-ext-fp8e4m3" + ], + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e4m3", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-ext-fp8e5m2": { + "supports_all": [ + "tosa-ext-fp8e5m2" + ], + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e5m2", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--num-rand-permutations", + "2" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + } + } + }, + "select": { + "group": "ew_ternary", + "support_for": [ + "lazy_data_gen" + ], + "generation": { + "tosa-bi-mi": { + "supports_any": [ + "tosa-bi", + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "bool", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ] + ] + }, + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "int8", + "--target-shape", + "1,1,65535,1", + "--target-shape", + "1,65538,1,2", + "--num-rand-permutations", + "2" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6" + ] + ] + }, + "8k_level-bi-mi": { + "supports_any": [ + "tosa-bi", + "tosa-mi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,2,65534,2,1", + "--target-shape", + "1,1,65539,2", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + } + } + }, + "sigmoid": { + "group": "activation", + "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ], + "generation": { + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-16.0,16.0", + "--tensor-dim-range", + "16,64", + "--target-rank", + "0", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-16.0,16.0", + "--tensor-dim-range", + "1,19", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-16.0,16.0", + "--target-shape", + "1,2,65538,2,1", + "--target-shape", + "1,3,1,65533" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-16.0,16.0", + "--tensor-dim-range", + "16,64", + "--target-rank", + "0", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-16.0,16.0", + "--tensor-dim-range", + "1,19", + "--target-rank", + "4", + "--target-rank", + "5" + ] + ] + } + }, + "selection": { + "default": { + "generator_select": "true" + } + } + }, + "sin": { + "group": "ew_unary", + "support_for": [ + "lazy_data_gen" + ], + "generation": { + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "15,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,15", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--target-shape", + "3,1,65534,2", + "--target-shape", + "65533,1,3,2,1" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "15,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + } + } + }, + "slice": { + "group": "data_layout", + "support_for": [ + "lazy_data_gen" + ], + "generation": { + "tosa-bi-mi": { + "supports_any": [ + "tosa-bi", + "tosa-mi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bool", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ], + [ + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--num-rand-permutations", + "2" + ] + ] + }, + "8k_level_bi_mi": { + "supports_any": [ + "tosa-bi", + "tosa-mi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,10", + "--target-rank", + "5", + "--target-rank", + "6", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ], + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--num-rand-permutations", + "2" + ], + [ + "--target-dtype", + "int8", + "--target-shape", + "1,1,65535,1", + "--target-shape", + "1,65538,1,2", + "--num-rand-permutations", + "2" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,10", + "--target-rank", + "5", + "--target-rank", + "6", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ], + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ], + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--num-rand-permutations", + "2" + ], + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-2.0,2.0", + "--target-shape", + "1,1,655341,1", + "--target-shape", + "1,65539,1,1", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-ext-fp8e4m3": { + "supports_all": [ + "tosa-ext-fp8e4m3" + ], + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e4m3", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-ext-fp8e5m2": { + "supports_all": [ + "tosa-ext-fp8e5m2" + ], + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e5m2", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type", + "perm" + ] + } + } + }, + "sub": { + "group": "ew_binary", + "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ], + "generation": { + "tosa-bi-mi": { + "supports_any": [ + "tosa-bi", + "tosa-mi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "int32", + "--target-shape", + "1,2,65533,1", + "--target-shape", + "1,65539,1,1,1" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "selector": "tosa-mi", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "16,64", + "--target-rank", + "0", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,1,65533,3", + "--target-shape", + "1,1,2,1,65545" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "selector": "tosa-mi", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "16,64", + "--target-rank", + "0", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + }, + "tosa-mi": { + "generator_select": "true" + } + } + }, + "sub_shape": { + "group": "shape", + "generation": { + "standard": { + "supports_any": [ + "tosa-bi", + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "shape", + "--tensor-dim-range", + "1,16", + "--target-rank", + "1" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape" + ] + } + } + }, + "table": { + "group": "ew_binary", + "generation": { + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "generator_args": [ + [ + "--target-dtype", + "int8", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int8", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "int8", + "--target-shape", + "1,2,65536,1", + "--target-shape", + "1,65537,1,1,3" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6" + ] + ] + }, + "8k_level_ext_int16": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int16", + "--tensor-dim-range", + "1,10", + "--target-rank", + "6" + ] + ] + }, + "tosa-ext-int16": { + "supports_all": [ + "tosa-ext-int16" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int16", + "--tensor-dim-range", + "16,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "int16", + "--tensor-dim-range", + "1,16", + "--target-rank", + "4", + "--target-rank", + "5" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + } + } + }, + "tanh": { + "group": "activation", + "support_for": [ "lazy_data_gen", "stable_random_gen", "random_const_inputs" ], + "generation": { + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-8.0,8.0", + "--tensor-dim-range", + "18,60", + "--target-rank", + "0", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-8.0,8.0", + "--tensor-dim-range", + "1,24", + "--target-rank", + "4", + "--target-rank", + "5" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-8.0,8.0", + "--target-shape", + "1,65535,2,1,1", + "--target-shape", + "1,65540,1,2" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-8.0,8.0", + "--tensor-dim-range", + "18,60", + "--target-rank", + "0", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3" + ], + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-8.0,8.0", + "--tensor-dim-range", + "1,24", + "--target-rank", + "4", + "--target-rank", + "5" + ] + ] + } + }, + "selection": { + "default": { + "generator_select": "true" + } + } + }, + "tile": { + "group": "data_layout", + "support_for": [ + "lazy_data_gen" + ], + "generation": { + "tosa-bi-mi": { + "supports_any": [ + "tosa-bi", + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,16", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--target-rank", + "4" + ] + ] + }, + "8k_level_bi_mi": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bool", + "--tensor-dim-range", + "1,10", + "--target-rank", + "5", + "--target-rank", + "6" + ] + ] + }, + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "4,32", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--target-rank", + "4" + ], + [ + "--target-dtype", + "int8", + "--target-shape", + "65533", + "--target-shape", + "1,2,65540,1" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,10", + "--target-rank", + "5", + "--target-rank", + "6" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "negative_dim_range": "1,10", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "4,32", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--target-rank", + "4" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--target-shape", + "65535", + "--target-shape", + "1,65543,2,1" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "no_negative_tests": "true", + "negative_dim_range": "1,10", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "4,32", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--target-rank", + "4" + ] + ] + }, + "tosa-ext-fp8e4m3": { + "supports_all": [ + "tosa-ext-fp8e4m3" + ], + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e4m3", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-ext-fp8e5m2": { + "supports_all": [ + "tosa-ext-fp8e5m2" + ], + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e5m2", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + } + } + }, + "transpose": { + "group": "data_layout", + "support_for": [ + "lazy_data_gen" + ], + "generation": { + "tosa-bi-mi": { + "supports_any": [ + "tosa-bi", + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "bool", + "--target-rank", + "3", + "--target-rank", + "4", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-rank", + "1", + "--target-rank", + "2", + "--num-rand-permutations", + "2" + ], + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--target-rank", + "3", + "--target-rank", + "4", + "--num-rand-permutations", + "2" + ], + [ + "--target-dtype", + "int8", + "--target-shape", + "1,65537,1,2", + "--target-shape", + "65535,1,1,1", + "--num-rand-permutations", + "1" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-dtype", + "int16", + "--target-dtype", + "int32", + "--tensor-dim-range", + "1,10", + "--target-rank", + "5", + "--target-rank", + "6", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--target-rank", + "1", + "--target-rank", + "2", + "--num-rand-permutations", + "2" + ], + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--target-rank", + "3", + "--target-rank", + "4", + "--num-rand-permutations", + "2" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,1,65539,2", + "--target-shape", + "1,1,1,65535", + "--num-rand-permutations", + "1" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--target-rank", + "3", + "--target-rank", + "4", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-ext-fp8e4m3": { + "supports_all": [ + "tosa-ext-fp8e4m3" + ], + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e4m3", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-ext-fp8e5m2": { + "supports_all": [ + "tosa-ext-fp8e4m3" + ], + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e5m2", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--target-rank", + "1", + "--target-rank", + "2", + "--target-rank", + "3", + "--num-rand-permutations", + "2" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type", + "perm" + ] + } + } + }, + "transpose_conv2d": { + "group": "tensor", + "support_for": [ + "lazy_data_gen" + ], + "gen_filter": "^transpose_conv2d", + "generation": { + "tosa-bi": { + "supports_all": [ + "tosa-bi" + ], + "selector": "tosa-bi", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--target-shape", + "1,49,33,1", + "--target-shape", + "1,11,33,3", + "--tensor-dim-range", + "1,18", + "--allow-pooling-and-conv-oversizes" + ], + [ + "--target-dtype", + "int8", + "--target-shape", + "1,65536,1,1", + "--tensor-dim-range", + "1,12", + "--max-conv-dilation", + "1" + ] + ] + }, + "8k_level": { + "supports_all": [ + "tosa-bi" + ], + "no_negative_tests": "true", + "selector": "8k_level", + "generator_args": [ + [ + "--target-dtype", + "int8", + "--tensor-dim-range", + "2,5", + "--level-8k-sizes" + ] + ] + }, + "8k_level_ext_int16": { + "supports_all": [ + "tosa-ext-int16" + ], + "no_negative_tests": "true", + "selector": "8k_level", + "generator_args": [ + [ + "--target-dtype", + "int16", + "--tensor-dim-range", + "2,5", + "--level-8k-sizes" + ] + ] + }, + "tosa-mi": { + "supports_all": [ + "tosa-mi" + ], + "selector": "tosa-mi", + "negative_dim_range": "1,10", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--fp-values-range", + "-max,max", + "--target-shape", + "10,24,9,13", + "--target-shape", + "8,14,20,5", + "--tensor-dim-range", + "1,16", + "--allow-pooling-and-conv-oversizes" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-max,max", + "--target-shape", + "1,65536,1,2", + "--tensor-dim-range", + "1,17", + "--max-conv-dilation", + "1" + ] + ] + }, + "tosa-ext-bf16": { + "supports_all": [ + "tosa-ext-bf16" + ], + "selector": "tosa-mi", + "no_negative_tests": "true", + "negative_dim_range": "1,10", + "generator_args": [ + [ + "--target-dtype", + "bf16", + "--fp-values-range", + "-max,max", + "--target-shape", + "10,24,9,13", + "--target-shape", + "8,14,20,5", + "--tensor-dim-range", + "1,16", + "--allow-pooling-and-conv-oversizes" + ] + ] + }, + "tosa-ext-fp8e4m3": { + "supports_all": [ + "tosa-ext-fp8e4m3" + ], + "selector": "tosa-mi", + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e4m3", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--target-shape", + "10,24,9,13", + "--target-shape", + "8,14,20,5", + "--tensor-dim-range", + "1,16", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-ext-fp8e5m2": { + "supports_all": [ + "tosa-ext-fp8e5m2" + ], + "selector": "tosa-mi", + "no_negative_tests": "true", + "from_version": "v0.100.0", + "generator_args": [ + [ + "--target-dtype", + "fp8e5m2", + "--fp-values-range", + "-max,max", + "--tensor-dim-range", + "32,64", + "--target-shape", + "10,24,9,13", + "--target-shape", + "8,14,20,5", + "--tensor-dim-range", + "1,16", + "--num-rand-permutations", + "2" + ] + ] + }, + "tosa-ext-int16": { + "supports_all": [ + "tosa-ext-int16" + ], + "selector": "tosa-bi", + "no_negative_tests": "true", + "generator_args": [ + [ + "--target-dtype", + "int16", + "--target-shape", + "1,49,33,1", + "--target-shape", + "1,11,33,3", + "--tensor-dim-range", + "1,18", + "--allow-pooling-and-conv-oversizes" + ] + ] + } + }, + "selection": { + "tosa-bi": { + "params": { + "shape": [], + "type": [ + "i8xi8", + "i16xi8", + "i8xi4" + ], + "kernel": [], + "stride": [], + "pad": [], + "out_shape": [] + }, + "permutes": [ + "shape", + "type", + "stride", + "pad" + ] + }, + "8k_level": { + "all": "true" + }, + "tosa-mi": { + "params": { + "shape": [], + "type": [], + "kernel": [], + "stride": [], + "pad": [], + "out_shape": [] + }, + "permutes": [ + "shape", + "type", + "stride", + "pad" + ] + } + } + }, + "while_loop": { + "group": "control_flow", + "generation": { + "standard": { + "supports_any": [ + "tosa-bi", + "tosa-mi" + ], + "generator_args": [ + [ + "--target-dtype", + "int32" + ], + [ + "--target-dtype", + "int32", + "--target-shape", + "3,22,17,1,2", + "--target-shape", + "1,65537,2,1" + ] + ] + } + }, + "selection": { + "default": { + "params": {}, + "permutes": [ + "shape", + "type" + ] + } + } + } +}
\ No newline at end of file |