Details | Last modification | View Log | RSS feed
| Rev | Author | Line No. | Line |
|---|---|---|---|
| 14 | pmbaty | 1 | //===- lib/CodeGen/CalcSpillWeights.h ---------------------------*- 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 | #ifndef LLVM_CODEGEN_CALCSPILLWEIGHTS_H |
||
| 10 | #define LLVM_CODEGEN_CALCSPILLWEIGHTS_H |
||
| 11 | |||
| 12 | #include "llvm/CodeGen/SlotIndexes.h" |
||
| 13 | |||
| 14 | namespace llvm { |
||
| 15 | |||
| 16 | class LiveInterval; |
||
| 17 | class LiveIntervals; |
||
| 18 | class MachineBlockFrequencyInfo; |
||
| 19 | class MachineFunction; |
||
| 20 | class MachineLoopInfo; |
||
| 21 | class VirtRegMap; |
||
| 22 | |||
| 23 | /// Normalize the spill weight of a live interval |
||
| 24 | /// |
||
| 25 | /// The spill weight of a live interval is computed as: |
||
| 26 | /// |
||
| 27 | /// (sum(use freq) + sum(def freq)) / (K + size) |
||
| 28 | /// |
||
| 29 | /// @param UseDefFreq Expected number of executed use and def instructions |
||
| 30 | /// per function call. Derived from block frequencies. |
||
| 31 | /// @param Size Size of live interval as returnexd by getSize() |
||
| 32 | /// @param NumInstr Number of instructions using this live interval |
||
| 33 | static inline float normalizeSpillWeight(float UseDefFreq, unsigned Size, |
||
| 34 | unsigned NumInstr) { |
||
| 35 | // The constant 25 instructions is added to avoid depending too much on |
||
| 36 | // accidental SlotIndex gaps for small intervals. The effect is that small |
||
| 37 | // intervals have a spill weight that is mostly proportional to the number |
||
| 38 | // of uses, while large intervals get a spill weight that is closer to a use |
||
| 39 | // density. |
||
| 40 | return UseDefFreq / (Size + 25*SlotIndex::InstrDist); |
||
| 41 | } |
||
| 42 | |||
| 43 | /// Calculate auxiliary information for a virtual register such as its |
||
| 44 | /// spill weight and allocation hint. |
||
| 45 | class VirtRegAuxInfo { |
||
| 46 | MachineFunction &MF; |
||
| 47 | LiveIntervals &LIS; |
||
| 48 | const VirtRegMap &VRM; |
||
| 49 | const MachineLoopInfo &Loops; |
||
| 50 | const MachineBlockFrequencyInfo &MBFI; |
||
| 51 | |||
| 52 | /// Returns true if Reg of live interval LI is used in instruction with many |
||
| 53 | /// operands like STATEPOINT. |
||
| 54 | bool isLiveAtStatepointVarArg(LiveInterval &LI); |
||
| 55 | |||
| 56 | public: |
||
| 57 | VirtRegAuxInfo(MachineFunction &MF, LiveIntervals &LIS, |
||
| 58 | const VirtRegMap &VRM, const MachineLoopInfo &Loops, |
||
| 59 | const MachineBlockFrequencyInfo &MBFI) |
||
| 60 | : MF(MF), LIS(LIS), VRM(VRM), Loops(Loops), MBFI(MBFI) {} |
||
| 61 | |||
| 62 | virtual ~VirtRegAuxInfo() = default; |
||
| 63 | |||
| 64 | /// (re)compute li's spill weight and allocation hint. |
||
| 65 | void calculateSpillWeightAndHint(LiveInterval &LI); |
||
| 66 | |||
| 67 | /// Compute spill weights and allocation hints for all virtual register |
||
| 68 | /// live intervals. |
||
| 69 | void calculateSpillWeightsAndHints(); |
||
| 70 | |||
| 71 | /// Return the preferred allocation register for reg, given a COPY |
||
| 72 | /// instruction. |
||
| 73 | static Register copyHint(const MachineInstr *MI, unsigned Reg, |
||
| 74 | const TargetRegisterInfo &TRI, |
||
| 75 | const MachineRegisterInfo &MRI); |
||
| 76 | |||
| 77 | /// Determine if all values in LI are rematerializable. |
||
| 78 | static bool isRematerializable(const LiveInterval &LI, |
||
| 79 | const LiveIntervals &LIS, |
||
| 80 | const VirtRegMap &VRM, |
||
| 81 | const TargetInstrInfo &TII); |
||
| 82 | |||
| 83 | protected: |
||
| 84 | /// Helper function for weight calculations. |
||
| 85 | /// (Re)compute LI's spill weight and allocation hint, or, for non null |
||
| 86 | /// start and end - compute future expected spill weight of a split |
||
| 87 | /// artifact of LI that will span between start and end slot indexes. |
||
| 88 | /// \param LI The live interval for which to compute the weight. |
||
| 89 | /// \param Start The expected beginning of the split artifact. Instructions |
||
| 90 | /// before start will not affect the weight. Relevant for |
||
| 91 | /// weight calculation of future split artifact. |
||
| 92 | /// \param End The expected end of the split artifact. Instructions |
||
| 93 | /// after end will not affect the weight. Relevant for |
||
| 94 | /// weight calculation of future split artifact. |
||
| 95 | /// \return The spill weight. Returns negative weight for unspillable LI. |
||
| 96 | float weightCalcHelper(LiveInterval &LI, SlotIndex *Start = nullptr, |
||
| 97 | SlotIndex *End = nullptr); |
||
| 98 | |||
| 99 | /// Weight normalization function. |
||
| 100 | virtual float normalize(float UseDefFreq, unsigned Size, |
||
| 101 | unsigned NumInstr) { |
||
| 102 | return normalizeSpillWeight(UseDefFreq, Size, NumInstr); |
||
| 103 | } |
||
| 104 | }; |
||
| 105 | } // end namespace llvm |
||
| 106 | |||
| 107 | #endif // LLVM_CODEGEN_CALCSPILLWEIGHTS_H |