71 lines · cpp
1//===-- HeatUtils.cpp - Utility for printing heat colors --------*- 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-exception6//7//===----------------------------------------------------------------------===//8//9// Utility for printing heat colors based on heuristics or profiling10// information.11//12//===----------------------------------------------------------------------===//13 14#include "llvm/Analysis/HeatUtils.h"15#include "llvm/Analysis/BlockFrequencyInfo.h"16#include "llvm/IR/Instructions.h"17 18#include <cmath>19 20using namespace llvm;21 22static constexpr unsigned HeatSize = 100;23static constexpr char HeatPalette[HeatSize][8] = {24 "#3d50c3", "#4055c8", "#4358cb", "#465ecf", "#4961d2", "#4c66d6", "#4f69d9",25 "#536edd", "#5572df", "#5977e3", "#5b7ae5", "#5f7fe8", "#6282ea", "#6687ed",26 "#6a8bef", "#6c8ff1", "#7093f3", "#7396f5", "#779af7", "#7a9df8", "#7ea1fa",27 "#81a4fb", "#85a8fc", "#88abfd", "#8caffe", "#8fb1fe", "#93b5fe", "#96b7ff",28 "#9abbff", "#9ebeff", "#a1c0ff", "#a5c3fe", "#a7c5fe", "#abc8fd", "#aec9fc",29 "#b2ccfb", "#b5cdfa", "#b9d0f9", "#bbd1f8", "#bfd3f6", "#c1d4f4", "#c5d6f2",30 "#c7d7f0", "#cbd8ee", "#cedaeb", "#d1dae9", "#d4dbe6", "#d6dce4", "#d9dce1",31 "#dbdcde", "#dedcdb", "#e0dbd8", "#e3d9d3", "#e5d8d1", "#e8d6cc", "#ead5c9",32 "#ecd3c5", "#eed0c0", "#efcebd", "#f1ccb8", "#f2cab5", "#f3c7b1", "#f4c5ad",33 "#f5c1a9", "#f6bfa6", "#f7bca1", "#f7b99e", "#f7b599", "#f7b396", "#f7af91",34 "#f7ac8e", "#f7a889", "#f6a385", "#f5a081", "#f59c7d", "#f4987a", "#f39475",35 "#f29072", "#f08b6e", "#ef886b", "#ed8366", "#ec7f63", "#e97a5f", "#e8765c",36 "#e57058", "#e36c55", "#e16751", "#de614d", "#dc5d4a", "#d85646", "#d65244",37 "#d24b40", "#d0473d", "#cc403a", "#ca3b37", "#c53334", "#c32e31", "#be242e",38 "#bb1b2c", "#b70d28"};39 40uint64_t llvm::getNumOfCalls(const Function &CallerFunction,41 const Function &CalledFunction) {42 uint64_t Counter = 0;43 for (const User *U : CalledFunction.users())44 if (auto CI = dyn_cast<CallInst>(U))45 Counter += CI->getCaller() == &CallerFunction;46 return Counter;47}48 49uint64_t llvm::getMaxFreq(const Function &F, const BlockFrequencyInfo *BFI) {50 uint64_t MaxFreq = 0;51 for (const BasicBlock &BB : F) {52 uint64_t FreqVal = BFI->getBlockFreq(&BB).getFrequency();53 if (FreqVal >= MaxFreq)54 MaxFreq = FreqVal;55 }56 return MaxFreq;57}58 59std::string llvm::getHeatColor(uint64_t Freq, uint64_t MaxFreq) {60 if (Freq > MaxFreq)61 Freq = MaxFreq;62 double Percent = (Freq > 0) ? log2(double(Freq)) / log2(MaxFreq) : 0;63 return getHeatColor(Percent);64}65 66std::string llvm::getHeatColor(double Percent) {67 Percent = std::clamp(Percent, 0.0, 1.0);68 unsigned ColorID = unsigned(round(Percent * (HeatSize - 1.0)));69 return HeatPalette[ColorID];70}71