llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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

Signed-off-by: Suraj Sudhir <suraj.sudhir@arm.com>
pull/531/head
Suraj Sudhir 2022-01-19 12:10:04 -08:00 committed by GitHub
parent 6fe70c7794
commit 5d6c4f48dc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
e2e_testing/torchscript/xfail_sets.py
View File

@ -51,4 +51,7 @@ TOSA_PASS_SET = {
"SqueezeDimModule_unitDim",
"ReturnTwoTensorF32I64_basic",
"ElementwisePowModule_basic",
"BmmModule_basic",
"Matmul_dot",
"Matmul_3d",
}

6
python/torch_mlir_e2e_test/tosa_backends/linalg_on_tensors.py
View File

@ -48,6 +48,12 @@ class LinalgOnTensorsTosaBackend(TosaBackend):
"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)",