From 886a90e94523545207b7c630d80fbfc6210600ab Mon Sep 17 00:00:00 2001 From: Suraj Sudhir Date: Wed, 19 Jan 2022 10:03:31 -0800 Subject: [PATCH] [tosa] Enable tosa-to-linalg-named so Matmul works again Signed-off-by: Suraj Sudhir --- e2e_testing/torchscript/xfail_sets.py | 3 +++ .../torch_mlir_e2e_test/tosa_backends/linalg_on_tensors.py | 6 ++++++ 2 files changed, 9 insertions(+) diff --git a/e2e_testing/torchscript/xfail_sets.py b/e2e_testing/torchscript/xfail_sets.py index 7f9be0987ac9..6275ab762013 100644 --- a/e2e_testing/torchscript/xfail_sets.py +++ b/e2e_testing/torchscript/xfail_sets.py @@ -51,4 +51,7 @@ "SqueezeDimModule_unitDim", "ReturnTwoTensorF32I64_basic", "ElementwisePowModule_basic", + "BmmModule_basic", + "Matmul_dot", + "Matmul_3d", } diff --git a/python/torch_mlir_e2e_test/tosa_backends/linalg_on_tensors.py b/python/torch_mlir_e2e_test/tosa_backends/linalg_on_tensors.py index f2755e1c3a3b..3b1287be3256 100644 --- a/python/torch_mlir_e2e_test/tosa_backends/linalg_on_tensors.py +++ b/python/torch_mlir_e2e_test/tosa_backends/linalg_on_tensors.py @@ -48,6 +48,12 @@ def compile(self, imported_module: Module): "builtin.func(tosa-to-standard)", "Lowering TOSA to Standard") + # Named ops must be legalized prior to general tosa-to-linalg + run_pipeline_with_repro_report( + imported_module, + "builtin.func(tosa-to-linalg-named)", + "Lowering TOSA to Linalg-on-Tensors for Named Ops") + run_pipeline_with_repro_report( imported_module, "builtin.func(tosa-to-linalg)",