lib/Target/NVPTX/NVPTXMachineFunctionInfo.h

   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   /// 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.
  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.
  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