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
19namespace llvm {
20class NVPTXMachineFunctionInfo : public MachineFunctionInfo {
21private:
22  /// Stores a mapping from index to symbol name for removing image handles
23  /// on Fermi.
24  SmallVector<std::string, 8> ImageHandleList;
25
26public:
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
51