[mlir][sparse][pytaco] migrate to sparse compiler pipeline

Reviewed By: bixia

Differential Revision: https://reviews.llvm.org/D119395
This commit is contained in:
Aart Bik 2022-02-09 17:07:54 -08:00
parent 02a05097e7
commit 8189a2b8bd
1 changed files with 1 additions and 13 deletions

View File

@ -154,19 +154,7 @@ def compile_and_build_engine(
A JIT execution engine for the MLIR module. A JIT execution engine for the MLIR module.
""" """
pipeline = ( pipeline = f"sparse-compiler"
f"sparsification,"
f"sparse-tensor-conversion,"
f"builtin.func(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{{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")
PassManager.parse(pipeline).run(module) PassManager.parse(pipeline).run(module)
return execution_engine.ExecutionEngine( return execution_engine.ExecutionEngine(
module, opt_level=_OPT_LEVEL, shared_libs=[_get_support_lib_name()]) module, opt_level=_OPT_LEVEL, shared_libs=[_get_support_lib_name()])