[tosa] Enable tosa-to-linalg-named so Matmul works again (#530)

sjarus · web-flow · commit 5d6c4f48dcdc · 2022-01-19T12:10:04.000-08:00
Signed-off-by: Suraj Sudhir &lt;suraj.sudhir@arm.com&gt;
diff --git 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
@@ -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)",

Original file line number	Diff line number	Diff line change
`@@ -51,4 +51,7 @@`
`51`	`51`	`"SqueezeDimModule_unitDim",`
`52`	`52`	`"ReturnTwoTensorF32I64_basic",`
`53`	`53`	`"ElementwisePowModule_basic",`
	`54`	`+ "BmmModule_basic",`
	`55`	`+ "Matmul_dot",`
	`56`	`+ "Matmul_3d",`
`54`	`57`	`}`