diff --git a/mlir/benchmark/python/common.py b/mlir/benchmark/python/common.py --- a/mlir/benchmark/python/common.py +++ b/mlir/benchmark/python/common.py @@ -19,25 +19,8 @@ "parallelization-strategy=0" " vectorization-strategy=0 vl=1 enable-simd-index32=False" ) - pipeline = ( - f"builtin.func" - f"(linalg-generalize-named-ops,linalg-fuse-elementwise-ops)," - f"sparsification{{{opt}}}," - f"sparse-tensor-conversion," - f"builtin.func" - f"(linalg-bufferize,convert-linalg-to-loops,convert-vector-to-scf)," - f"convert-scf-to-std," - f"func-bufferize," - f"tensor-constant-bufferize," - f"builtin.func(tensor-bufferize,std-bufferize,finalizing-bufferize)," - f"convert-vector-to-llvm" - f"{{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(mlir_module) + pipeline_new = f"builtin.module(sparse-compiler{{{opt}}})" + PassManager.parse(pipeline_new).run(mlir_module) def create_sparse_np_tensor(dimensions, number_of_elements):