Create MI tests for Data Layout: CONCAT, PAD, RESHAPE, REVERSE, SLICE, TILE, TRANSPOSE

Signed-off-by: James Ward <james.ward@arm.com>
Change-Id: I626d9ece0290c5104c9ac42d20be2f9db1e55fa7

2 files changed, 314 insertions, 1 deletions

diff --git a/verif/conformance/tosa_main_profile_ops_info.json b/verif/conformance/tosa_main_profile_ops_info.json
index 9846c29..9d2ae09 100644
--- a/verif/conformance/tosa_main_profile_ops_info.json
+++ b/verif/conformance/tosa_main_profile_ops_info.json
@@ -105,6 +105,69 @@
             "tosa-mi"
         ]
     },
+    "concat": {
+        "group": "data_layout",
+        "generator_args": [
+            [
+                "--target-dtype",
+                "fp32",
+                "--target-dtype",
+                "fp16",
+                "--target-dtype",
+                "bf16",
+                "--fp-values-range",
+                "-2.0,2.0",
+                "--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",
+                "-2.0,2.0",
+                "--tensor-dim-range",
+                "1,16",
+                "--target-rank",
+                "4",
+                "--target-rank",
+                "5",
+                "--num-const-inputs-concat",
+                "1"
+            ],
+            [
+                "--target-dtype",
+                "fp32",
+                "--fp-values-range",
+                "-2.0,2.0",
+                "--target-shape",
+                "1,1,65534,1,1",
+                "--target-shape",
+                "1,65537,1,2",
+                "--num-const-inputs-concat",
+                "2"
+            ]
+        ],
+        "params": {},
+        "permutes": [
+            "shape",
+            "type"
+        ],
+        "profile": [
+            "tosa-mi"
+        ]
+    },
     "conv2d": {
         "group": "tensor",
         "generator_args": [
@@ -246,6 +309,159 @@
             "tosa-mi"
         ]
     },
