| 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>(Old: *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(Elt: 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 | |
|---|
