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(const Function &F, const TargetSubtargetInfo *STI) {}
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