Details | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
14 | pmbaty | 1 | //== llvm/CodeGen/GlobalISel/Localizer.h - Localizer -------------*- 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 | /// \file This file describes the interface of the Localizer pass. |
||
10 | /// This pass moves/duplicates constant-like instructions close to their uses. |
||
11 | /// Its primarily goal is to workaround the deficiencies of the fast register |
||
12 | /// allocator. |
||
13 | /// With GlobalISel constants are all materialized in the entry block of |
||
14 | /// a function. However, the fast allocator cannot rematerialize constants and |
||
15 | /// has a lot more live-ranges to deal with and will most likely end up |
||
16 | /// spilling a lot. |
||
17 | /// By pushing the constants close to their use, we only create small |
||
18 | /// live-ranges. |
||
19 | //===----------------------------------------------------------------------===// |
||
20 | |||
21 | #ifndef LLVM_CODEGEN_GLOBALISEL_LOCALIZER_H |
||
22 | #define LLVM_CODEGEN_GLOBALISEL_LOCALIZER_H |
||
23 | |||
24 | #include "llvm/ADT/SetVector.h" |
||
25 | #include "llvm/CodeGen/MachineFunctionPass.h" |
||
26 | |||
27 | namespace llvm { |
||
28 | // Forward declarations. |
||
29 | class AnalysisUsage; |
||
30 | class MachineBasicBlock; |
||
31 | class MachineInstr; |
||
32 | class MachineOperand; |
||
33 | class MachineRegisterInfo; |
||
34 | class TargetTransformInfo; |
||
35 | |||
36 | /// This pass implements the localization mechanism described at the |
||
37 | /// top of this file. One specificity of the implementation is that |
||
38 | /// it will materialize one and only one instance of a constant per |
||
39 | /// basic block, thus enabling reuse of that constant within that block. |
||
40 | /// Moreover, it only materializes constants in blocks where they |
||
41 | /// are used. PHI uses are considered happening at the end of the |
||
42 | /// related predecessor. |
||
43 | class Localizer : public MachineFunctionPass { |
||
44 | public: |
||
45 | static char ID; |
||
46 | |||
47 | private: |
||
48 | /// An input function to decide if the pass should run or not |
||
49 | /// on the given MachineFunction. |
||
50 | std::function<bool(const MachineFunction &)> DoNotRunPass; |
||
51 | |||
52 | /// MRI contains all the register class/bank information that this |
||
53 | /// pass uses and updates. |
||
54 | MachineRegisterInfo *MRI; |
||
55 | /// TTI used for getting remat costs for instructions. |
||
56 | TargetTransformInfo *TTI; |
||
57 | |||
58 | /// Check if \p MOUse is used in the same basic block as \p Def. |
||
59 | /// If the use is in the same block, we say it is local. |
||
60 | /// When the use is not local, \p InsertMBB will contain the basic |
||
61 | /// block when to insert \p Def to have a local use. |
||
62 | static bool isLocalUse(MachineOperand &MOUse, const MachineInstr &Def, |
||
63 | MachineBasicBlock *&InsertMBB); |
||
64 | |||
65 | /// Initialize the field members using \p MF. |
||
66 | void init(MachineFunction &MF); |
||
67 | |||
68 | typedef SmallSetVector<MachineInstr *, 32> LocalizedSetVecT; |
||
69 | |||
70 | /// If \p Op is a phi operand and not unique in that phi, that is, |
||
71 | /// there are other operands in the phi with the same register, |
||
72 | /// return true. |
||
73 | bool isNonUniquePhiValue(MachineOperand &Op) const; |
||
74 | |||
75 | /// Do inter-block localization from the entry block. |
||
76 | bool localizeInterBlock(MachineFunction &MF, |
||
77 | LocalizedSetVecT &LocalizedInstrs); |
||
78 | |||
79 | /// Do intra-block localization of already localized instructions. |
||
80 | bool localizeIntraBlock(LocalizedSetVecT &LocalizedInstrs); |
||
81 | |||
82 | public: |
||
83 | Localizer(); |
||
84 | Localizer(std::function<bool(const MachineFunction &)>); |
||
85 | |||
86 | StringRef getPassName() const override { return "Localizer"; } |
||
87 | |||
88 | MachineFunctionProperties getRequiredProperties() const override { |
||
89 | return MachineFunctionProperties() |
||
90 | .set(MachineFunctionProperties::Property::IsSSA); |
||
91 | } |
||
92 | |||
93 | void getAnalysisUsage(AnalysisUsage &AU) const override; |
||
94 | |||
95 | bool runOnMachineFunction(MachineFunction &MF) override; |
||
96 | }; |
||
97 | |||
98 | } // End namespace llvm. |
||
99 | |||
100 | #endif |