Arm Backend: Fixes related to pytorch updates

- Update to error message for unsupported types
- Check file extension in test_debug_feats
- Adding failures for some unsupported operators in tests

Author: SaoirseARM

backends/arm/operators/op_table.py

@@ -41,7 +41,7 @@ def define_node(
 
         if inputs[0].dtype not in (ts.DType.INT8, ts.DType.INT16):
             raise ValueError(
-                f"TOSA.TABLE only supports int8 or int16 inputs, got {ts.DTypeNames[inputs[0]]}"
+                f"TOSA.TABLE only supports int8 or int16 inputs, got {ts.DTypeNames[inputs[0].dtype]}"
             )
 
         table = self._exported_program.state_dict[node.name]  # type: ignore[union-attr]

backends/arm/test/misc/test_debug_feats.py

@@ -192,16 +192,13 @@ def test_collate_tosa_BI_tests(self):
             .to_edge_transform_and_lower()
             .to_executorch()
         )
+
+        test_collate_dir = "test_collate_tosa_tests/tosa-bi/TestCollateTosaTests/test_collate_tosa_BI_tests"
         # test that the output directory is created and contains the expected files
-        assert os.path.exists(
-            "test_collate_tosa_tests/tosa-bi/TestCollateTosaTests/test_collate_tosa_BI_tests"
-        )
-        assert os.path.exists(
-            "test_collate_tosa_tests/tosa-bi/TestCollateTosaTests/test_collate_tosa_BI_tests/output_tag6_TOSA-0.80+BI.tosa"
-        )
-        assert os.path.exists(
-            "test_collate_tosa_tests/tosa-bi/TestCollateTosaTests/test_collate_tosa_BI_tests/desc_tag6_TOSA-0.80+BI.json"
-        )
+        assert os.path.exists(test_collate_dir)
+
+        for file in os.listdir(test_collate_dir):
+            assert file.endswith(("TOSA-0.80+BI.json", "TOSA-0.80+BI.tosa"))
 
         os.environ.pop("TOSA_TESTCASES_BASE_PATH")
         shutil.rmtree("test_collate_tosa_tests", ignore_errors=True)

backends/arm/test/misc/test_partition_decomposed_quantized_ops.py

@@ -145,7 +145,12 @@ def test_linear_residaul_tosa_MI(test_data: input_t1):
     pipeline.run()
 
 
-@common.parametrize("test_data", test_data)
+@common.parametrize(
+    "test_data",
+    test_data,
+    {"3d_rand": "MLETORCH-855: Issue with Quantization folding."},
+    strict=False,
+)
 def test_linear_residual_tosa_BI(test_data: input_t1):
     pipeline = TosaPipelineBI[input_t1](
         LinearResidualModule(),

backends/arm/test/models/test_nn_functional.py

@@ -104,7 +104,15 @@ def test_nn_functional_MI(test_data):
             raise e
 
 
-@parametrize("test_data", module_tests)
+x_fails = {
+    "normalize": "MLETORCH-852: Support aten.index_put.default",
+    "cosine_similarity": "MLETORCH-854: Support aten.linalg_vector_norm.default",
+    "unfold": "Int64 input && MLETORCH-827: Support aten.index.Tensor",
+    "fold": "Int64 input && MLETORCH-827: Support aten.index_put.default",
+}
+
+
+@parametrize("test_data", module_tests, x_fails, strict=False)
 def test_nn_functional_BI(test_data):
     module, inputs = test_data
     pipeline = TosaPipelineBI[input_t](