1//===-- AMDGPUTargetStreamer.h - AMDGPU Target Streamer --------*- C++ -*--===//
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#ifndef LLVM_LIB_TARGET_AMDGPU_MCTARGETDESC_AMDGPUTARGETSTREAMER_H
10#define LLVM_LIB_TARGET_AMDGPU_MCTARGETDESC_AMDGPUTARGETSTREAMER_H
11
12#include "Utils/AMDGPUBaseInfo.h"
13#include "Utils/AMDGPUPALMetadata.h"
14#include "llvm/ADT/SmallVector.h"
15#include "llvm/MC/MCStreamer.h"
16#include <string>
17#include <utility>
18
19namespace llvm {
20
21class MCELFStreamer;
22class MCSymbol;
23class formatted_raw_ostream;
24
25namespace AMDGPU {
26
27struct AMDGPUMCKernelCodeT;
28struct MCKernelDescriptor;
29namespace HSAMD {
30struct Metadata;
31}
32
33struct FuncInfo {
34 uint32_t NumSGPR = 0;
35 uint32_t NumArchVGPR = 0;
36 uint32_t NumAccVGPR = 0;
37 uint32_t PrivateSegmentSize = 0;
38 bool UsesVCC = false;
39 bool UsesFlatScratch = false;
40 bool HasDynStack = false;
41
42 MCSymbol *Sym = nullptr;
43};
44
45struct InfoSectionData {
46 SmallVector<FuncInfo, 8> Funcs;
47 SmallVector<std::pair<MCSymbol *, MCSymbol *>, 4> Uses;
48 SmallVector<std::pair<MCSymbol *, MCSymbol *>, 8> Calls;
49 SmallVector<std::pair<MCSymbol *, std::string>, 4> IndirectCalls;
50 SmallVector<std::pair<MCSymbol *, std::string>, 4> TypeIds;
51};
52
53} // namespace AMDGPU
54
55class AMDGPUTargetStreamer : public MCTargetStreamer {
56 AMDGPUPALMetadata PALMetadata;
57
58protected:
59 // TODO: Move HSAMetadataStream to AMDGPUTargetStreamer.
60 std::optional<AMDGPU::TargetID> TargetID;
61 unsigned CodeObjectVersion;
62
63 MCContext &getContext() const { return Streamer.getContext(); }
64
65public:
66 AMDGPUTargetStreamer(MCStreamer &S)
67 : MCTargetStreamer(S),
68 // Assume the default COV for now, EmitDirectiveAMDHSACodeObjectVersion
69 // will update this if it is encountered.
70 CodeObjectVersion(AMDGPU::getDefaultAMDHSACodeObjectVersion()) {}
71
72 AMDGPUPALMetadata *getPALMetadata() { return &PALMetadata; }
73
74 virtual void EmitDirectiveAMDGCNTarget(){};
75
76 virtual void EmitDirectiveAMDHSACodeObjectVersion(unsigned COV) {
77 CodeObjectVersion = COV;
78 }
79
80 virtual void EmitAMDKernelCodeT(AMDGPU::AMDGPUMCKernelCodeT &Header) {};
81
82 virtual void EmitAMDGPUSymbolType(StringRef SymbolName, unsigned Type){};
83
84 virtual void emitAMDGPULDS(MCSymbol *Symbol, unsigned Size, Align Alignment) {
85 }
86
87 virtual void EmitMCResourceInfo(
88 const MCSymbol *NumVGPR, const MCSymbol *NumAGPR,
89 const MCSymbol *NumExplicitSGPR, const MCSymbol *NumNamedBarrier,
90 const MCSymbol *PrivateSegmentSize, const MCSymbol *UsesVCC,
91 const MCSymbol *UsesFlatScratch, const MCSymbol *HasDynamicallySizedStack,
92 const MCSymbol *HasRecursion, const MCSymbol *HasIndirectCall) {};
93
94 virtual void EmitMCResourceMaximums(const MCSymbol *MaxVGPR,
95 const MCSymbol *MaxAGPR,
96 const MCSymbol *MaxSGPR,
97 const MCSymbol *MaxNamedBarrier) {};
98
99 /// \returns True on success, false on failure.
100 virtual bool EmitISAVersion() { return true; }
101
102 /// \returns True on success, false on failure.
103 virtual bool EmitHSAMetadataV3(StringRef HSAMetadataString);
104
105 /// Emit HSA Metadata
106 ///
107 /// When \p Strict is true, known metadata elements must already be
108 /// well-typed. When \p Strict is false, known types are inferred and
109 /// the \p HSAMetadata structure is updated with the correct types.
110 ///
111 /// \returns True on success, false on failure.
112 virtual bool EmitHSAMetadata(msgpack::Document &HSAMetadata, bool Strict) {
113 return true;
114 }
115
116 /// \returns True on success, false on failure.
117 virtual bool EmitHSAMetadata(const AMDGPU::HSAMD::Metadata &HSAMetadata) {
118 return true;
119 }
120
121 /// \returns True on success, false on failure.
122 virtual bool EmitCodeEnd(const MCSubtargetInfo &STI) { return true; }
123
124 virtual void
125 EmitAmdhsaKernelDescriptor(const MCSubtargetInfo &STI, StringRef KernelName,
126 const AMDGPU::MCKernelDescriptor &KernelDescriptor,
127 const MCExpr *NextVGPR, const MCExpr *NextSGPR,
128 const MCExpr *ReserveVCC,
129 const MCExpr *ReserveFlatScr) {}
130
131 virtual void emitAMDGPUInfo(const AMDGPU::InfoSectionData &Data) {}
132
133 static unsigned getElfMach(StringRef GPU);
134
135 const std::optional<AMDGPU::TargetID> &getTargetID() const {
136 return TargetID;
137 }
138 std::optional<AMDGPU::TargetID> &getTargetID() { return TargetID; }
139 void initializeTargetID(const MCSubtargetInfo &STI,
140 bool ApplyFeatureString = false);
141};
142
143class AMDGPUTargetAsmStreamer final : public AMDGPUTargetStreamer {
144 formatted_raw_ostream &OS;
145public:
146 AMDGPUTargetAsmStreamer(MCStreamer &S, formatted_raw_ostream &OS);
147
148 void finish() override;
149
150 void EmitDirectiveAMDGCNTarget() override;
151
152 void EmitDirectiveAMDHSACodeObjectVersion(unsigned COV) override;
153
154 void EmitAMDKernelCodeT(AMDGPU::AMDGPUMCKernelCodeT &Header) override;
155
156 void EmitAMDGPUSymbolType(StringRef SymbolName, unsigned Type) override;
157
158 void emitAMDGPULDS(MCSymbol *Sym, unsigned Size, Align Alignment) override;
159
160 void EmitMCResourceInfo(
161 const MCSymbol *NumVGPR, const MCSymbol *NumAGPR,
162 const MCSymbol *NumExplicitSGPR, const MCSymbol *NumNamedBarrier,
163 const MCSymbol *PrivateSegmentSize, const MCSymbol *UsesVCC,
164 const MCSymbol *UsesFlatScratch, const MCSymbol *HasDynamicallySizedStack,
165 const MCSymbol *HasRecursion, const MCSymbol *HasIndirectCall) override;
166
167 void EmitMCResourceMaximums(const MCSymbol *MaxVGPR, const MCSymbol *MaxAGPR,
168 const MCSymbol *MaxSGPR,
169 const MCSymbol *MaxNamedBarrier) override;
170
171 /// \returns True on success, false on failure.
172 bool EmitISAVersion() override;
173
174 /// \returns True on success, false on failure.
175 bool EmitHSAMetadata(msgpack::Document &HSAMetadata, bool Strict) override;
176
177 /// \returns True on success, false on failure.
178 bool EmitCodeEnd(const MCSubtargetInfo &STI) override;
179
180 void
181 EmitAmdhsaKernelDescriptor(const MCSubtargetInfo &STI, StringRef KernelName,
182 const AMDGPU::MCKernelDescriptor &KernelDescriptor,
183 const MCExpr *NextVGPR, const MCExpr *NextSGPR,
184 const MCExpr *ReserveVCC,
185 const MCExpr *ReserveFlatScr) override;
186
187 void emitAMDGPUInfo(const AMDGPU::InfoSectionData &Data) override;
188};
189
190class AMDGPUTargetELFStreamer final : public AMDGPUTargetStreamer {
191 const MCSubtargetInfo &STI;
192 MCStreamer &Streamer;
193
194 void EmitNote(StringRef Name, const MCExpr *DescSize, unsigned NoteType,
195 function_ref<void(MCELFStreamer &)> EmitDesc);
196
197 unsigned getEFlags();
198
199 unsigned getEFlagsR600();
200 unsigned getEFlagsAMDGCN();
201
202 unsigned getEFlagsUnknownOS();
203 unsigned getEFlagsAMDHSA();
204 unsigned getEFlagsAMDPAL();
205 unsigned getEFlagsMesa3D();
206
207 unsigned getEFlagsV3();
208 unsigned getEFlagsV4();
209 unsigned getEFlagsV6();
210
211public:
212 AMDGPUTargetELFStreamer(MCStreamer &S, const MCSubtargetInfo &STI);
213
214 MCELFStreamer &getStreamer();
215
216 void finish() override;
217
218 void EmitDirectiveAMDGCNTarget() override;
219
220 void EmitAMDKernelCodeT(AMDGPU::AMDGPUMCKernelCodeT &Header) override;
221
222 void EmitAMDGPUSymbolType(StringRef SymbolName, unsigned Type) override;
223
224 void emitAMDGPULDS(MCSymbol *Sym, unsigned Size, Align Alignment) override;
225
226 /// \returns True on success, false on failure.
227 bool EmitISAVersion() override;
228
229 /// \returns True on success, false on failure.
230 bool EmitHSAMetadata(msgpack::Document &HSAMetadata, bool Strict) override;
231
232 /// \returns True on success, false on failure.
233 bool EmitCodeEnd(const MCSubtargetInfo &STI) override;
234
235 void
236 EmitAmdhsaKernelDescriptor(const MCSubtargetInfo &STI, StringRef KernelName,
237 const AMDGPU::MCKernelDescriptor &KernelDescriptor,
238 const MCExpr *NextVGPR, const MCExpr *NextSGPR,
239 const MCExpr *ReserveVCC,
240 const MCExpr *ReserveFlatScr) override;
241
242 void emitAMDGPUInfo(const AMDGPU::InfoSectionData &Data) override;
243};
244}
245#endif
246