1 //===- SparseTensorPipelines.cpp - Pipelines for sparse tensor code -------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 
9 #include "mlir/Dialect/SparseTensor/Pipelines/Passes.h"
10 
11 #include "mlir/Conversion/Passes.h"
12 #include "mlir/Dialect/Arithmetic/Transforms/Passes.h"
13 #include "mlir/Dialect/Bufferization/Transforms/Passes.h"
14 #include "mlir/Dialect/Linalg/Passes.h"
15 #include "mlir/Dialect/SparseTensor/IR/SparseTensor.h"
16 #include "mlir/Dialect/SparseTensor/Transforms/Passes.h"
17 #include "mlir/Dialect/StandardOps/Transforms/Passes.h"
18 #include "mlir/Dialect/Tensor/Transforms/Passes.h"
19 #include "mlir/Pass/PassManager.h"
20 
21 using namespace mlir;
22 using namespace mlir::sparse_tensor;
23 
24 //===----------------------------------------------------------------------===//
25 // Pipeline implementation.
26 //===----------------------------------------------------------------------===//
27 
28 void mlir::sparse_tensor::buildSparseCompiler(
29     OpPassManager &pm, const SparseCompilerOptions &options) {
30   pm.addPass(createSparsificationPass(options.sparsificationOptions()));
31   pm.addPass(createSparseTensorConversionPass());
32   pm.addPass(createLinalgBufferizePass());
33   pm.addPass(createConvertLinalgToLoopsPass());
34   pm.addPass(createConvertVectorToSCFPass());
35   pm.addPass(createLowerToCFGPass()); // --convert-scf-to-std
36   pm.addPass(createFuncBufferizePass());
37   pm.addPass(arith::createConstantBufferizePass());
38   pm.addPass(createTensorBufferizePass());
39   pm.addPass(createStdBufferizePass());
40   pm.addPass(mlir::bufferization::createFinalizingBufferizePass());
41   pm.addPass(createLowerAffinePass());
42   pm.addPass(createConvertVectorToLLVMPass());
43   pm.addPass(createMemRefToLLVMPass());
44   pm.addPass(createConvertMathToLLVMPass());
45   pm.addPass(createLowerToLLVMPass()); // --convert-std-to-llvm
46   pm.addPass(createReconcileUnrealizedCastsPass());
47 }
48 
49 //===----------------------------------------------------------------------===//
50 // Pipeline registration.
51 //===----------------------------------------------------------------------===//
52 
53 void mlir::sparse_tensor::registerSparseTensorPipelines() {
54   PassPipelineRegistration<SparseCompilerOptions>(
55       "sparse-compiler",
56       "The standard pipeline for taking sparsity-agnostic IR using the"
57       " sparse-tensor type, and lowering it to LLVM IR with concrete"
58       " representations and algorithms for sparse tensors.",
59       buildSparseCompiler);
60 }
61