This is an archive of the discontinued LLVM Phabricator instance.

Show First 20 Lines • Show All 148 Lines • ▼ Show 20 Lines	def compile_and_build_engine(

Args:		Args:
module: The MLIR module.		module: The MLIR module.

Returns:		Returns:
A JIT execution engine for the MLIR module.		A JIT execution engine for the MLIR module.

"""		"""
pipeline = (		pipeline = f"sparse-compiler"
		bixiaUnsubmitted Not Done Reply Inline Actions Nit: f is not needed. bixia: Nit: f is not needed.
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()])


class _SparseTensorDescriptor(ctypes.Structure):		class _SparseTensorDescriptor(ctypes.Structure):
"""A C structure for an MLIR sparse tensor."""		"""A C structure for an MLIR sparse tensor."""
_fields_ = [		_fields_ = [
▲ Show 20 Lines • Show All 159 Lines • Show Last 20 Lines

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