MLIR  18.0.0git
Passes.h
Go to the documentation of this file.
1 //===- Passes.h - Linalg pass entry points ----------------------*- C++ -*-===//
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 // This header file defines prototypes that expose pass constructors.
10 //
11 //===----------------------------------------------------------------------===//
12 
13 #ifndef MLIR_DIALECT_LINALG_PASSES_H_
14 #define MLIR_DIALECT_LINALG_PASSES_H_
15 
18 #include "mlir/Pass/Pass.h"
19 
20 namespace mlir {
21 namespace func {
22 class FuncOp;
23 } // namespace func
24 
25 namespace bufferization {
26 struct OneShotBufferizationOptions;
27 } // namespace bufferization
28 
29 #define GEN_PASS_DECL
30 #include "mlir/Dialect/Linalg/Passes.h.inc"
31 
32 std::unique_ptr<Pass> createConvertElementwiseToLinalgPass();
33 
34 std::unique_ptr<Pass> createLinalgFoldUnitExtentDimsPass();
35 
36 std::unique_ptr<Pass> createLinalgElementwiseOpFusionPass();
38 
39 std::unique_ptr<Pass> createLinalgNamedOpConversionPass();
40 
41 std::unique_ptr<Pass> createLinalgInlineScalarOperandsPass();
42 
43 /// Create a pass to convert Linalg operations to scf.for loops and
44 /// memref.load/memref.store accesses.
45 std::unique_ptr<Pass> createConvertLinalgToLoopsPass();
46 
47 /// Create a pass to convert Linalg operations to scf.parallel loops and
48 /// memref.load/memref.store accesses.
49 std::unique_ptr<Pass> createConvertLinalgToParallelLoopsPass();
50 
51 /// Create a pass to convert Linalg operations to affine.for loops and
52 /// affine_load/affine_store accesses.
53 /// Placeholder for now, this is NYI.
54 std::unique_ptr<Pass> createConvertLinalgToAffineLoopsPass();
55 
56 /// Create a pass to convert Linalg operations which work on tensors to use
57 /// buffers instead.
58 std::unique_ptr<Pass> createLinalgBufferizePass();
59 
60 /// Create a pass to convert named Linalg operations to Linalg generic
61 /// operations.
62 std::unique_ptr<Pass> createLinalgGeneralizationPass();
63 
64 /// Create a pass to convert Linalg operations to equivalent operations that
65 /// work on primitive types, if possible.
66 std::unique_ptr<Pass> createLinalgDetensorizePass();
67 
68 //===----------------------------------------------------------------------===//
69 // Registration
70 //===----------------------------------------------------------------------===//
71 
72 /// Generate the code for registering passes.
73 #define GEN_PASS_REGISTRATION
74 #include "mlir/Dialect/Linalg/Passes.h.inc"
75 
76 } // namespace mlir
77 
78 #endif // MLIR_DIALECT_LINALG_PASSES_H_
This header declares functions that assist transformations in the MemRef dialect.
std::unique_ptr< Pass > createLinalgInlineScalarOperandsPass()
std::unique_ptr< Pass > createFoldReshapeOpsByLinearizationPass()
std::unique_ptr< Pass > createLinalgBufferizePass()
Create a pass to convert Linalg operations which work on tensors to use buffers instead.
Definition: Bufferize.cpp:52
std::unique_ptr< Pass > createLinalgGeneralizationPass()
Create a pass to convert named Linalg operations to Linalg generic operations.
std::unique_ptr< Pass > createLinalgDetensorizePass()
Create a pass to convert Linalg operations to equivalent operations that work on primitive types,...
std::unique_ptr< Pass > createLinalgElementwiseOpFusionPass()
std::unique_ptr< Pass > createConvertLinalgToLoopsPass()
Create a pass to convert Linalg operations to scf.for loops and memref.load/memref....
Definition: Loops.cpp:356
std::unique_ptr< Pass > createLinalgFoldUnitExtentDimsPass()
std::unique_ptr< Pass > createConvertLinalgToParallelLoopsPass()
Create a pass to convert Linalg operations to scf.parallel loops and memref.load/memref....
Definition: Loops.cpp:360
std::unique_ptr< Pass > createConvertElementwiseToLinalgPass()
std::unique_ptr< Pass > createConvertLinalgToAffineLoopsPass()
Create a pass to convert Linalg operations to affine.for loops and affine_load/affine_store accesses.
Definition: Loops.cpp:364
std::unique_ptr< Pass > createLinalgNamedOpConversionPass()