[mlir] Use sparse-compiler pass in sparse benchmark

This patch uses sparse-compiler option in mlir sparse tensor benchmark

Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D118579
This commit is contained in:
Saurabh Jha
2022-02-09 20:10:30 +00:00
parent aa845d7a24
commit 0ed0a8e2f0

View File

@@ -19,24 +19,7 @@ def setup_passes(mlir_module):
"parallelization-strategy=0"
" vectorization-strategy=0 vl=1 enable-simd-index32=False"
)
pipeline = (
f"builtin.func"
f"(linalg-generalize-named-ops,linalg-fuse-elementwise-ops),"
f"sparsification{{{opt}}},"
f"sparse-tensor-conversion,"
f"builtin.func"
f"(linalg-bufferize,convert-linalg-to-loops,convert-vector-to-scf),"
f"convert-scf-to-cf,"
f"func-bufferize,"
f"arith-bufferize,"
f"builtin.func(tensor-bufferize,finalizing-bufferize),"
f"convert-vector-to-llvm"
f"{{reassociate-fp-reductions=1 enable-index-optimizations=1}},"
f"lower-affine,"
f"convert-memref-to-llvm,"
f"convert-std-to-llvm,"
f"reconcile-unrealized-casts"
)
pipeline = f"sparse-compiler{{{opt}}}"
PassManager.parse(pipeline).run(mlir_module)