62e4436bb8
Prior to this change NVPTXReplaceImageHandles replaced operands with indices and populated a table matching these indices to strings to be used in AsmPrinter. We can clean this up by simply inserting the correct external symbol or global address operands during NVPTXReplaceImageHandles, largely removing the need for the table.
60 lines
2.1 KiB
C++
60 lines
2.1 KiB
C++
//===-- NVPTXMachineFunctionInfo.h - NVPTX-specific Function Info --------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This class is attached to a MachineFunction instance and tracks target-
|
|
// dependent information
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_LIB_TARGET_NVPTX_NVPTXMACHINEFUNCTIONINFO_H
|
|
#define LLVM_LIB_TARGET_NVPTX_NVPTXMACHINEFUNCTIONINFO_H
|
|
|
|
#include "llvm/ADT/StringRef.h"
|
|
#include "llvm/CodeGen/MachineFunction.h"
|
|
|
|
namespace llvm {
|
|
class NVPTXMachineFunctionInfo : public MachineFunctionInfo {
|
|
private:
|
|
/// Stores a mapping from index to symbol name for image handles that are
|
|
/// replaced with image references
|
|
SmallVector<std::string, 8> ImageHandleList;
|
|
|
|
public:
|
|
NVPTXMachineFunctionInfo(const Function &F, const TargetSubtargetInfo *STI) {}
|
|
|
|
MachineFunctionInfo *
|
|
clone(BumpPtrAllocator &Allocator, MachineFunction &DestMF,
|
|
const DenseMap<MachineBasicBlock *, MachineBasicBlock *> &Src2DstMBB)
|
|
const override {
|
|
return DestMF.cloneInfo<NVPTXMachineFunctionInfo>(*this);
|
|
}
|
|
|
|
/// Returns the index for the symbol \p Symbol. If the symbol was previously,
|
|
/// added, the same index is returned. Otherwise, the symbol is added and the
|
|
/// new index is returned.
|
|
unsigned getImageHandleSymbolIndex(StringRef Symbol) {
|
|
// Is the symbol already present?
|
|
for (unsigned i = 0, e = ImageHandleList.size(); i != e; ++i)
|
|
if (ImageHandleList[i] == Symbol)
|
|
return i;
|
|
// Nope, insert it
|
|
ImageHandleList.push_back(Symbol.str());
|
|
return ImageHandleList.size()-1;
|
|
}
|
|
|
|
/// Check if the symbol has a mapping. Having a mapping means the handle is
|
|
/// replaced with a reference
|
|
bool checkImageHandleSymbol(StringRef Symbol) const {
|
|
return ImageHandleList.end() !=
|
|
std::find(ImageHandleList.begin(), ImageHandleList.end(), Symbol);
|
|
}
|
|
};
|
|
}
|
|
|
|
#endif
|