blob: d5d08ba0631ea34a128bf70c23e999bbfbddc43a [file] [log] [blame]
{
"abs": {
"group": "ew_unary",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--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",
"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": {
"default": {
"params": {},
"permutes": [
"shape",
"type"
]
}
}
},
"add": {
"group": "ew_binary",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--fp-values-range",
"-max,max",
"--tensor-dim-range",
"16,64",
"--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"
]
]
}
},
"selection": {
"default": {
"params": {},
"permutes": [
"shape",
"type"
]
}
}
},
"argmax": {
"group": "tensor",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--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",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--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"
]
]
},
"float8": {
"from_version" : "v0.100.0",
"no_negative_tests": "true",
"generator_args": [
[
"--target-dtype",
"fp8e4m3",
"--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"
]
}
}
},
"avg_pool2d": {
"group": "tensor",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--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"
],
[
"--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"
]
]
},
"float8": {
"from_version" : "v0.100.0",
"no_negative_tests": "true",
"generator_args": [
[
"--target-dtype",
"fp8e4m3",
"--target-dtype",
"fp8e5m2",
"--fp-values-range",
"-max,max",
"--tensor-dim-range",
"32,64",
"--num-rand-permutations",
"2"
]
]
}
},
"selection": {
"default": {
"params": {},
"permutes": [
"shape",
"type"
],
"sparsity": {
"pad": 17
}
}
}
},
"cast": {
"group": "type_conversion",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"negative_dim_range": "1,10",
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--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",
"fp16",
"--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"
],
[
"--target-dtype",
"fp16",
"--fp-values-range",
"-max,max",
"--target-shape",
"1,1,1,65533,1",
"--target-shape",
"2,65538,1,1"
]
]
},
"float8": {
"from_version" : "v0.100.0",
"no_negative_tests": "true",
"generator_args": [
[
"--target-dtype",
"fp8e4m3",
"--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": {
"shape": [],
"type": [],
"output_type": []
},
"permutes": [
"shape",
"type",
"output_type"
],
"exclude_patterns": [
"^((?!ERRORIF).)*_(i8|i16|i32|b)_out(i8|i16|i32|b)"
]
}
}
},
"ceil": {
"group": "ew_unary",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--fp-values-range",
"-max,max",
"--tensor-dim-range",
"4,64",
"--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",
"fp16",
"--fp-values-range",
"-max,max",
"--target-shape",
"2,1,65530,1",
"--target-shape",
"1,1,65539,1"
]
]
}
},
"selection": {
"default": {
"params": {},
"permutes": [
"shape",
"type"
]
}
}
},
"clamp": {
"group": "activation",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--fp-values-range",
"-max,max",
"--tensor-dim-range",
"20,64",
"--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,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"
]
]
}
},
"selection": {
"default": {
"params": {},
"permutes": [
"shape",
"type"
]
}
}
},
"concat": {
"group": "data_layout",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--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",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--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"
]
]
},
"float8": {
"from_version" : "v0.100.0",
"no_negative_tests": "true",
"generator_args": [
[
"--target-dtype",
"fp8e4m3",
"--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"
]
}
}
},
"const": {
"group": "data_nodes",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"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"
]
]
},
"float8": {
"from_version" : "v0.100.0",
"no_negative_tests": "true",
"generator_args": [
[
"--target-dtype",
"fp8e4m3",
"--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"
]
}
}
},
"conv2d": {
"group": "tensor",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen", "generator_select", "stable_random_gen" ],
"gen_filter": "^conv2d",
"generation": {
"standard": {
"negative_dim_range": "1,10",
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--fp-values-range",
"-max,max",
"--target-shape",
"2,34,19,27",
"--target-shape",
"3,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",
"--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"
]
]
},
"float8": {
"from_version" : "v0.100.0",
"no_negative_tests": "true",
"generator_args": [
[
"--target-dtype",
"fp8e4m3",
"--target-dtype",
"fp8e5m2",
"--fp-values-range",
"-max,max",
"--tensor-dim-range",
"32,64",
"--num-rand-permutations",
"2"
]
]
}
},
"selection": {
"default": {
"permutes": [ "rank", "dtype", "kernel", "acc_type" ],
"full_params": [ "stride", "dilation" ],
"specifics": { "pad": [ "(0, 0, 0, 0)" ] },
"groups": [ "s" ],
"maximum": 3
}
}
},
"conv3d": {
"group": "tensor",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"gen_filter": "^conv3d",
"generation": {
"standard": {
"negative_dim_range": "1,10",
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--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"
],
[
"--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"
]
]
},
"float8": {
"from_version" : "v0.100.0",
"no_negative_tests": "true",
"generator_args": [
[
"--target-dtype",
"fp8e4m3",
"--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"
]
]
}
},
"selection": {
"default": {
"params": {
"shape": [],
"type": [],
"kernel": [],
"stride": [],
"pad": [],
"dilation": []
},
"permutes": [
"kernel",
"shape",
"type",
"pad",
"stride"
]
}
}
},
"depthwise_conv2d": {
"group": "tensor",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"gen_filter": "^depthwise_conv2d",
"generation": {
"standard": {
"negative_dim_range": "1,10",
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--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"
],
[
"--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"
]
]
},
"float8": {
"from_version" : "v0.100.0",
"no_negative_tests": "true",
"generator_args": [
[
"--target-dtype",
"fp8e4m3",
"--target-dtype",
"fp8e5m2",
"--fp-values-range",
"-max,max",
"--tensor-dim-range",
"32,64",
"--num-rand-permutations",
"2"
]
]
}
},
"selection": {
"default": {
"params": {
"shape": [],
"type": [
"f16xf16",
"bf16xbf16",
"f32xf32"
],
"kernel": [],
"stride": [],
"pad": [],
"dilation": []
},
"permutes": [
"kernel",
"shape",
"type",
"pad"
]
}
}
},
"dim": {
"group": "data_layout",
"profile": [
"tosa-mi"
],
"generation": {
"standard": {
"from_version": "v0.80.0",
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--fp-values-range",
"-2.0,2.0",
"--tensor-dim-range",
"1,65",
"--target-rank",
"1",
"--target-rank",
"2",
"--target-rank",
"3"
]
]
},
"float8": {
"from_version" : "v0.100.0",
"no_negative_tests": "true",
"generator_args": [
[
"--target-dtype",
"fp8e4m3",
"--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"
]
}
}
},
"equal": {
"group": "comparison",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--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"
]
]
}
},
"selection": {
"default": {
"params": {},
"permutes": [
"shape",
"type"
]
}
}
},
"exp": {
"group": "ew_unary",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--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",
"1,1,65535,4",
"--target-shape",
"2,65536,1,1,1"
]
]
}
},
"selection": {
"default": {
"params": {},
"permutes": [
"shape",
"type"
]
}
}
},
"fft2d": {
"group": "tensor",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"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"
]
}
}
},
"fully_connected": {
"group": "tensor",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"negative_dim_range": "1,10",
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--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",
"--target-dtype",
"bf16",
"--fp-values-range",
"-max,max",
"--tensor-dim-range",
"35,64",
"--target-shape",
"30,16",
"--target-shape",
"100,23"
]
]
}
},
"selection": {
"default": {
"params": {},
"permutes": [
"shape",
"type"
]
}
}
},
"floor": {
"group": "ew_unary",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--fp-values-range",
"-max,max",
"--tensor-dim-range",
"4,64",
"--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,1,65533,2",
"--target-shape",
"1,65537,1,1"
]
]
}
},
"selection": {
"default": {
"params": {},
"permutes": [
"shape",
"type"
]
}
}
},
"gather": {
"group": "scatter_gather",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"negative_dim_range": "1,10",
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--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"
]
]
},
"float8": {
"from_version" : "v0.100.0",
"no_negative_tests": "true",
"generator_args": [
[
"--target-dtype",
"fp8e4m3",
"--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",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--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"
],
[
"--target-dtype",
"fp32",
"--fp-values-range",
"-max,max",
"--target-shape",
"3,65551,1,1",
"--target-shape",
"1,1,3,65531,1"
]
]
}
},
"selection": {
"default": {
"params": {},
"permutes": [
"shape",
"type"
]
}
}
},
"greater_equal": {
"group": "comparison",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--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"
],
[
"--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",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"no_negative_tests": "true",
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--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",
"--target-dtype",
"bf16",
"--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"
]
]
}
},
"selection": {
"default": {
"params": {},
"permutes": [
"shape",
"type"
]
}
}
},
"log": {
"group": "ew_unary",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--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",
"1,65530,1,2",
"--target-shape",
"3,1,65540,1,1"
]
]
}
},
"selection": {
"default": {
"params": {},
"permutes": [
"shape",
"type"
]
}
}
},
"matmul": {
"group": "tensor",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"negative_dim_range": "1,10",
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--fp-values-range",
"-2.0,2.0",
"--tensor-dim-range",
"13,70"
],
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--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"
]
]
},
"float8": {
"from_version" : "v0.100.0",
"no_negative_tests": "true",
"generator_args": [
[
"--target-dtype",
"fp8e4m3",
"--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": {
"all": "true"
}
}
},
"max_pool2d": {
"group": "tensor",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--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"
],
[
"--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"
]
]
},
"float8": {
"from_version" : "v0.100.0",
"no_negative_tests": "true",
"generator_args": [
[
"--target-dtype",
"fp8e4m3",
"--target-dtype",
"fp8e5m2",
"--fp-values-range",
"-max,max",
"--tensor-dim-range",
"32,64",
"--num-rand-permutations",
"2"
]
]
}
},
"selection": {
"default": {
"params": {
"shape": [],
"type": [
"f32",
"f16",
"bf16"
],
"stride": [],
"pad": []
},
"permutes": [
"shape",
"type",
"pad"
],
"sparsity": {
"pad": 11
}
}
}
},
"negate": {
"group": "ew_unary",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"negative_dim_range": "1,10",
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--fp-values-range",
"-max,max",
"--tensor-dim-range",
"17,64",
"--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,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"
]
]
}
},
"selection": {
"default": {
"params": {},
"permutes": [
"shape",
"type"
]
}
}
},
"pad": {
"group": "data_layout",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen", "generator_select", "stable_random_gen" ],
"generation": {
"standard": {
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--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",
"--target-dtype",
"bf16",
"--fp-values-range",
"-max,max",
"--tensor-dim-range",
"1,11",
"--target-rank",
"4",
"--target-rank",
"5",
"--target-rank",
"6"
],
[
"--target-dtype",
"fp32",
"--fp-values-range",
"-max,max",
"--target-shape",
"1,1,65539,1"
]
]
},
"float8": {
"from_version" : "v0.100.0",
"no_negative_tests": "true",
"generator_args": [
[
"--target-dtype",
"fp8e4m3",
"--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": {
"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]]"
]
}
}
}
},
"pow": {
"group": "ew_binary",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--fp-values-range",
"-max,max",
"--tensor-dim-range",
"16,64",
"--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"
],
[
"--target-dtype",
"bf16",
"--fp-values-range",
"-max,max",
"--target-shape",
"65534,3,1,1",
"--target-shape",
"2,65541,1,1,1"
]
]
}
},
"selection": {
"default": {
"params": {},
"permutes": [
"shape",
"type"
]
}
}
},
"reciprocal": {
"group": "ew_unary",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--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",
"bf16",
"--fp-values-range",
"-max,max",
"--target-shape",
"1,1,65536,4",
"--target-shape",
"2,65535,1,1,1"
]
]
}
},
"selection": {
"default": {
"params": {},
"permutes": [
"shape",
"type"
]
}
}
},
"reshape": {
"group": "data_layout",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--fp-values-range",
"-max,max",
"--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"
]
]
},
"float8": {
"from_version" : "v0.100.0",
"no_negative_tests": "true",
"generator_args": [
[
"--target-dtype",
"fp8e4m3",
"--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"
]
}
}
},
"reverse": {
"group": "data_layout",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--fp-values-range",
"-max,max",
"--tensor-dim-range",
"1,48"
],
[
"--target-dtype",
"fp32",
"--fp-values-range",
"-max,max",
"--target-shape",
"1,1,65537",
"--target-shape",
"1,65535,1,2"
]
]
},
"float8": {
"from_version" : "v0.100.0",
"no_negative_tests": "true",
"generator_args": [
[
"--target-dtype",
"fp8e4m3",
"--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"
]
}
}
},
"cos": {
"group": "ew_unary",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--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"
]
]
}
},
"selection": {
"default": {
"params": {},
"permutes": [
"shape",
"type"
]
}
}
},
"sin": {
"group": "ew_unary",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--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"
]
]
}
},
"selection": {
"default": {
"params": {},
"permutes": [
"shape",
"type"
]
}
}
},
"rsqrt": {
"group": "ew_unary",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--fp-values-range",
"-max,max",
"--tensor-dim-range",
"15,64",
"--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"
],
[
"--target-dtype",
"fp32",
"--fp-values-range",
"-max,max",
"--target-shape",
"1,1,3,65541",
"--target-shape",
"65532,1,3,1,1"
]
]
}
},
"selection": {
"default": {
"params": {},
"permutes": [
"shape",
"type"
]
}
}
},
"scatter": {
"group": "scatter_gather",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"negative_dim_range": "1,10",
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--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"
]
]
},
"float8": {
"from_version" : "v0.100.0",
"no_negative_tests": "true",
"generator_args": [
[
"--target-dtype",
"fp8e4m3",
"--target-dtype",
"fp8e5m2",
"--fp-values-range",
"-max,max",
"--tensor-dim-range",
"32,64",
"--num-rand-permutations",
"2"
]
]
}
},
"selection": {
"default": {
"params": {},
"permutes": [
"shape",
"type"
]
}
}
},
"tile": {
"group": "data_layout",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"negative_dim_range": "1,10",
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--fp-values-range",
"-max,max",
"--tensor-dim-range",
"4,32"
],
[
"--target-dtype",
"fp32",
"--fp-values-range",
"-max,max",
"--target-shape",
"65535",
"--target-shape",
"1,65543,2,1"
]
]
},
"float8": {
"from_version" : "v0.100.0",
"no_negative_tests": "true",
"generator_args": [
[
"--target-dtype",
"fp8e4m3",
"--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",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"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",
"--target-dtype",
"bf16",
"--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"
]
]
},
"float8": {
"from_version" : "v0.100.0",
"no_negative_tests": "true",
"generator_args": [
[
"--target-dtype",
"fp8e4m3",
"--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",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"gen_filter": "^transpose_conv2d",
"generation": {
"standard": {
"negative_dim_range": "1,10",
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--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"
],
[
"--target-dtype",
"fp32",
"--fp-values-range",
"-max,max",
"--target-shape",
"1,65536,1,2",
"--tensor-dim-range",
"1,17",
"--max-conv-dilation",
"1"
]
]
},
"float8": {
"from_version" : "v0.100.0",
"no_negative_tests": "true",
"generator_args": [
[
"--target-dtype",
"fp8e4m3",
"--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"
]
]
}
},
"selection": {
"default": {
"params": {
"shape": [],
"type": [],
"kernel": [],
"stride": [],
"pad": [],
"out_shape": []
},
"permutes": [
"shape",
"type",
"stride",
"pad"
]
}
}
},
"maximum": {
"group": "ew_binary",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--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",
"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",
"65531,1,2,1",
"--target-shape",
"1,1,65541,1"
]
]
}
},
"selection": {
"default": {
"params": {},
"permutes": [
"shape",
"type"
]
}
}
},
"minimum": {
"group": "ew_binary",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--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",
"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,65531,2,1",
"--target-shape",
"1,1,1,65538"
]
]
}
},
"selection": {
"default": {
"params": {},
"permutes": [
"shape",
"type"
]
}
}
},
"mul": {
"group": "ew_binary",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"negative_dim_range": "1,10",
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--fp-values-range",
"-max,max",
"--tensor-dim-range",
"16,64",
"--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"
],
[
"--target-dtype",
"bf16",
"--fp-values-range",
"-max,max",
"--target-shape",
"1,1,3,65534",
"--target-shape",
"1,2,1,65541,1"
]
]
}
},
"selection": {
"default": {
"params": {},
"permutes": [
"shape",
"type"
]
}
}
},
"reduce_max": {
"group": "reduction",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"negative_dim_range": "1,10",
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--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"
],
[
"--target-dtype",
"fp32",
"--fp-values-range",
"-max,max",
"--target-shape",
"1,1,1,65531",
"--target-shape",
"1,1,65539,2"
]
]
},
"8k_level": {
"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,10",
"--target-rank",
"5",
"--target-rank",
"6"
]
]
}
},
"selection": {
"default": {
"params": {},
"permutes": [
"shape",
"type",
"axis"
]
}
}
},
"reduce_min": {
"group": "reduction",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"negative_dim_range": "1,10",
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--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"
],
[
"--target-dtype",
"fp32",
"--fp-values-range",
"-max,max",
"--target-shape",
"1,1,65531,1",
"--target-shape",
"1,65539,2,1"
]
]
},
"8k_level": {
"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,9",
"--target-rank",
"5",
"--target-rank",
"6"
]
]
}
},
"selection": {
"default": {
"params": {},
"permutes": [
"shape",
"type",
"axis"
]
}
}
},
"reduce_product": {
"group": "reduction",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--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"
],
[
"--target-dtype",
"fp16",
"--fp-values-range",
"-max,max",
"--target-shape",
"2,65527,3,1",
"--target-shape",
"1,3,2,65541"
]
]
},
"8k_level": {
"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",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--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",
"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"
],
[
"--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": {
"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",
"5,12",
"--target-rank",
"5",
"--target-rank",
"6"
]
]
}
},
"selection": {
"default": {
"params": {},
"permutes": [
"shape",
"type",
"axis"
]
}
}
},
"resize": {
"group": "image",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"negative_dim_range": "1,10",
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--fp-values-range",
"-max,max"
],
[
"--target-dtype",
"fp32",
"--fp-values-range",
"-max,max",
"--target-shape",
"1,1103,1,2",
"--max-resize-output-dim",
"9100"
]
]
}
},
"selection": {
"default": {
"params": {
"offset": [],
"shape": [],
"type": [],
"mode": [],
"output_type": [],
"scale": [
"sc2x1x3x1"
],
"border": [
"bor0x1"
]
},
"permutes": [
"shape",
"type"
],
"exclude_patterns": [
".*_ERRORIF_MaxDimExceeded_1x16584x5x1_.*"
]
}
}
},
"rfft2d": {
"group": "tensor",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"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"
]
}
}
},
"select": {
"group": "ew_ternary",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--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",
"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,2,65534,2,1",
"--target-shape",
"1,1,65539,2",
"--num-rand-permutations",
"2"
]
]
}
},
"selection": {
"default": {
"params": {},
"permutes": [
"shape",
"type"
]
}
}
},
"sigmoid": {
"group": "activation",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--fp-values-range",
"-16.0,16.0",
"--tensor-dim-range",
"16,64",
"--target-rank",
"1",
"--target-rank",
"2",
"--target-rank",
"3"
],
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--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"
]
]
}
},
"selection": {
"default": {
"params": {},
"permutes": [
"shape",
"type"
]
}
}
},
"sub": {
"group": "ew_binary",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--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"
],
[
"--target-dtype",
"fp32",
"--fp-values-range",
"-max,max",
"--target-shape",
"1,1,65533,3",
"--target-shape",
"1,1,2,1,65545"
]
]
}
},
"selection": {
"default": {
"params": {},
"permutes": [
"shape",
"type"
]
}
}
},
"slice": {
"group": "data_layout",
"profile": [
"tosa-bi",
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--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",
"fp32",
"--target-dtype",
"fp16",
"--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"
]
]
},
"float8": {
"from_version" : "v0.100.0",
"no_negative_tests": "true",
"generator_args": [
[
"--target-dtype",
"fp8e4m3",
"--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"
]
}
}
},
"tanh": {
"group": "activation",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--fp-values-range",
"-8.0,8.0",
"--tensor-dim-range",
"18,60",
"--target-rank",
"1",
"--target-rank",
"2",
"--target-rank",
"3"
],
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--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"
]
]
}
},
"selection": {
"default": {
"params": {},
"permutes": [
"shape",
"type"
]
}
}
},
"erf": {
"group": "activation",
"profile": [
"tosa-mi"
],
"support_for": [ "lazy_data_gen" ],
"generation": {
"standard": {
"generator_args": [
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--fp-values-range",
"-4.0,4.0",
"--tensor-dim-range",
"18,60",
"--target-rank",
"1",
"--target-rank",
"2",
"--target-rank",
"3"
],
[
"--target-dtype",
"fp32",
"--target-dtype",
"fp16",
"--target-dtype",
"bf16",
"--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"
]
]
}
},
"selection": {
"default": {
"params": {},
"permutes": [
"shape",
"type"
]
}
}
}
}