forked from OSchip/llvm-project
[mlir][sparse][pytaco] migrate to sparse compiler pipeline
Reviewed By: bixia Differential Revision: https://reviews.llvm.org/D119395
This commit is contained in:
parent
02a05097e7
commit
8189a2b8bd
|
@ -154,19 +154,7 @@ def compile_and_build_engine(
|
|||
A JIT execution engine for the MLIR module.
|
||||
|
||||
"""
|
||||
pipeline = (
|
||||
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")
|
||||
pipeline = f"sparse-compiler"
|
||||
PassManager.parse(pipeline).run(module)
|
||||
return execution_engine.ExecutionEngine(
|
||||
module, opt_level=_OPT_LEVEL, shared_libs=[_get_support_lib_name()])
|
||||
|
|
Loading…
Reference in New Issue