LLVM  10.0.0svn
NVPTXMachineFunctionInfo.h
Go to the documentation of this file.
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 
18 
19 namespace llvm {
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:
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
This class represents lattice values for constants.
Definition: AllocatorList.h:23
void push_back(const T &Elt)
Definition: SmallVector.h:211
unsigned getImageHandleSymbolIndex(const char *Symbol)
Returns the index for the symbol Symbol.
MachineFunctionInfo - This class can be derived from and used by targets to hold private target-speci...
NVPTXMachineFunctionInfo(MachineFunction &MF)
const char * getImageHandleSymbol(unsigned Idx) const
Returns the symbol name at the given index.
size_t size() const
Definition: SmallVector.h:52
assert(ImpDefSCC.getReg()==AMDGPU::SCC &&ImpDefSCC.isDef())