1 //===- llvm/IR/ProfDataUtils.h - Profiling Metadata Utilities ---*- 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 10 /// This file contains the declarations for profiling metadata utility 11 /// functions. 12 // 13 //===----------------------------------------------------------------------===// 14 15 #ifndef LLVM_IR_PROFDATAUTILS_H 16 #define LLVM_IR_PROFDATAUTILS_H 17 18 #include "llvm/ADT/SmallVector.h" 19 #include "llvm/ADT/Twine.h" 20 #include "llvm/IR/Metadata.h" 21 22 namespace llvm { 23 24 /// Checks if an Instruction has MD_prof Metadata 25 bool hasProfMD(const Instruction &I); 26 27 /// Checks if an MDNode contains Branch Weight Metadata 28 bool isBranchWeightMD(const MDNode *ProfileData); 29 30 /// Checks if an instructions has Branch Weight Metadata 31 /// 32 /// \param I The instruction to check 33 /// \returns True if I has an MD_prof node containing Branch Weights. False 34 /// otherwise. 35 bool hasBranchWeightMD(const Instruction &I); 36 37 /// Checks if an instructions has valid Branch Weight Metadata 38 /// 39 /// \param I The instruction to check 40 /// \returns True if I has an MD_prof node containing valid Branch Weights, 41 /// i.e., one weight for each successor. False otherwise. 42 bool hasValidBranchWeightMD(const Instruction &I); 43 44 /// Get the branch weights metadata node 45 /// 46 /// \param I The Instruction to get the weights from. 47 /// \returns A pointer to I's branch weights metadata node, if it exists. 48 /// Nullptr otherwise. 49 MDNode *getBranchWeightMDNode(const Instruction &I); 50 51 /// Get the valid branch weights metadata node 52 /// 53 /// \param I The Instruction to get the weights from. 54 /// \returns A pointer to I's valid branch weights metadata node, if it exists. 55 /// Nullptr otherwise. 56 MDNode *getValidBranchWeightMDNode(const Instruction &I); 57 58 /// Extract branch weights from MD_prof metadata 59 /// 60 /// \param ProfileData A pointer to an MDNode. 61 /// \param [out] Weights An output vector to fill with branch weights 62 /// \returns True if weights were extracted, False otherwise. When false Weights 63 /// will be cleared. 64 bool extractBranchWeights(const MDNode *ProfileData, 65 SmallVectorImpl<uint32_t> &Weights); 66 67 /// Faster version of extractBranchWeights() that skips checks and must only 68 /// be called with "branch_weights" metadata nodes. Supports uint32_t. 69 void extractFromBranchWeightMD32(const MDNode *ProfileData, 70 SmallVectorImpl<uint32_t> &Weights); 71 72 /// Faster version of extractBranchWeights() that skips checks and must only 73 /// be called with "branch_weights" metadata nodes. Supports uint64_t. 74 void extractFromBranchWeightMD64(const MDNode *ProfileData, 75 SmallVectorImpl<uint64_t> &Weights); 76 77 /// Extract branch weights attatched to an Instruction 78 /// 79 /// \param I The Instruction to extract weights from. 80 /// \param [out] Weights An output vector to fill with branch weights 81 /// \returns True if weights were extracted, False otherwise. When false Weights 82 /// will be cleared. 83 bool extractBranchWeights(const Instruction &I, 84 SmallVectorImpl<uint32_t> &Weights); 85 86 /// Extract branch weights from a conditional branch or select Instruction. 87 /// 88 /// \param I The instruction to extract branch weights from. 89 /// \param [out] TrueVal will contain the branch weight for the True branch 90 /// \param [out] FalseVal will contain the branch weight for the False branch 91 /// \returns True on success with profile weights filled in. False if no 92 /// metadata or invalid metadata was found. 93 bool extractBranchWeights(const Instruction &I, uint64_t &TrueVal, 94 uint64_t &FalseVal); 95 96 /// Retrieve the total of all weights from MD_prof data. 97 /// 98 /// \param ProfileData The profile data to extract the total weight from 99 /// \param [out] TotalWeights input variable to fill with total weights 100 /// \returns True on success with profile total weights filled in. False if no 101 /// metadata was found. 102 bool extractProfTotalWeight(const MDNode *ProfileData, uint64_t &TotalWeights); 103 104 /// Retrieve the total of all weights from an instruction. 105 /// 106 /// \param I The instruction to extract the total weight from 107 /// \param [out] TotalWeights input variable to fill with total weights 108 /// \returns True on success with profile total weights filled in. False if no 109 /// metadata was found. 110 bool extractProfTotalWeight(const Instruction &I, uint64_t &TotalWeights); 111 112 /// Create a new `branch_weights` metadata node and add or overwrite 113 /// a `prof` metadata reference to instruction `I`. 114 void setBranchWeights(Instruction &I, ArrayRef<uint32_t> Weights); 115 116 /// Scaling the profile data attached to 'I' using the ratio of S/T. 117 void scaleProfData(Instruction &I, uint64_t S, uint64_t T); 118 119 } // namespace llvm 120 #endif 121