1 //===- AMDGPUPerfHintAnalysis.h ---- analysis of memory traffic -*- 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 /// \brief Analyzes if a function potentially memory bound and if a kernel 11 /// kernel may benefit from limiting number of waves to reduce cache thrashing. 12 /// 13 //===----------------------------------------------------------------------===// 14 15 #ifndef LLVM_LIB_TARGET_AMDGPU_MDGPUPERFHINTANALYSIS_H 16 #define LLVM_LIB_TARGET_AMDGPU_MDGPUPERFHINTANALYSIS_H 17 18 #include "llvm/Analysis/CallGraphSCCPass.h" 19 #include "llvm/IR/ValueMap.h" 20 21 namespace llvm { 22 23 struct AMDGPUPerfHintAnalysis : public CallGraphSCCPass { 24 static char ID; 25 26 public: 27 AMDGPUPerfHintAnalysis() : CallGraphSCCPass(ID) {} 28 29 bool runOnSCC(CallGraphSCC &SCC) override; 30 31 void getAnalysisUsage(AnalysisUsage &AU) const override { 32 AU.setPreservesAll(); 33 } 34 35 bool isMemoryBound(const Function *F) const; 36 37 bool needsWaveLimiter(const Function *F) const; 38 39 struct FuncInfo { 40 unsigned MemInstCost; 41 unsigned InstCost; 42 unsigned IAMInstCost; // Indirect access memory instruction count 43 unsigned LSMInstCost; // Large stride memory instruction count 44 FuncInfo() : MemInstCost(0), InstCost(0), IAMInstCost(0), LSMInstCost(0) {} 45 }; 46 47 typedef ValueMap<const Function*, FuncInfo> FuncInfoMap; 48 49 private: 50 51 FuncInfoMap FIM; 52 }; 53 } // namespace llvm 54 #endif // LLVM_LIB_TARGET_AMDGPU_MDGPUPERFHINTANALYSIS_H 55