MLIR  16.0.0git
Passes.h
Go to the documentation of this file.
1 //===- Passes.h - Pass Entrypoints ------------------------------*- 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_SCF_TRANSFORMS_PASSES_H_
14 #define MLIR_DIALECT_SCF_TRANSFORMS_PASSES_H_
15 
16 #include "mlir/Pass/Pass.h"
17 
18 namespace mlir {
19 
20 #define GEN_PASS_DECL
21 #include "mlir/Dialect/SCF/Transforms/Passes.h.inc"
22 
23 /// Creates a pass that bufferizes the SCF dialect.
24 std::unique_ptr<Pass> createSCFBufferizePass();
25 
26 /// Creates a pass that specializes for loop for unrolling and
27 /// vectorization.
28 std::unique_ptr<Pass> createForLoopSpecializationPass();
29 
30 /// Creates a pass that peels for loops at their upper bounds for
31 /// better vectorization.
32 std::unique_ptr<Pass> createForLoopPeelingPass();
33 
34 /// Creates a pass that canonicalizes affine.min and affine.max operations
35 /// inside of scf.for loops with known lower and upper bounds.
36 std::unique_ptr<Pass> createSCFForLoopCanonicalizationPass();
37 
38 /// Creates a pass that transforms a single ParallelLoop over N induction
39 /// variables into another ParallelLoop over less than N induction variables.
40 std::unique_ptr<Pass> createParallelLoopCollapsingPass();
41 
42 /// Creates a loop fusion pass which fuses parallel loops.
43 std::unique_ptr<Pass> createParallelLoopFusionPass();
44 
45 /// Creates a pass that specializes parallel loop for unrolling and
46 /// vectorization.
47 std::unique_ptr<Pass> createParallelLoopSpecializationPass();
48 
49 /// Creates a pass which tiles innermost parallel loops.
50 /// If noMinMaxBounds, the upper bound of the inner loop will
51 /// be a same value among different outter loop iterations, and
52 /// an additional inbound check will be emitted inside the internal
53 /// loops.
54 std::unique_ptr<Pass>
56  bool noMinMaxBounds = false);
57 
58 /// Creates a pass which folds arith ops on induction variable into
59 /// loop range.
60 std::unique_ptr<Pass> createForLoopRangeFoldingPass();
61 
62 // Creates a pass which lowers for loops into while loops.
63 std::unique_ptr<Pass> createForToWhileLoopPass();
64 
65 //===----------------------------------------------------------------------===//
66 // Registration
67 //===----------------------------------------------------------------------===//
68 
69 /// Generate the code for registering passes.
70 #define GEN_PASS_REGISTRATION
71 #include "mlir/Dialect/SCF/Transforms/Passes.h.inc"
72 
73 } // namespace mlir
74 
75 #endif // MLIR_DIALECT_SCF_TRANSFORMS_PASSES_H_
Include the generated interface declarations.
std::unique_ptr< Pass > createParallelLoopSpecializationPass()
Creates a pass that specializes parallel loop for unrolling and vectorization.
std::unique_ptr< Pass > createForLoopSpecializationPass()
Creates a pass that specializes for loop for unrolling and vectorization.
std::unique_ptr< Pass > createParallelLoopTilingPass(llvm::ArrayRef< int64_t > tileSize={}, bool noMinMaxBounds=false)
Creates a pass which tiles innermost parallel loops.
std::unique_ptr< Pass > createForLoopRangeFoldingPass()
Creates a pass which folds arith ops on induction variable into loop range.
std::unique_ptr< Pass > createParallelLoopCollapsingPass()
Creates a pass that transforms a single ParallelLoop over N induction variables into another Parallel...
std::unique_ptr< Pass > createForToWhileLoopPass()
Definition: ForToWhile.cpp:117
std::unique_ptr< Pass > createForLoopPeelingPass()
Creates a pass that peels for loops at their upper bounds for better vectorization.
std::unique_ptr< Pass > createSCFBufferizePass()
Creates a pass that bufferizes the SCF dialect.
Definition: Bufferize.cpp:45
std::unique_ptr< Pass > createSCFForLoopCanonicalizationPass()
Creates a pass that canonicalizes affine.min and affine.max operations inside of scf....
std::unique_ptr< Pass > createParallelLoopFusionPass()
Creates a loop fusion pass which fuses parallel loops.