[Mlir-commits] [mlir] 8189a2b - [mlir][sparse][pytaco] migrate to sparse compiler pipeline

[Aart Bik]

Author: Aart Bik
Date: 2022-02-10T07:42:54-08:00
New Revision: 8189a2b8bd52d88ee6968e252f8c96295d431ff4

URL: https://github.com/llvm/llvm-project/commit/8189a2b8bd52d88ee6968e252f8c96295d431ff4
DIFF: https://github.com/llvm/llvm-project/commit/8189a2b8bd52d88ee6968e252f8c96295d431ff4.diff

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

Reviewed By: bixia

Differential Revision: https://reviews.llvm.org/D119395

Added: 
    

Modified: 
    mlir/test/Integration/Dialect/SparseTensor/taco/tools/mlir_pytaco_utils.py

Removed: 
    


################################################################################
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 c815220815126..25221f32fff99 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()])