diff --git a/backends/arm/operators/op_table.py b/backends/arm/operators/op_table.py index da7e2e8be95..40214b265f0 100644 --- a/backends/arm/operators/op_table.py +++ b/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] diff --git a/backends/arm/test/misc/test_debug_feats.py b/backends/arm/test/misc/test_debug_feats.py index 4a3767833a3..945c940a20b 100644 --- a/backends/arm/test/misc/test_debug_feats.py +++ b/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) diff --git a/backends/arm/test/misc/test_partition_decomposed_quantized_ops.py b/backends/arm/test/misc/test_partition_decomposed_quantized_ops.py index 5bb692ebcaf..f69d9d34462 100644 --- a/backends/arm/test/misc/test_partition_decomposed_quantized_ops.py +++ b/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(), diff --git a/backends/arm/test/models/test_nn_functional.py b/backends/arm/test/models/test_nn_functional.py index 40e83aa05c4..f258e54ac7f 100644 --- a/backends/arm/test/models/test_nn_functional.py +++ b/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](