Details | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
14 | pmbaty | 1 | //===- llvm/Analysis/DemandedBits.h - Determine demanded bits ---*- 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 pass implements a demanded bits analysis. A demanded bit is one that |
||
10 | // contributes to a result; bits that are not demanded can be either zero or |
||
11 | // one without affecting control or data flow. For example in this sequence: |
||
12 | // |
||
13 | // %1 = add i32 %x, %y |
||
14 | // %2 = trunc i32 %1 to i16 |
||
15 | // |
||
16 | // Only the lowest 16 bits of %1 are demanded; the rest are removed by the |
||
17 | // trunc. |
||
18 | // |
||
19 | //===----------------------------------------------------------------------===// |
||
20 | |||
21 | #ifndef LLVM_ANALYSIS_DEMANDEDBITS_H |
||
22 | #define LLVM_ANALYSIS_DEMANDEDBITS_H |
||
23 | |||
24 | #include "llvm/ADT/APInt.h" |
||
25 | #include "llvm/ADT/DenseMap.h" |
||
26 | #include "llvm/ADT/SmallPtrSet.h" |
||
27 | #include "llvm/IR/PassManager.h" |
||
28 | #include "llvm/Pass.h" |
||
29 | #include <optional> |
||
30 | |||
31 | namespace llvm { |
||
32 | |||
33 | class AssumptionCache; |
||
34 | class DominatorTree; |
||
35 | class Function; |
||
36 | class Instruction; |
||
37 | struct KnownBits; |
||
38 | class raw_ostream; |
||
39 | |||
40 | class DemandedBits { |
||
41 | public: |
||
42 | DemandedBits(Function &F, AssumptionCache &AC, DominatorTree &DT) : |
||
43 | F(F), AC(AC), DT(DT) {} |
||
44 | |||
45 | /// Return the bits demanded from instruction I. |
||
46 | /// |
||
47 | /// For vector instructions individual vector elements are not distinguished: |
||
48 | /// A bit is demanded if it is demanded for any of the vector elements. The |
||
49 | /// size of the return value corresponds to the type size in bits of the |
||
50 | /// scalar type. |
||
51 | /// |
||
52 | /// Instructions that do not have integer or vector of integer type are |
||
53 | /// accepted, but will always produce a mask with all bits set. |
||
54 | APInt getDemandedBits(Instruction *I); |
||
55 | |||
56 | /// Return the bits demanded from use U. |
||
57 | APInt getDemandedBits(Use *U); |
||
58 | |||
59 | /// Return true if, during analysis, I could not be reached. |
||
60 | bool isInstructionDead(Instruction *I); |
||
61 | |||
62 | /// Return whether this use is dead by means of not having any demanded bits. |
||
63 | bool isUseDead(Use *U); |
||
64 | |||
65 | void print(raw_ostream &OS); |
||
66 | |||
67 | /// Compute alive bits of one addition operand from alive output and known |
||
68 | /// operand bits |
||
69 | static APInt determineLiveOperandBitsAdd(unsigned OperandNo, |
||
70 | const APInt &AOut, |
||
71 | const KnownBits &LHS, |
||
72 | const KnownBits &RHS); |
||
73 | |||
74 | /// Compute alive bits of one subtraction operand from alive output and known |
||
75 | /// operand bits |
||
76 | static APInt determineLiveOperandBitsSub(unsigned OperandNo, |
||
77 | const APInt &AOut, |
||
78 | const KnownBits &LHS, |
||
79 | const KnownBits &RHS); |
||
80 | |||
81 | private: |
||
82 | void performAnalysis(); |
||
83 | void determineLiveOperandBits(const Instruction *UserI, |
||
84 | const Value *Val, unsigned OperandNo, |
||
85 | const APInt &AOut, APInt &AB, |
||
86 | KnownBits &Known, KnownBits &Known2, bool &KnownBitsComputed); |
||
87 | |||
88 | Function &F; |
||
89 | AssumptionCache &AC; |
||
90 | DominatorTree &DT; |
||
91 | |||
92 | bool Analyzed = false; |
||
93 | |||
94 | // The set of visited instructions (non-integer-typed only). |
||
95 | SmallPtrSet<Instruction*, 32> Visited; |
||
96 | DenseMap<Instruction *, APInt> AliveBits; |
||
97 | // Uses with no demanded bits. If the user also has no demanded bits, the use |
||
98 | // might not be stored explicitly in this map, to save memory during analysis. |
||
99 | SmallPtrSet<Use *, 16> DeadUses; |
||
100 | }; |
||
101 | |||
102 | class DemandedBitsWrapperPass : public FunctionPass { |
||
103 | private: |
||
104 | mutable std::optional<DemandedBits> DB; |
||
105 | |||
106 | public: |
||
107 | static char ID; // Pass identification, replacement for typeid |
||
108 | |||
109 | DemandedBitsWrapperPass(); |
||
110 | |||
111 | bool runOnFunction(Function &F) override; |
||
112 | void getAnalysisUsage(AnalysisUsage &AU) const override; |
||
113 | |||
114 | /// Clean up memory in between runs |
||
115 | void releaseMemory() override; |
||
116 | |||
117 | DemandedBits &getDemandedBits() { return *DB; } |
||
118 | |||
119 | void print(raw_ostream &OS, const Module *M) const override; |
||
120 | }; |
||
121 | |||
122 | /// An analysis that produces \c DemandedBits for a function. |
||
123 | class DemandedBitsAnalysis : public AnalysisInfoMixin<DemandedBitsAnalysis> { |
||
124 | friend AnalysisInfoMixin<DemandedBitsAnalysis>; |
||
125 | |||
126 | static AnalysisKey Key; |
||
127 | |||
128 | public: |
||
129 | /// Provide the result type for this analysis pass. |
||
130 | using Result = DemandedBits; |
||
131 | |||
132 | /// Run the analysis pass over a function and produce demanded bits |
||
133 | /// information. |
||
134 | DemandedBits run(Function &F, FunctionAnalysisManager &AM); |
||
135 | }; |
||
136 | |||
137 | /// Printer pass for DemandedBits |
||
138 | class DemandedBitsPrinterPass : public PassInfoMixin<DemandedBitsPrinterPass> { |
||
139 | raw_ostream &OS; |
||
140 | |||
141 | public: |
||
142 | explicit DemandedBitsPrinterPass(raw_ostream &OS) : OS(OS) {} |
||
143 | |||
144 | PreservedAnalyses run(Function &F, FunctionAnalysisManager &AM); |
||
145 | }; |
||
146 | |||
147 | /// Create a demanded bits analysis pass. |
||
148 | FunctionPass *createDemandedBitsWrapperPass(); |
||
149 | |||
150 | } // end namespace llvm |
||
151 | |||
152 | #endif // LLVM_ANALYSIS_DEMANDEDBITS_H |