From 8189a2b8bd52d88ee6968e252f8c96295d431ff4 Mon Sep 17 00:00:00 2001 From: Aart Bik Date: Wed, 9 Feb 2022 17:07:54 -0800 Subject: [PATCH] [mlir][sparse][pytaco] migrate to sparse compiler pipeline Reviewed By: bixia Differential Revision: https://reviews.llvm.org/D119395 --- .../SparseTensor/taco/tools/mlir_pytaco_utils.py | 14 +------------- 1 file changed, 1 insertion(+), 13 deletions(-) diff --git a/mlir/test/Integration/Dialect/SparseTensor/taco/tools/mlir_pytaco_utils.py b/mlir/test/Integration/Dialect/SparseTensor/taco/tools/mlir_pytaco_utils.py index c81522081512..25221f32fff9 100644 --- a/mlir/test/Integration/Dialect/SparseTensor/taco/tools/mlir_pytaco_utils.py +++ b/mlir/test/Integration/Dialect/SparseTensor/taco/tools/mlir_pytaco_utils.py @@ -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()])