1 //===-- NVPTXMachineFunctionInfo.h - NVPTX-specific Function Info --------===// 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 class is attached to a MachineFunction instance and tracks target- 10 // dependent information 11 // 12 //===----------------------------------------------------------------------===// 13 14 #ifndef LLVM_LIB_TARGET_NVPTX_NVPTXMACHINEFUNCTIONINFO_H 15 #define LLVM_LIB_TARGET_NVPTX_NVPTXMACHINEFUNCTIONINFO_H 16 17 #include "llvm/CodeGen/MachineFunction.h" 18 19 namespace llvm { 20 class NVPTXMachineFunctionInfo : public MachineFunctionInfo { 21 private: 22 /// Stores a mapping from index to symbol name for removing image handles 23 /// on Fermi. 24 SmallVector<std::string, 8> ImageHandleList; 25 26 public: 27 NVPTXMachineFunctionInfo(MachineFunction &MF) {} 28 29 MachineFunctionInfo * 30 clone(BumpPtrAllocator &Allocator, MachineFunction &DestMF, 31 const DenseMap<MachineBasicBlock *, MachineBasicBlock *> &Src2DstMBB) 32 const override { 33 return DestMF.cloneInfo<NVPTXMachineFunctionInfo>(*this); 34 } 35 36 /// Returns the index for the symbol \p Symbol. If the symbol was previously, 37 /// added, the same index is returned. Otherwise, the symbol is added and the 38 /// new index is returned. 39 unsigned getImageHandleSymbolIndex(const char *Symbol) { 40 // Is the symbol already present? 41 for (unsigned i = 0, e = ImageHandleList.size(); i != e; ++i) 42 if (ImageHandleList[i] == std::string(Symbol)) 43 return i; 44 // Nope, insert it 45 ImageHandleList.push_back(Symbol); 46 return ImageHandleList.size()-1; 47 } 48 49 /// Returns the symbol name at the given index. 50 const char *getImageHandleSymbol(unsigned Idx) const { 51 assert(ImageHandleList.size() > Idx && "Bad index"); 52 return ImageHandleList[Idx].c_str(); 53 } 54 }; 55 } 56 57 #endif 58