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.
|
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()])
|
||||||
|
|
Loading…
Reference in New Issue