+    "pad": {
+        "group": "data_layout",
+        "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"
+            ],
+            [
+                "--target-dtype",
+                "fp16",
+                "--fp-values-range",
+                "-2.0,2.0",
+                "--tensor-dim-range",
+                "1,17",
+                "--target-rank",
+                "4"
+            ],
+            [
+                "--target-dtype",
+                "bf16",
+                "--fp-values-range",
+                "-2.0,2.0",
+                "--tensor-dim-range",
+                "1,16",
+                "--target-rank",
+                "5"
+            ],
+            [
+                "--target-dtype",
+                "fp32",
+                "--fp-values-range",
+                "-2.0,2.0",
+                "--target-shape",
+                "1,1,65539,1"
+            ]
+        ],
+        "params": {},
+        "permutes": [
+            "shape",
+            "type"
+        ],
+        "sparsity": {
+            "pad": 21
+        },
+        "preselected": [
+            {
+                "shape": "50",
+                "type": "bf16",
+                "pad": "pad11"
+            },
+            {
+                "shape": "63x46",
+                "type": "bf16",
+                "pad": "pad1010"
+            },
+            {
+                "shape": "6",
+                "type": "f16",
+                "pad": "pad01"
+            }
+        ],
+        "profile": [
+            "tosa-mi"
+        ]
+    },
+    "reshape": {
+        "group": "data_layout",
+        "generator_args": [
+            [
+                "--target-dtype",
+                "fp32",
+                "--target-dtype",
+                "fp16",
+                "--target-dtype",
+                "bf16",
+                "--fp-values-range",
+                "-2.0,2.0",
+                "--num-rand-permutations",
+                "2"
+            ],
+            [
+                "--target-dtype",
+                "fp32",
+                "--fp-values-range",
+                "-2.0,2.0",
+                "--target-shape",
+                "1,1,65536,1",
+                "--target-shape",
+                "1,65535,1,2",
+                "--num-rand-permutations",
+                "2"
+            ]
+        ],
+        "params": {},
+        "permutes": [
+            "shape",
+            "type",
+            "rank",
+            "perm"
+        ],
+        "profile": [
+            "tosa-mi"
+        ]
+    },
+    "reverse": {
+        "group": "data_layout",
+        "generator_args": [
+            [
+                "--target-dtype",
+                "fp32",
+                "--target-dtype",
+                "fp16",
+                "--target-dtype",
+                "bf16",
+                "--fp-values-range",
+                "-2.0,2.0",
+                "--tensor-dim-range",
+                "1,48"
+            ],
+            [
+                "--target-dtype",
+                "fp32",
+                "--fp-values-range",
+                "-2.0,2.0",
+                "--target-shape",
+                "1,1,65537",
+                "--target-shape",
+                "1,65535,1,2"
+            ]
+        ],
+        "params": {},
+        "permutes": [
+            "shape",
+            "type"
+        ],
+        "profile": [
+            "tosa-mi"
+        ]
+    },
     "scatter": {
         "group": "scatter_gather",
         "generator_args": [
@@ -282,5 +498,102 @@
         "profile": [
             "tosa-mi"
         ]
+    },
+    "tile": {
+        "group": "data_layout",
+        "generator_args": [
+            [
+                "--target-dtype",
+                "fp32",
+                "--target-dtype",
+                "fp16",
+                "--target-dtype",
+                "bf16",
+                "--fp-values-range",
+                "-2.0,2.0",
+                "--tensor-dim-range",
+                "4,32"
+            ],
+            [
+                "--target-dtype",
+                "fp32",
+                "--fp-values-range",
+                "-2.0,2.0",
+                "--target-shape",
+                "65535",
+                "--target-shape",
+                "1,65543,2,1"
+            ]
+        ],
+        "params": {},
+        "permutes": [
+            "shape",
+            "type"
+        ],
+        "profile": [
+            "tosa-mi"
+        ]
+    },
+    "transpose": {
+        "group": "data_layout",
+        "generator_args": [
+            [
+                "--filter",
+                "^transpose$",
+                "--target-dtype",
+                "fp32",
+                "--target-dtype",
+                "fp16",
+                "--fp-values-range",
+                "-2.0,2.0",
+                "--target-rank",
+                "1",
+                "--target-rank",
+                "2",
+                "--num-rand-permutations",
+                "2"
+            ],
+            [
+                "--filter",
+                "^transpose$",
+                "--target-dtype",
+                "fp32",
+                "--target-dtype",
+                "fp16",
+                "--target-dtype",
+                "bf16",
+                "--fp-values-range",
+                "-2.0,2.0",
+                "--target-rank",
+                "3",
+                "--target-rank",
+                "4",
+                "--num-rand-permutations",
+                "2"
+            ],
+            [
+                "--filter",
+                "^transpose$",
+                "--target-dtype",
+                "fp32",
+                "--fp-values-range",
+                "-2.0,2.0",
+                "--target-shape",
+                "1,1,65539,2",
+                "--target-shape",
+                "1,1,1,65535",
+                "--num-rand-permutations",
+                "1"
+            ]
+        ],
+        "params": {},
+        "permutes": [
+            "shape",
+            "type",
+            "perm"
+        ],
+        "profile": [
+            "tosa-mi"
+        ]
     }
 }
\ No newline at end of file
diff --git a/verif/generator/tosa_arg_gen.py b/verif/generator/tosa_arg_gen.py
index 932ad55..4e15b06 100644
--- a/verif/generator/tosa_arg_gen.py
+++ b/verif/generator/tosa_arg_gen.py
@@ -1247,7 +1247,7 @@ class TosaArgGen:
         if dtype in [DType.BOOL, DType.INT8, DType.INT16, DType.INT32]:
             pad_const_int = testGen.getRandNumberDType(dtype)
             pad_const_fp = 0
-        elif dtype in (DType.FP16, DType.FP32):
+        elif dtype in (DType.FP16, DType.BF16, DType.FP32):
             pad_const_int = 0
             pad_const_fp = testGen.getRandNumberDType(dtype)
         else: