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:
NVPTXMachineFunctionInfo(MachineFunction & MF)27   NVPTXMachineFunctionInfo(MachineFunction &MF) {}
28 
29   /// Returns the index for the symbol \p Symbol. If the symbol was previously,
30   /// added, the same index is returned. Otherwise, the symbol is added and the
31   /// new index is returned.
getImageHandleSymbolIndex(const char * Symbol)32   unsigned getImageHandleSymbolIndex(const char *Symbol) {
33     // Is the symbol already present?
34     for (unsigned i = 0, e = ImageHandleList.size(); i != e; ++i)
35       if (ImageHandleList[i] == std::string(Symbol))
36         return i;
37     // Nope, insert it
38     ImageHandleList.push_back(Symbol);
39     return ImageHandleList.size()-1;
40   }
41 
42   /// Returns the symbol name at the given index.
getImageHandleSymbol(unsigned Idx)43   const char *getImageHandleSymbol(unsigned Idx) const {
44     assert(ImageHandleList.size() > Idx && "Bad index");
45     return ImageHandleList[Idx].c_str();
46   }
47 };
48 }
49 
50 #endif
51