fe6fb910df
clang -fexperimental-relative-c++-abi-vtables might generate `@plt` and
`@gotpcrel` specifiers in data directives. The syntax is not used in
humand-written assembly code, and is not supported by GNU assembler.
Note: the `@plt` in `.word foo@plt` is different from
the legacy `call func@plt` (where `@plt` is simply ignored).
The `@plt` syntax was selected was simply due to a quirk of AsmParser:
the syntax was supported by all targets until I updated it
to be an opt-in feature in a0671758eb
RISC-V favors the `%specifier(expr)` syntax following MIPS and Sparc,
and we should follow this convention.
This PR adds support for `.word %pltpcrel(foo+offset)` and
`.word %gotpcrel(foo)`, and drops `@plt` and `@gotpcrel`.
* MCValue::SymA can no longer have a SymbolVariant. Add an assert
similar to that of AArch64ELFObjectWriter.cpp before
https://reviews.llvm.org/D81446 (see my analysis at
https://maskray.me/blog/2025-03-16-relocation-generation-in-assemblers
if intrigued)
* `jump foo@plt, x31` now has a different diagnostic.
Pull Request: https://github.com/llvm/llvm-project/pull/132569
150 lines
5.0 KiB
C++
150 lines
5.0 KiB
C++
//===-- AMDGPUAsmPrinter.h - Print AMDGPU assembly code ---------*- C++ -*-===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
/// \file
|
|
/// AMDGPU Assembly printer class.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_LIB_TARGET_AMDGPU_AMDGPUASMPRINTER_H
|
|
#define LLVM_LIB_TARGET_AMDGPU_AMDGPUASMPRINTER_H
|
|
|
|
#include "AMDGPUMCResourceInfo.h"
|
|
#include "SIProgramInfo.h"
|
|
#include "llvm/CodeGen/AsmPrinter.h"
|
|
|
|
namespace llvm {
|
|
|
|
class AMDGPUMachineFunction;
|
|
struct AMDGPUResourceUsageAnalysis;
|
|
class AMDGPUTargetStreamer;
|
|
class MCCodeEmitter;
|
|
class MCOperand;
|
|
class MCResourceInfo;
|
|
|
|
namespace AMDGPU {
|
|
struct MCKernelDescriptor;
|
|
struct AMDGPUMCKernelCodeT;
|
|
namespace HSAMD {
|
|
class MetadataStreamer;
|
|
}
|
|
} // namespace AMDGPU
|
|
|
|
class AMDGPUAsmPrinter final : public AsmPrinter {
|
|
private:
|
|
unsigned CodeObjectVersion;
|
|
void initializeTargetID(const Module &M);
|
|
|
|
AMDGPUResourceUsageAnalysis *ResourceUsage;
|
|
|
|
MCResourceInfo RI;
|
|
|
|
SIProgramInfo CurrentProgramInfo;
|
|
|
|
std::unique_ptr<AMDGPU::HSAMD::MetadataStreamer> HSAMetadataStream;
|
|
|
|
MCCodeEmitter *DumpCodeInstEmitter = nullptr;
|
|
|
|
void getSIProgramInfo(SIProgramInfo &Out, const MachineFunction &MF);
|
|
void getAmdKernelCode(AMDGPU::AMDGPUMCKernelCodeT &Out,
|
|
const SIProgramInfo &KernelInfo,
|
|
const MachineFunction &MF) const;
|
|
|
|
/// Emit register usage information so that the GPU driver
|
|
/// can correctly setup the GPU state.
|
|
void EmitProgramInfoSI(const MachineFunction &MF,
|
|
const SIProgramInfo &KernelInfo);
|
|
void EmitPALMetadata(const MachineFunction &MF,
|
|
const SIProgramInfo &KernelInfo);
|
|
void emitPALFunctionMetadata(const MachineFunction &MF);
|
|
void emitCommonFunctionComments(const MCExpr *NumVGPR, const MCExpr *NumAGPR,
|
|
const MCExpr *TotalNumVGPR,
|
|
const MCExpr *NumSGPR,
|
|
const MCExpr *ScratchSize, uint64_t CodeSize,
|
|
const AMDGPUMachineFunction *MFI);
|
|
void emitResourceUsageRemarks(const MachineFunction &MF,
|
|
const SIProgramInfo &CurrentProgramInfo,
|
|
bool isModuleEntryFunction, bool hasMAIInsts);
|
|
|
|
const MCExpr *getAmdhsaKernelCodeProperties(const MachineFunction &MF) const;
|
|
|
|
AMDGPU::MCKernelDescriptor
|
|
getAmdhsaKernelDescriptor(const MachineFunction &MF,
|
|
const SIProgramInfo &PI) const;
|
|
|
|
void initTargetStreamer(Module &M);
|
|
|
|
SmallString<128> getMCExprStr(const MCExpr *Value);
|
|
|
|
/// Attempts to replace the validation that is missed in getSIProgramInfo due
|
|
/// to MCExpr being unknown. Invoked during doFinalization such that the
|
|
/// MCResourceInfo symbols are known.
|
|
void validateMCResourceInfo(Function &F);
|
|
|
|
public:
|
|
explicit AMDGPUAsmPrinter(TargetMachine &TM,
|
|
std::unique_ptr<MCStreamer> Streamer);
|
|
|
|
StringRef getPassName() const override;
|
|
|
|
const MCSubtargetInfo* getGlobalSTI() const;
|
|
|
|
AMDGPUTargetStreamer* getTargetStreamer() const;
|
|
|
|
bool doInitialization(Module &M) override;
|
|
bool doFinalization(Module &M) override;
|
|
bool runOnMachineFunction(MachineFunction &MF) override;
|
|
|
|
/// Wrapper for MCInstLowering.lowerOperand() for the tblgen'erated
|
|
/// pseudo lowering.
|
|
bool lowerOperand(const MachineOperand &MO, MCOperand &MCOp) const;
|
|
|
|
/// Lower the specified LLVM Constant to an MCExpr.
|
|
/// The AsmPrinter::lowerConstantof does not know how to lower
|
|
/// addrspacecast, therefore they should be lowered by this function.
|
|
const MCExpr *lowerConstant(const Constant *CV, const Constant *BaseCV,
|
|
uint64_t Offset) override;
|
|
|
|
/// tblgen'erated driver function for lowering simple MI->MC pseudo
|
|
/// instructions.
|
|
bool lowerPseudoInstExpansion(const MachineInstr *MI, MCInst &Inst);
|
|
|
|
/// Implemented in AMDGPUMCInstLower.cpp
|
|
void emitInstruction(const MachineInstr *MI) override;
|
|
|
|
void emitFunctionBodyStart() override;
|
|
|
|
void emitFunctionBodyEnd() override;
|
|
|
|
void emitImplicitDef(const MachineInstr *MI) const override;
|
|
|
|
void emitFunctionEntryLabel() override;
|
|
|
|
void emitBasicBlockStart(const MachineBasicBlock &MBB) override;
|
|
|
|
void emitGlobalVariable(const GlobalVariable *GV) override;
|
|
|
|
void emitStartOfAsmFile(Module &M) override;
|
|
|
|
void emitEndOfAsmFile(Module &M) override;
|
|
|
|
bool PrintAsmOperand(const MachineInstr *MI, unsigned OpNo,
|
|
const char *ExtraCode, raw_ostream &O) override;
|
|
|
|
protected:
|
|
void getAnalysisUsage(AnalysisUsage &AU) const override;
|
|
|
|
std::vector<std::string> DisasmLines, HexLines;
|
|
size_t DisasmLineMaxLen;
|
|
bool IsTargetStreamerInitialized;
|
|
};
|
|
|
|
} // end namespace llvm
|
|
|
|
#endif // LLVM_LIB_TARGET_AMDGPU_AMDGPUASMPRINTER_H
|