1//===-- AMDGPUTargetStreamer.cpp - Mips Target Streamer Methods -----------===//
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 file provides AMDGPU specific target streamer methods.
10//
11//===----------------------------------------------------------------------===//
12
13#include "AMDGPUTargetStreamer.h"
14#include "AMDGPUMCExpr.h"
15#include "AMDGPUMCKernelDescriptor.h"
16#include "AMDGPUMCTargetDesc.h"
17#include "AMDGPUPTNote.h"
18#include "Utils/AMDGPUBaseInfo.h"
19#include "Utils/AMDKernelCodeTUtils.h"
20#include "llvm/BinaryFormat/AMDGPUMetadataVerifier.h"
21#include "llvm/BinaryFormat/ELF.h"
22#include "llvm/MC/MCAsmInfo.h"
23#include "llvm/MC/MCAssembler.h"
24#include "llvm/MC/MCContext.h"
25#include "llvm/MC/MCELFObjectWriter.h"
26#include "llvm/MC/MCELFStreamer.h"
27#include "llvm/MC/MCSubtargetInfo.h"
28#include "llvm/MC/StringTableBuilder.h"
29#include "llvm/Support/AMDGPUMetadata.h"
30#include "llvm/Support/AMDGPUObjLinkingInfo.h"
31#include "llvm/Support/AMDHSAKernelDescriptor.h"
32#include "llvm/Support/CommandLine.h"
33#include "llvm/Support/FormattedStream.h"
34#include "llvm/TargetParser/AMDGPUTargetParser.h"
35
36using namespace llvm;
37using namespace llvm::AMDGPU;
38
39//===----------------------------------------------------------------------===//
40// AMDGPUTargetStreamer
41//===----------------------------------------------------------------------===//
42
43static cl::opt<unsigned>
44 ForceGenericVersion("amdgpu-force-generic-version",
45 cl::desc("Force a specific generic_v<N> flag to be "
46 "added. For testing purposes only."),
47 cl::ReallyHidden, cl::init(Val: 0));
48
49void AMDGPUTargetStreamer::initializeTargetID(const MCSubtargetInfo &STI,
50 bool ApplyFeatureString) {
51 assert(TargetID == std::nullopt && "TargetID can only be initialized once");
52 // Apply xnack/sramecc from subtarget features only in MC contexts
53 // (assembler), not in codegen where they come from module flags
54 TargetID = AMDGPU::createAMDGPUTargetID(
55 STI, FeatureString: ApplyFeatureString ? STI.getFeatureString() : "");
56}
57
58bool AMDGPUTargetStreamer::EmitHSAMetadataV3(StringRef HSAMetadataString) {
59 msgpack::Document HSAMetadataDoc;
60 if (!HSAMetadataDoc.fromYAML(S: HSAMetadataString))
61 return false;
62 return EmitHSAMetadata(HSAMetadata&: HSAMetadataDoc, Strict: false);
63}
64
65StringRef AMDGPUTargetStreamer::getArchNameFromElfMach(unsigned ElfMach) {
66 AMDGPU::GPUKind AK;
67
68 // clang-format off
69 switch (ElfMach) {
70 case ELF::EF_AMDGPU_MACH_R600_R600: AK = GK_R600; break;
71 case ELF::EF_AMDGPU_MACH_R600_R630: AK = GK_R630; break;
72 case ELF::EF_AMDGPU_MACH_R600_RS880: AK = GK_RS880; break;
73 case ELF::EF_AMDGPU_MACH_R600_RV670: AK = GK_RV670; break;
74 case ELF::EF_AMDGPU_MACH_R600_RV710: AK = GK_RV710; break;
75 case ELF::EF_AMDGPU_MACH_R600_RV730: AK = GK_RV730; break;
76 case ELF::EF_AMDGPU_MACH_R600_RV770: AK = GK_RV770; break;
77 case ELF::EF_AMDGPU_MACH_R600_CEDAR: AK = GK_CEDAR; break;
78 case ELF::EF_AMDGPU_MACH_R600_CYPRESS: AK = GK_CYPRESS; break;
79 case ELF::EF_AMDGPU_MACH_R600_JUNIPER: AK = GK_JUNIPER; break;
80 case ELF::EF_AMDGPU_MACH_R600_REDWOOD: AK = GK_REDWOOD; break;
81 case ELF::EF_AMDGPU_MACH_R600_SUMO: AK = GK_SUMO; break;
82 case ELF::EF_AMDGPU_MACH_R600_BARTS: AK = GK_BARTS; break;
83 case ELF::EF_AMDGPU_MACH_R600_CAICOS: AK = GK_CAICOS; break;
84 case ELF::EF_AMDGPU_MACH_R600_CAYMAN: AK = GK_CAYMAN; break;
85 case ELF::EF_AMDGPU_MACH_R600_TURKS: AK = GK_TURKS; break;
86 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX600: AK = GK_GFX600; break;
87 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX601: AK = GK_GFX601; break;
88 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX602: AK = GK_GFX602; break;
89 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX700: AK = GK_GFX700; break;
90 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX701: AK = GK_GFX701; break;
91 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX702: AK = GK_GFX702; break;
92 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX703: AK = GK_GFX703; break;
93 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX704: AK = GK_GFX704; break;
94 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX705: AK = GK_GFX705; break;
95 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX801: AK = GK_GFX801; break;
96 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX802: AK = GK_GFX802; break;
97 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX803: AK = GK_GFX803; break;
98 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX805: AK = GK_GFX805; break;
99 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX810: AK = GK_GFX810; break;
100 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX900: AK = GK_GFX900; break;
101 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX902: AK = GK_GFX902; break;
102 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX904: AK = GK_GFX904; break;
103 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX906: AK = GK_GFX906; break;
104 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX908: AK = GK_GFX908; break;
105 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX909: AK = GK_GFX909; break;
106 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX90A: AK = GK_GFX90A; break;
107 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX90C: AK = GK_GFX90C; break;
108 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX942: AK = GK_GFX942; break;
109 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX950: AK = GK_GFX950; break;
110 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1010: AK = GK_GFX1010; break;
111 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1011: AK = GK_GFX1011; break;
112 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1012: AK = GK_GFX1012; break;
113 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1013: AK = GK_GFX1013; break;
114 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1030: AK = GK_GFX1030; break;
115 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1031: AK = GK_GFX1031; break;
116 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1032: AK = GK_GFX1032; break;
117 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1033: AK = GK_GFX1033; break;
118 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1034: AK = GK_GFX1034; break;
119 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1035: AK = GK_GFX1035; break;
120 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1036: AK = GK_GFX1036; break;
121 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1100: AK = GK_GFX1100; break;
122 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1101: AK = GK_GFX1101; break;
123 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1102: AK = GK_GFX1102; break;
124 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1103: AK = GK_GFX1103; break;
125 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1150: AK = GK_GFX1150; break;
126 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1151: AK = GK_GFX1151; break;
127 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1152: AK = GK_GFX1152; break;
128 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1153: AK = GK_GFX1153; break;
129 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1154: AK = GK_GFX1154; break;
130 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1170: AK = GK_GFX1170; break;
131 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1171: AK = GK_GFX1171; break;
132 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1172: AK = GK_GFX1172; break;
133 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1200: AK = GK_GFX1200; break;
134 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1201: AK = GK_GFX1201; break;
135 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1250: AK = GK_GFX1250; break;
136 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1251: AK = GK_GFX1251; break;
137 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX1310: AK = GK_GFX1310; break;
138 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX9_GENERIC: AK = GK_GFX9_GENERIC; break;
139 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX9_4_GENERIC: AK = GK_GFX9_4_GENERIC; break;
140 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX10_1_GENERIC: AK = GK_GFX10_1_GENERIC; break;
141 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX10_3_GENERIC: AK = GK_GFX10_3_GENERIC; break;
142 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX11_GENERIC: AK = GK_GFX11_GENERIC; break;
143 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX11_7_GENERIC: AK = GK_GFX11_7_GENERIC; break;
144 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX12_GENERIC: AK = GK_GFX12_GENERIC; break;
145 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX12_5_GENERIC: AK = GK_GFX12_5_GENERIC; break;
146 case ELF::EF_AMDGPU_MACH_AMDGCN_GFX13_GENERIC: AK = GK_GFX13_GENERIC; break;
147 case ELF::EF_AMDGPU_MACH_NONE: AK = GK_NONE; break;
148 default: AK = GK_NONE; break;
149 }
150 // clang-format on
151
152 StringRef GPUName = getArchNameAMDGCN(AK);
153 if (GPUName != "")
154 return GPUName;
155 return getArchNameR600(AK);
156}
157
158unsigned AMDGPUTargetStreamer::getElfMach(StringRef GPU) {
159 AMDGPU::GPUKind AK = parseArchAMDGCN(CPU: GPU);
160 if (AK == AMDGPU::GPUKind::GK_NONE)
161 AK = parseArchR600(CPU: GPU);
162
163 // clang-format off
164 switch (AK) {
165 case GK_R600: return ELF::EF_AMDGPU_MACH_R600_R600;
166 case GK_R630: return ELF::EF_AMDGPU_MACH_R600_R630;
167 case GK_RS880: return ELF::EF_AMDGPU_MACH_R600_RS880;
168 case GK_RV670: return ELF::EF_AMDGPU_MACH_R600_RV670;
169 case GK_RV710: return ELF::EF_AMDGPU_MACH_R600_RV710;
170 case GK_RV730: return ELF::EF_AMDGPU_MACH_R600_RV730;
171 case GK_RV770: return ELF::EF_AMDGPU_MACH_R600_RV770;
172 case GK_CEDAR: return ELF::EF_AMDGPU_MACH_R600_CEDAR;
173 case GK_CYPRESS: return ELF::EF_AMDGPU_MACH_R600_CYPRESS;
174 case GK_JUNIPER: return ELF::EF_AMDGPU_MACH_R600_JUNIPER;
175 case GK_REDWOOD: return ELF::EF_AMDGPU_MACH_R600_REDWOOD;
176 case GK_SUMO: return ELF::EF_AMDGPU_MACH_R600_SUMO;
177 case GK_BARTS: return ELF::EF_AMDGPU_MACH_R600_BARTS;
178 case GK_CAICOS: return ELF::EF_AMDGPU_MACH_R600_CAICOS;
179 case GK_CAYMAN: return ELF::EF_AMDGPU_MACH_R600_CAYMAN;
180 case GK_TURKS: return ELF::EF_AMDGPU_MACH_R600_TURKS;
181 case GK_GFX600: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX600;
182 case GK_GFX601: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX601;
183 case GK_GFX602: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX602;
184 case GK_GFX700: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX700;
185 case GK_GFX701: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX701;
186 case GK_GFX702: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX702;
187 case GK_GFX703: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX703;
188 case GK_GFX704: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX704;
189 case GK_GFX705: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX705;
190 case GK_GFX801: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX801;
191 case GK_GFX802: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX802;
192 case GK_GFX803: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX803;
193 case GK_GFX805: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX805;
194 case GK_GFX810: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX810;
195 case GK_GFX900: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX900;
196 case GK_GFX902: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX902;
197 case GK_GFX904: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX904;
198 case GK_GFX906: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX906;
199 case GK_GFX908: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX908;
200 case GK_GFX909: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX909;
201 case GK_GFX90A: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX90A;
202 case GK_GFX90C: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX90C;
203 case GK_GFX942: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX942;
204 case GK_GFX950: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX950;
205 case GK_GFX1010: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX1010;
206 case GK_GFX1011: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX1011;
207 case GK_GFX1012: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX1012;
208 case GK_GFX1013: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX1013;
209 case GK_GFX1030: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX1030;
210 case GK_GFX1031: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX1031;
211 case GK_GFX1032: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX1032;
212 case GK_GFX1033: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX1033;
213 case GK_GFX1034: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX1034;
214 case GK_GFX1035: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX1035;
215 case GK_GFX1036: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX1036;
216 case GK_GFX1100: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX1100;
217 case GK_GFX1101: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX1101;
218 case GK_GFX1102: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX1102;
219 case GK_GFX1103: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX1103;
220 case GK_GFX1150: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX1150;
221 case GK_GFX1151: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX1151;
222 case GK_GFX1152: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX1152;
223 case GK_GFX1153: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX1153;
224 case GK_GFX1154: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX1154;
225 case GK_GFX1170: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX1170;
226 case GK_GFX1171: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX1171;
227 case GK_GFX1172: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX1172;
228 case GK_GFX1200: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX1200;
229 case GK_GFX1201: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX1201;
230 case GK_GFX1250: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX1250;
231 case GK_GFX1251: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX1251;
232 case GK_GFX1310: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX1310;
233 case GK_GFX9_GENERIC: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX9_GENERIC;
234 case GK_GFX9_4_GENERIC: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX9_4_GENERIC;
235 case GK_GFX10_1_GENERIC: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX10_1_GENERIC;
236 case GK_GFX10_3_GENERIC: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX10_3_GENERIC;
237 case GK_GFX11_GENERIC: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX11_GENERIC;
238 case GK_GFX11_7_GENERIC: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX11_7_GENERIC;
239 case GK_GFX12_GENERIC: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX12_GENERIC;
240 case GK_GFX12_5_GENERIC: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX12_5_GENERIC;
241 case GK_GFX13_GENERIC: return ELF::EF_AMDGPU_MACH_AMDGCN_GFX13_GENERIC;
242 case GK_NONE: return ELF::EF_AMDGPU_MACH_NONE;
243 }
244 // clang-format on
245
246 llvm_unreachable("unknown GPU");
247}
248
249//===----------------------------------------------------------------------===//
250// AMDGPUTargetAsmStreamer
251//===----------------------------------------------------------------------===//
252
253AMDGPUTargetAsmStreamer::AMDGPUTargetAsmStreamer(MCStreamer &S,
254 formatted_raw_ostream &OS)
255 : AMDGPUTargetStreamer(S), OS(OS) {}
256
257// A hook for emitting stuff at the end.
258// We use it for emitting the accumulated PAL metadata as directives.
259// The PAL metadata is reset after it is emitted.
260void AMDGPUTargetAsmStreamer::finish() {
261 std::string S;
262 getPALMetadata()->toString(S);
263 OS << S;
264
265 // Reset the pal metadata so its data will not affect a compilation that
266 // reuses this object.
267 getPALMetadata()->reset();
268}
269
270void AMDGPUTargetAsmStreamer::EmitDirectiveAMDGCNTarget() {
271 OS << "\t.amdgcn_target \"" << *getTargetID() << "\"\n";
272}
273
274void AMDGPUTargetAsmStreamer::EmitDirectiveAMDHSACodeObjectVersion(
275 unsigned COV) {
276 AMDGPUTargetStreamer::EmitDirectiveAMDHSACodeObjectVersion(COV);
277 OS << "\t.amdhsa_code_object_version " << COV << '\n';
278}
279
280void AMDGPUTargetAsmStreamer::EmitAMDKernelCodeT(AMDGPUMCKernelCodeT &Header) {
281 auto FoldAndPrint = [&](const MCExpr *Expr, raw_ostream &OS,
282 const MCAsmInfo *MAI) {
283 printAMDGPUMCExpr(Expr: foldAMDGPUMCExpr(Expr, Ctx&: getContext()), OS, MAI);
284 };
285
286 OS << "\t.amd_kernel_code_t\n";
287 Header.EmitKernelCodeT(OS, Ctx&: getContext(), Helper: FoldAndPrint);
288 OS << "\t.end_amd_kernel_code_t\n";
289}
290
291void AMDGPUTargetAsmStreamer::EmitAMDGPUSymbolType(StringRef SymbolName,
292 unsigned Type) {
293 switch (Type) {
294 default:
295 llvm_unreachable("Invalid AMDGPU symbol type");
296 case ELF::STT_AMDGPU_HSA_KERNEL:
297 OS << "\t.amdgpu_hsa_kernel " << SymbolName << '\n';
298 break;
299 }
300}
301
302void AMDGPUTargetAsmStreamer::emitAMDGPULDS(MCSymbol *Symbol, unsigned Size,
303 Align Alignment) {
304 OS << "\t.amdgpu_lds " << Symbol->getName() << ", " << Size << ", "
305 << Alignment.value() << '\n';
306}
307
308void AMDGPUTargetAsmStreamer::EmitMCResourceInfo(
309 const MCSymbol *NumVGPR, const MCSymbol *NumAGPR,
310 const MCSymbol *NumExplicitSGPR, const MCSymbol *NumNamedBarrier,
311 const MCSymbol *PrivateSegmentSize, const MCSymbol *UsesVCC,
312 const MCSymbol *UsesFlatScratch, const MCSymbol *HasDynamicallySizedStack,
313 const MCSymbol *HasRecursion, const MCSymbol *HasIndirectCall) {
314#define PRINT_RES_INFO(ARG) \
315 OS << "\t.set "; \
316 ARG->print(OS, &getContext().getAsmInfo()); \
317 OS << ", "; \
318 getContext().getAsmInfo().printExpr(OS, *ARG->getVariableValue()); \
319 Streamer.addBlankLine();
320
321 PRINT_RES_INFO(NumVGPR);
322 PRINT_RES_INFO(NumAGPR);
323 PRINT_RES_INFO(NumExplicitSGPR);
324 PRINT_RES_INFO(NumNamedBarrier);
325 PRINT_RES_INFO(PrivateSegmentSize);
326 PRINT_RES_INFO(UsesVCC);
327 PRINT_RES_INFO(UsesFlatScratch);
328 PRINT_RES_INFO(HasDynamicallySizedStack);
329 PRINT_RES_INFO(HasRecursion);
330 PRINT_RES_INFO(HasIndirectCall);
331#undef PRINT_RES_INFO
332}
333
334void AMDGPUTargetAsmStreamer::EmitMCResourceMaximums(
335 const MCSymbol *MaxVGPR, const MCSymbol *MaxAGPR, const MCSymbol *MaxSGPR,
336 const MCSymbol *MaxNamedBarrier) {
337#define PRINT_RES_INFO(ARG) \
338 OS << "\t.set "; \
339 ARG->print(OS, &getContext().getAsmInfo()); \
340 OS << ", "; \
341 getContext().getAsmInfo().printExpr(OS, *ARG->getVariableValue()); \
342 Streamer.addBlankLine();
343
344 PRINT_RES_INFO(MaxVGPR);
345 PRINT_RES_INFO(MaxAGPR);
346 PRINT_RES_INFO(MaxSGPR);
347 PRINT_RES_INFO(MaxNamedBarrier);
348#undef PRINT_RES_INFO
349}
350
351bool AMDGPUTargetAsmStreamer::EmitISAVersion() {
352 OS << "\t.amd_amdgpu_isa \"" << getTargetID() << "\"\n";
353 return true;
354}
355
356bool AMDGPUTargetAsmStreamer::EmitHSAMetadata(msgpack::Document &HSAMetadataDoc,
357 bool Strict) {
358 HSAMD::V3::MetadataVerifier Verifier(Strict);
359 if (!Verifier.verify(HSAMetadataRoot&: HSAMetadataDoc.getRoot()))
360 return false;
361
362 std::string HSAMetadataString;
363 raw_string_ostream StrOS(HSAMetadataString);
364 HSAMetadataDoc.toYAML(OS&: StrOS);
365
366 OS << '\t' << HSAMD::V3::AssemblerDirectiveBegin << '\n';
367 OS << StrOS.str() << '\n';
368 OS << '\t' << HSAMD::V3::AssemblerDirectiveEnd << '\n';
369 return true;
370}
371
372bool AMDGPUTargetAsmStreamer::EmitCodeEnd(const MCSubtargetInfo &STI) {
373 const uint32_t Encoded_s_code_end = 0xbf9f0000;
374 const uint32_t Encoded_s_nop = 0xbf800000;
375 uint32_t Encoded_pad = Encoded_s_code_end;
376
377 // Instruction cache line size in bytes.
378 const unsigned Log2CacheLineSize = AMDGPU::isGFX11Plus(STI) ? 7 : 6;
379 const unsigned CacheLineSize = 1u << Log2CacheLineSize;
380
381 // Extra padding amount in bytes to support prefetch mode 3.
382 unsigned FillSize = 3 * CacheLineSize;
383
384 if (AMDGPU::isGFX90A(STI)) {
385 Encoded_pad = Encoded_s_nop;
386 FillSize = 16 * CacheLineSize;
387 }
388
389 OS << "\t.p2alignl " << Log2CacheLineSize << ", " << Encoded_pad << '\n';
390 OS << "\t.fill " << (FillSize / 4) << ", 4, " << Encoded_pad << '\n';
391 return true;
392}
393
394void AMDGPUTargetAsmStreamer::EmitAmdhsaKernelDescriptor(
395 const MCSubtargetInfo &STI, StringRef KernelName,
396 const MCKernelDescriptor &KD, const MCExpr *NextVGPR,
397 const MCExpr *NextSGPR, const MCExpr *ReserveVCC,
398 const MCExpr *ReserveFlatScr) {
399 IsaVersion IVersion = getIsaVersion(GPU: STI.getCPU());
400 const MCAsmInfo &MAI = getContext().getAsmInfo();
401
402 OS << "\t.amdhsa_kernel " << KernelName << '\n';
403
404 auto PrintField = [&](const MCExpr *Expr, uint32_t Shift, uint32_t Mask,
405 StringRef Directive) {
406 OS << "\t\t" << Directive << ' ';
407 const MCExpr *ShiftedAndMaskedExpr =
408 MCKernelDescriptor::bits_get(Src: Expr, Shift, Mask, Ctx&: getContext());
409 const MCExpr *New = foldAMDGPUMCExpr(Expr: ShiftedAndMaskedExpr, Ctx&: getContext());
410 printAMDGPUMCExpr(Expr: New, OS, MAI: &MAI);
411 OS << '\n';
412 };
413
414 auto EmitMCExpr = [&](const MCExpr *Value) {
415 const MCExpr *NewExpr = foldAMDGPUMCExpr(Expr: Value, Ctx&: getContext());
416 printAMDGPUMCExpr(Expr: NewExpr, OS, MAI: &MAI);
417 };
418
419 OS << "\t\t.amdhsa_group_segment_fixed_size ";
420 EmitMCExpr(KD.group_segment_fixed_size);
421 OS << '\n';
422
423 OS << "\t\t.amdhsa_private_segment_fixed_size ";
424 EmitMCExpr(KD.private_segment_fixed_size);
425 OS << '\n';
426
427 OS << "\t\t.amdhsa_kernarg_size ";
428 EmitMCExpr(KD.kernarg_size);
429 OS << '\n';
430
431 if (isGFX1250Plus(STI)) {
432 PrintField(KD.compute_pgm_rsrc2,
433 amdhsa::COMPUTE_PGM_RSRC2_GFX125_USER_SGPR_COUNT_SHIFT,
434 amdhsa::COMPUTE_PGM_RSRC2_GFX125_USER_SGPR_COUNT,
435 ".amdhsa_user_sgpr_count");
436 } else {
437 PrintField(KD.compute_pgm_rsrc2,
438 amdhsa::COMPUTE_PGM_RSRC2_GFX6_GFX120_USER_SGPR_COUNT_SHIFT,
439 amdhsa::COMPUTE_PGM_RSRC2_GFX6_GFX120_USER_SGPR_COUNT,
440 ".amdhsa_user_sgpr_count");
441 }
442
443 if (!hasArchitectedFlatScratch(STI))
444 PrintField(
445 KD.kernel_code_properties,
446 amdhsa::KERNEL_CODE_PROPERTY_ENABLE_SGPR_PRIVATE_SEGMENT_BUFFER_SHIFT,
447 amdhsa::KERNEL_CODE_PROPERTY_ENABLE_SGPR_PRIVATE_SEGMENT_BUFFER,
448 ".amdhsa_user_sgpr_private_segment_buffer");
449 PrintField(KD.kernel_code_properties,
450 amdhsa::KERNEL_CODE_PROPERTY_ENABLE_SGPR_DISPATCH_PTR_SHIFT,
451 amdhsa::KERNEL_CODE_PROPERTY_ENABLE_SGPR_DISPATCH_PTR,
452 ".amdhsa_user_sgpr_dispatch_ptr");
453 PrintField(KD.kernel_code_properties,
454 amdhsa::KERNEL_CODE_PROPERTY_ENABLE_SGPR_QUEUE_PTR_SHIFT,
455 amdhsa::KERNEL_CODE_PROPERTY_ENABLE_SGPR_QUEUE_PTR,
456 ".amdhsa_user_sgpr_queue_ptr");
457 PrintField(KD.kernel_code_properties,
458 amdhsa::KERNEL_CODE_PROPERTY_ENABLE_SGPR_KERNARG_SEGMENT_PTR_SHIFT,
459 amdhsa::KERNEL_CODE_PROPERTY_ENABLE_SGPR_KERNARG_SEGMENT_PTR,
460 ".amdhsa_user_sgpr_kernarg_segment_ptr");
461 PrintField(KD.kernel_code_properties,
462 amdhsa::KERNEL_CODE_PROPERTY_ENABLE_SGPR_DISPATCH_ID_SHIFT,
463 amdhsa::KERNEL_CODE_PROPERTY_ENABLE_SGPR_DISPATCH_ID,
464 ".amdhsa_user_sgpr_dispatch_id");
465 if (!hasArchitectedFlatScratch(STI))
466 PrintField(KD.kernel_code_properties,
467 amdhsa::KERNEL_CODE_PROPERTY_ENABLE_SGPR_FLAT_SCRATCH_INIT_SHIFT,
468 amdhsa::KERNEL_CODE_PROPERTY_ENABLE_SGPR_FLAT_SCRATCH_INIT,
469 ".amdhsa_user_sgpr_flat_scratch_init");
470 if (hasKernargPreload(STI)) {
471 PrintField(KD.kernarg_preload, amdhsa::KERNARG_PRELOAD_SPEC_LENGTH_SHIFT,
472 amdhsa::KERNARG_PRELOAD_SPEC_LENGTH,
473 ".amdhsa_user_sgpr_kernarg_preload_length");
474 PrintField(KD.kernarg_preload, amdhsa::KERNARG_PRELOAD_SPEC_OFFSET_SHIFT,
475 amdhsa::KERNARG_PRELOAD_SPEC_OFFSET,
476 ".amdhsa_user_sgpr_kernarg_preload_offset");
477 }
478 PrintField(
479 KD.kernel_code_properties,
480 amdhsa::KERNEL_CODE_PROPERTY_ENABLE_SGPR_PRIVATE_SEGMENT_SIZE_SHIFT,
481 amdhsa::KERNEL_CODE_PROPERTY_ENABLE_SGPR_PRIVATE_SEGMENT_SIZE,
482 ".amdhsa_user_sgpr_private_segment_size");
483 if (IVersion.Major >= 10)
484 PrintField(KD.kernel_code_properties,
485 amdhsa::KERNEL_CODE_PROPERTY_ENABLE_WAVEFRONT_SIZE32_SHIFT,
486 amdhsa::KERNEL_CODE_PROPERTY_ENABLE_WAVEFRONT_SIZE32,
487 ".amdhsa_wavefront_size32");
488 if (CodeObjectVersion >= AMDGPU::AMDHSA_COV5)
489 PrintField(KD.kernel_code_properties,
490 amdhsa::KERNEL_CODE_PROPERTY_USES_DYNAMIC_STACK_SHIFT,
491 amdhsa::KERNEL_CODE_PROPERTY_USES_DYNAMIC_STACK,
492 ".amdhsa_uses_dynamic_stack");
493 PrintField(KD.compute_pgm_rsrc2,
494 amdhsa::COMPUTE_PGM_RSRC2_ENABLE_PRIVATE_SEGMENT_SHIFT,
495 amdhsa::COMPUTE_PGM_RSRC2_ENABLE_PRIVATE_SEGMENT,
496 (hasArchitectedFlatScratch(STI)
497 ? ".amdhsa_enable_private_segment"
498 : ".amdhsa_system_sgpr_private_segment_wavefront_offset"));
499 PrintField(KD.compute_pgm_rsrc2,
500 amdhsa::COMPUTE_PGM_RSRC2_ENABLE_SGPR_WORKGROUP_ID_X_SHIFT,
501 amdhsa::COMPUTE_PGM_RSRC2_ENABLE_SGPR_WORKGROUP_ID_X,
502 ".amdhsa_system_sgpr_workgroup_id_x");
503 PrintField(KD.compute_pgm_rsrc2,
504 amdhsa::COMPUTE_PGM_RSRC2_ENABLE_SGPR_WORKGROUP_ID_Y_SHIFT,
505 amdhsa::COMPUTE_PGM_RSRC2_ENABLE_SGPR_WORKGROUP_ID_Y,
506 ".amdhsa_system_sgpr_workgroup_id_y");
507 PrintField(KD.compute_pgm_rsrc2,
508 amdhsa::COMPUTE_PGM_RSRC2_ENABLE_SGPR_WORKGROUP_ID_Z_SHIFT,
509 amdhsa::COMPUTE_PGM_RSRC2_ENABLE_SGPR_WORKGROUP_ID_Z,
510 ".amdhsa_system_sgpr_workgroup_id_z");
511 PrintField(KD.compute_pgm_rsrc2,
512 amdhsa::COMPUTE_PGM_RSRC2_ENABLE_SGPR_WORKGROUP_INFO_SHIFT,
513 amdhsa::COMPUTE_PGM_RSRC2_ENABLE_SGPR_WORKGROUP_INFO,
514 ".amdhsa_system_sgpr_workgroup_info");
515 PrintField(KD.compute_pgm_rsrc2,
516 amdhsa::COMPUTE_PGM_RSRC2_ENABLE_VGPR_WORKITEM_ID_SHIFT,
517 amdhsa::COMPUTE_PGM_RSRC2_ENABLE_VGPR_WORKITEM_ID,
518 ".amdhsa_system_vgpr_workitem_id");
519
520 // These directives are required.
521 OS << "\t\t.amdhsa_next_free_vgpr ";
522 EmitMCExpr(NextVGPR);
523 OS << '\n';
524
525 OS << "\t\t.amdhsa_next_free_sgpr ";
526 EmitMCExpr(NextSGPR);
527 OS << '\n';
528
529 if (AMDGPU::isGFX90A(STI)) {
530 // MCExpr equivalent of taking the (accum_offset + 1) * 4.
531 const MCExpr *accum_bits = MCKernelDescriptor::bits_get(
532 Src: KD.compute_pgm_rsrc3,
533 Shift: amdhsa::COMPUTE_PGM_RSRC3_GFX90A_ACCUM_OFFSET_SHIFT,
534 Mask: amdhsa::COMPUTE_PGM_RSRC3_GFX90A_ACCUM_OFFSET, Ctx&: getContext());
535 accum_bits = MCBinaryExpr::createAdd(
536 LHS: accum_bits, RHS: MCConstantExpr::create(Value: 1, Ctx&: getContext()), Ctx&: getContext());
537 accum_bits = MCBinaryExpr::createMul(
538 LHS: accum_bits, RHS: MCConstantExpr::create(Value: 4, Ctx&: getContext()), Ctx&: getContext());
539 OS << "\t\t.amdhsa_accum_offset ";
540 const MCExpr *New = foldAMDGPUMCExpr(Expr: accum_bits, Ctx&: getContext());
541 printAMDGPUMCExpr(Expr: New, OS, MAI: &MAI);
542 OS << '\n';
543 }
544
545 if (isGFX1250Plus(STI))
546 PrintField(KD.compute_pgm_rsrc3,
547 amdhsa::COMPUTE_PGM_RSRC3_GFX125_NAMED_BAR_CNT_SHIFT,
548 amdhsa::COMPUTE_PGM_RSRC3_GFX125_NAMED_BAR_CNT,
549 ".amdhsa_named_barrier_count");
550
551 OS << "\t\t.amdhsa_reserve_vcc ";
552 EmitMCExpr(ReserveVCC);
553 OS << '\n';
554
555 if (IVersion.Major >= 7 && !hasArchitectedFlatScratch(STI)) {
556 OS << "\t\t.amdhsa_reserve_flat_scratch ";
557 EmitMCExpr(ReserveFlatScr);
558 OS << '\n';
559 }
560
561 switch (CodeObjectVersion) {
562 default:
563 break;
564 case AMDGPU::AMDHSA_COV4:
565 case AMDGPU::AMDHSA_COV5:
566 if (STI.hasFeature(Feature: AMDGPU::FeatureSupportsXNACK)) {
567 bool XnackOn = getTargetID()->isXnackOnOrAny() ||
568 STI.hasFeature(Feature: AMDGPU::FeatureXNACK);
569 OS << "\t\t.amdhsa_reserve_xnack_mask " << XnackOn << '\n';
570 }
571 break;
572 }
573
574 PrintField(KD.compute_pgm_rsrc1,
575 amdhsa::COMPUTE_PGM_RSRC1_FLOAT_ROUND_MODE_32_SHIFT,
576 amdhsa::COMPUTE_PGM_RSRC1_FLOAT_ROUND_MODE_32,
577 ".amdhsa_float_round_mode_32");
578 PrintField(KD.compute_pgm_rsrc1,
579 amdhsa::COMPUTE_PGM_RSRC1_FLOAT_ROUND_MODE_16_64_SHIFT,
580 amdhsa::COMPUTE_PGM_RSRC1_FLOAT_ROUND_MODE_16_64,
581 ".amdhsa_float_round_mode_16_64");
582 PrintField(KD.compute_pgm_rsrc1,
583 amdhsa::COMPUTE_PGM_RSRC1_FLOAT_DENORM_MODE_32_SHIFT,
584 amdhsa::COMPUTE_PGM_RSRC1_FLOAT_DENORM_MODE_32,
585 ".amdhsa_float_denorm_mode_32");
586 PrintField(KD.compute_pgm_rsrc1,
587 amdhsa::COMPUTE_PGM_RSRC1_FLOAT_DENORM_MODE_16_64_SHIFT,
588 amdhsa::COMPUTE_PGM_RSRC1_FLOAT_DENORM_MODE_16_64,
589 ".amdhsa_float_denorm_mode_16_64");
590 if (STI.hasFeature(Feature: AMDGPU::FeatureDX10ClampAndIEEEMode)) {
591 PrintField(KD.compute_pgm_rsrc1,
592 amdhsa::COMPUTE_PGM_RSRC1_GFX6_GFX11_ENABLE_DX10_CLAMP_SHIFT,
593 amdhsa::COMPUTE_PGM_RSRC1_GFX6_GFX11_ENABLE_DX10_CLAMP,
594 ".amdhsa_dx10_clamp");
595 PrintField(KD.compute_pgm_rsrc1,
596 amdhsa::COMPUTE_PGM_RSRC1_GFX6_GFX11_ENABLE_IEEE_MODE_SHIFT,
597 amdhsa::COMPUTE_PGM_RSRC1_GFX6_GFX11_ENABLE_IEEE_MODE,
598 ".amdhsa_ieee_mode");
599 }
600 if (IVersion.Major >= 9) {
601 PrintField(KD.compute_pgm_rsrc1,
602 amdhsa::COMPUTE_PGM_RSRC1_GFX9_PLUS_FP16_OVFL_SHIFT,
603 amdhsa::COMPUTE_PGM_RSRC1_GFX9_PLUS_FP16_OVFL,
604 ".amdhsa_fp16_overflow");
605 }
606 if (AMDGPU::isGFX90A(STI))
607 PrintField(KD.compute_pgm_rsrc3,
608 amdhsa::COMPUTE_PGM_RSRC3_GFX90A_TG_SPLIT_SHIFT,
609 amdhsa::COMPUTE_PGM_RSRC3_GFX90A_TG_SPLIT, ".amdhsa_tg_split");
610 if (AMDGPU::supportsWGP(STI))
611 PrintField(KD.compute_pgm_rsrc1,
612 amdhsa::COMPUTE_PGM_RSRC1_GFX10_PLUS_WGP_MODE_SHIFT,
613 amdhsa::COMPUTE_PGM_RSRC1_GFX10_PLUS_WGP_MODE,
614 ".amdhsa_workgroup_processor_mode");
615 if (IVersion.Major >= 10) {
616 PrintField(KD.compute_pgm_rsrc1,
617 amdhsa::COMPUTE_PGM_RSRC1_GFX10_PLUS_MEM_ORDERED_SHIFT,
618 amdhsa::COMPUTE_PGM_RSRC1_GFX10_PLUS_MEM_ORDERED,
619 ".amdhsa_memory_ordered");
620 PrintField(KD.compute_pgm_rsrc1,
621 amdhsa::COMPUTE_PGM_RSRC1_GFX10_PLUS_FWD_PROGRESS_SHIFT,
622 amdhsa::COMPUTE_PGM_RSRC1_GFX10_PLUS_FWD_PROGRESS,
623 ".amdhsa_forward_progress");
624 }
625 if (IVersion.Major >= 10 && IVersion.Major < 12) {
626 PrintField(KD.compute_pgm_rsrc3,
627 amdhsa::COMPUTE_PGM_RSRC3_GFX10_GFX11_SHARED_VGPR_COUNT_SHIFT,
628 amdhsa::COMPUTE_PGM_RSRC3_GFX10_GFX11_SHARED_VGPR_COUNT,
629 ".amdhsa_shared_vgpr_count");
630 }
631 if (IVersion.Major == 11) {
632 PrintField(KD.compute_pgm_rsrc3,
633 amdhsa::COMPUTE_PGM_RSRC3_GFX11_INST_PREF_SIZE_SHIFT,
634 amdhsa::COMPUTE_PGM_RSRC3_GFX11_INST_PREF_SIZE,
635 ".amdhsa_inst_pref_size");
636 }
637 if (IVersion.Major >= 12) {
638 PrintField(KD.compute_pgm_rsrc3,
639 amdhsa::COMPUTE_PGM_RSRC3_GFX12_PLUS_INST_PREF_SIZE_SHIFT,
640 amdhsa::COMPUTE_PGM_RSRC3_GFX12_PLUS_INST_PREF_SIZE,
641 ".amdhsa_inst_pref_size");
642 PrintField(KD.compute_pgm_rsrc1,
643 amdhsa::COMPUTE_PGM_RSRC1_GFX12_PLUS_ENABLE_WG_RR_EN_SHIFT,
644 amdhsa::COMPUTE_PGM_RSRC1_GFX12_PLUS_ENABLE_WG_RR_EN,
645 ".amdhsa_round_robin_scheduling");
646 }
647 PrintField(
648 KD.compute_pgm_rsrc2,
649 amdhsa::
650 COMPUTE_PGM_RSRC2_ENABLE_EXCEPTION_IEEE_754_FP_INVALID_OPERATION_SHIFT,
651 amdhsa::COMPUTE_PGM_RSRC2_ENABLE_EXCEPTION_IEEE_754_FP_INVALID_OPERATION,
652 ".amdhsa_exception_fp_ieee_invalid_op");
653 PrintField(
654 KD.compute_pgm_rsrc2,
655 amdhsa::COMPUTE_PGM_RSRC2_ENABLE_EXCEPTION_FP_DENORMAL_SOURCE_SHIFT,
656 amdhsa::COMPUTE_PGM_RSRC2_ENABLE_EXCEPTION_FP_DENORMAL_SOURCE,
657 ".amdhsa_exception_fp_denorm_src");
658 PrintField(
659 KD.compute_pgm_rsrc2,
660 amdhsa::
661 COMPUTE_PGM_RSRC2_ENABLE_EXCEPTION_IEEE_754_FP_DIVISION_BY_ZERO_SHIFT,
662 amdhsa::COMPUTE_PGM_RSRC2_ENABLE_EXCEPTION_IEEE_754_FP_DIVISION_BY_ZERO,
663 ".amdhsa_exception_fp_ieee_div_zero");
664 PrintField(
665 KD.compute_pgm_rsrc2,
666 amdhsa::COMPUTE_PGM_RSRC2_ENABLE_EXCEPTION_IEEE_754_FP_OVERFLOW_SHIFT,
667 amdhsa::COMPUTE_PGM_RSRC2_ENABLE_EXCEPTION_IEEE_754_FP_OVERFLOW,
668 ".amdhsa_exception_fp_ieee_overflow");
669 PrintField(
670 KD.compute_pgm_rsrc2,
671 amdhsa::COMPUTE_PGM_RSRC2_ENABLE_EXCEPTION_IEEE_754_FP_UNDERFLOW_SHIFT,
672 amdhsa::COMPUTE_PGM_RSRC2_ENABLE_EXCEPTION_IEEE_754_FP_UNDERFLOW,
673 ".amdhsa_exception_fp_ieee_underflow");
674 PrintField(
675 KD.compute_pgm_rsrc2,
676 amdhsa::COMPUTE_PGM_RSRC2_ENABLE_EXCEPTION_IEEE_754_FP_INEXACT_SHIFT,
677 amdhsa::COMPUTE_PGM_RSRC2_ENABLE_EXCEPTION_IEEE_754_FP_INEXACT,
678 ".amdhsa_exception_fp_ieee_inexact");
679 PrintField(
680 KD.compute_pgm_rsrc2,
681 amdhsa::COMPUTE_PGM_RSRC2_ENABLE_EXCEPTION_INT_DIVIDE_BY_ZERO_SHIFT,
682 amdhsa::COMPUTE_PGM_RSRC2_ENABLE_EXCEPTION_INT_DIVIDE_BY_ZERO,
683 ".amdhsa_exception_int_div_zero");
684
685 OS << "\t.end_amdhsa_kernel\n";
686}
687
688namespace {
689/// Callback type invoked by \c forEachInfoScope for each function scope in
690/// the canonical iteration order. The scope is emitted exactly once per
691/// unique \p Sym regardless of how many flat entries reference it.
692using InfoScopeEmitter = function_ref<void(
693 MCSymbol *Sym, const AMDGPU::FuncInfo *Info, ArrayRef<MCSymbol *> Uses,
694 ArrayRef<MCSymbol *> Calls, ArrayRef<StringRef> IndirectCallTypeIds,
695 ArrayRef<StringRef> TypeIds)>;
696
697/// Group the flat edge lists in \p Data by source function symbol and drive
698/// per-scope emission. A scope is opened for every function with attached
699/// info and for every function that appears only as an edge source; each
700/// scope is emitted exactly once. Both the asm and ELF streamers share this
701/// iteration logic and only differ in the per-scope emission callback.
702static void forEachInfoScope(const AMDGPU::InfoSectionData &Data,
703 InfoScopeEmitter Emit) {
704 DenseMap<MCSymbol *, SmallVector<MCSymbol *, 2>> FuncUses;
705 DenseMap<MCSymbol *, SmallVector<MCSymbol *, 4>> FuncCalls;
706 DenseMap<MCSymbol *, SmallVector<StringRef, 2>> FuncIndirectCalls;
707 DenseMap<MCSymbol *, SmallVector<StringRef, 1>> FuncTypeIds;
708 for (const auto &[Func, Res] : Data.Uses)
709 FuncUses[Func].push_back(Elt: Res);
710 for (const auto &[Src, Dst] : Data.Calls)
711 FuncCalls[Src].push_back(Elt: Dst);
712 for (const auto &[Func, TypeId] : Data.IndirectCalls)
713 FuncIndirectCalls[Func].push_back(Elt: TypeId);
714 for (const auto &[Sym, TypeId] : Data.TypeIds)
715 FuncTypeIds[Sym].push_back(Elt: TypeId);
716
717 DenseSet<MCSymbol *> Emitted;
718 auto EmitIfNew = [&](MCSymbol *Sym, const AMDGPU::FuncInfo *Info) {
719 if (!Emitted.insert(V: Sym).second)
720 return;
721 ArrayRef<MCSymbol *> Uses, Calls;
722 ArrayRef<StringRef> IndirectCallTypeIds, TypeIds;
723 if (auto It = FuncUses.find(Val: Sym); It != FuncUses.end())
724 Uses = It->second;
725 if (auto It = FuncCalls.find(Val: Sym); It != FuncCalls.end())
726 Calls = It->second;
727 if (auto It = FuncIndirectCalls.find(Val: Sym); It != FuncIndirectCalls.end())
728 IndirectCallTypeIds = It->second;
729 if (auto It = FuncTypeIds.find(Val: Sym); It != FuncTypeIds.end())
730 TypeIds = It->second;
731 Emit(Sym, Info, Uses, Calls, IndirectCallTypeIds, TypeIds);
732 };
733
734 for (const AMDGPU::FuncInfo &Func : Data.Funcs)
735 EmitIfNew(Func.Sym, &Func);
736 // Emit scopes for functions that only appear as edge sources (e.g. typeid
737 // tags on address-taken declarations, or callers of external functions).
738 for (const auto &[Sym, TypeId] : Data.TypeIds)
739 EmitIfNew(Sym, nullptr);
740 for (const auto &[Sym, Res] : Data.Uses)
741 EmitIfNew(Sym, nullptr);
742 for (const auto &[Sym, Dst] : Data.Calls)
743 EmitIfNew(Sym, nullptr);
744 for (const auto &[Sym, TypeId] : Data.IndirectCalls)
745 EmitIfNew(Sym, nullptr);
746}
747} // namespace
748
749void AMDGPUTargetAsmStreamer::emitAMDGPUInfo(
750 const AMDGPU::InfoSectionData &Data) {
751 forEachInfoScope(Data, Emit: [&](MCSymbol *Sym, const AMDGPU::FuncInfo *Info,
752 ArrayRef<MCSymbol *> Uses,
753 ArrayRef<MCSymbol *> Calls,
754 ArrayRef<StringRef> IndirectCallTypeIds,
755 ArrayRef<StringRef> TypeIds) {
756 OS << "\t.amdgpu_info " << Sym->getName() << '\n';
757 if (Info) {
758 AMDGPU::FuncInfoFlags Flags{};
759 if (Info->UsesVCC)
760 Flags |= AMDGPU::FuncInfoFlags::FUNC_USES_VCC;
761 if (Info->UsesFlatScratch)
762 Flags |= AMDGPU::FuncInfoFlags::FUNC_USES_FLAT_SCRATCH;
763 if (Info->HasDynStack)
764 Flags |= AMDGPU::FuncInfoFlags::FUNC_HAS_DYN_STACK;
765 OS << "\t\t.amdgpu_flags " << llvm::to_underlying(E: Flags) << '\n';
766 OS << "\t\t.amdgpu_num_sgpr " << Info->NumSGPR << '\n';
767 OS << "\t\t.amdgpu_num_vgpr " << Info->NumArchVGPR << '\n';
768 if (Info->NumAccVGPR)
769 OS << "\t\t.amdgpu_num_agpr " << Info->NumAccVGPR << '\n';
770 OS << "\t\t.amdgpu_private_segment_size " << Info->PrivateSegmentSize
771 << '\n';
772 }
773 for (MCSymbol *Res : Uses)
774 OS << "\t\t.amdgpu_use " << Res->getName() << '\n';
775 for (MCSymbol *Dst : Calls)
776 OS << "\t\t.amdgpu_call " << Dst->getName() << '\n';
777 for (StringRef TypeId : IndirectCallTypeIds)
778 OS << "\t\t.amdgpu_indirect_call \"" << TypeId << "\"\n";
779 for (StringRef TypeId : TypeIds)
780 OS << "\t\t.amdgpu_typeid \"" << TypeId << "\"\n";
781 OS << "\t.end_amdgpu_info\n\n";
782 });
783}
784
785//===----------------------------------------------------------------------===//
786// AMDGPUTargetELFStreamer
787//===----------------------------------------------------------------------===//
788
789AMDGPUTargetELFStreamer::AMDGPUTargetELFStreamer(MCStreamer &S,
790 const MCSubtargetInfo &STI)
791 : AMDGPUTargetStreamer(S), STI(STI), Streamer(S) {}
792
793MCELFStreamer &AMDGPUTargetELFStreamer::getStreamer() {
794 return static_cast<MCELFStreamer &>(Streamer);
795}
796
797// A hook for emitting stuff at the end.
798// We use it for emitting the accumulated PAL metadata as a .note record.
799// The PAL metadata is reset after it is emitted.
800void AMDGPUTargetELFStreamer::finish() {
801 ELFObjectWriter &W = getStreamer().getWriter();
802 W.setELFHeaderEFlags(getEFlags());
803 W.setOverrideABIVersion(
804 getELFABIVersion(OS: STI.getTargetTriple(), CodeObjectVersion));
805
806 std::string Blob;
807 const char *Vendor = getPALMetadata()->getVendor();
808 unsigned Type = getPALMetadata()->getType();
809 getPALMetadata()->toBlob(Type, S&: Blob);
810 if (Blob.empty())
811 return;
812 EmitNote(Name: Vendor, DescSize: MCConstantExpr::create(Value: Blob.size(), Ctx&: getContext()), NoteType: Type,
813 EmitDesc: [&](MCELFStreamer &OS) { OS.emitBytes(Data: Blob); });
814
815 // Reset the pal metadata so its data will not affect a compilation that
816 // reuses this object.
817 getPALMetadata()->reset();
818}
819
820void AMDGPUTargetELFStreamer::EmitNote(
821 StringRef Name, const MCExpr *DescSZ, unsigned NoteType,
822 function_ref<void(MCELFStreamer &)> EmitDesc) {
823 auto &S = getStreamer();
824 auto &Context = S.getContext();
825
826 auto NameSZ = Name.size() + 1;
827
828 unsigned NoteFlags = 0;
829 // TODO Apparently, this is currently needed for OpenCL as mentioned in
830 // https://reviews.llvm.org/D74995
831 if (isHsaAbi(STI))
832 NoteFlags = ELF::SHF_ALLOC;
833
834 S.pushSection();
835 S.switchSection(
836 Section: Context.getELFSection(Section: ElfNote::SectionName, Type: ELF::SHT_NOTE, Flags: NoteFlags));
837 S.emitInt32(Value: NameSZ); // namesz
838 S.emitValue(Value: DescSZ, Size: 4); // descz
839 S.emitInt32(Value: NoteType); // type
840 S.emitBytes(Data: Name); // name
841 S.emitInt8(Value: 0); // null terminator
842 S.emitValueToAlignment(Alignment: Align(4), Fill: 0, FillLen: 1, MaxBytesToEmit: 0); // padding 0
843 EmitDesc(S); // desc
844 S.emitValueToAlignment(Alignment: Align(4), Fill: 0, FillLen: 1, MaxBytesToEmit: 0); // padding 0
845 S.popSection();
846}
847
848unsigned AMDGPUTargetELFStreamer::getEFlags() {
849 switch (STI.getTargetTriple().getArch()) {
850 default:
851 llvm_unreachable("Unsupported Arch");
852 case Triple::r600:
853 return getEFlagsR600();
854 case Triple::amdgpu:
855 return getEFlagsAMDGCN();
856 }
857}
858
859unsigned AMDGPUTargetELFStreamer::getEFlagsR600() {
860 assert(STI.getTargetTriple().getArch() == Triple::r600);
861
862 return getElfMach(GPU: STI.getCPU());
863}
864
865unsigned AMDGPUTargetELFStreamer::getEFlagsAMDGCN() {
866 assert(STI.getTargetTriple().isAMDGCN());
867
868 switch (STI.getTargetTriple().getOS()) {
869 default:
870 // TODO: Why are some tests have "mingw" listed as OS?
871 // llvm_unreachable("Unsupported OS");
872 case Triple::UnknownOS:
873 return getEFlagsUnknownOS();
874 case Triple::AMDHSA:
875 return getEFlagsAMDHSA();
876 case Triple::AMDPAL:
877 return getEFlagsAMDPAL();
878 case Triple::Mesa3D:
879 return getEFlagsMesa3D();
880 }
881}
882
883unsigned AMDGPUTargetELFStreamer::getEFlagsUnknownOS() {
884 // TODO: Why are some tests have "mingw" listed as OS?
885 // assert(STI.getTargetTriple().getOS() == Triple::UnknownOS);
886
887 return getEFlagsV3();
888}
889
890unsigned AMDGPUTargetELFStreamer::getEFlagsAMDHSA() {
891 assert(isHsaAbi(STI));
892
893 if (CodeObjectVersion >= 6)
894 return getEFlagsV6();
895 return getEFlagsV4();
896}
897
898unsigned AMDGPUTargetELFStreamer::getEFlagsAMDPAL() {
899 assert(STI.getTargetTriple().getOS() == Triple::AMDPAL);
900
901 return getEFlagsV3();
902}
903
904unsigned AMDGPUTargetELFStreamer::getEFlagsMesa3D() {
905 assert(STI.getTargetTriple().getOS() == Triple::Mesa3D);
906
907 return getEFlagsV3();
908}
909
910unsigned AMDGPUTargetELFStreamer::getEFlagsV3() {
911 unsigned EFlagsV3 = 0;
912
913 // mach.
914 EFlagsV3 |= getElfMach(GPU: STI.getCPU());
915
916 // xnack.
917 if (getTargetID()->isXnackOnOrAny())
918 EFlagsV3 |= ELF::EF_AMDGPU_FEATURE_XNACK_V3;
919 // sramecc.
920 if (getTargetID()->isSramEccOnOrAny())
921 EFlagsV3 |= ELF::EF_AMDGPU_FEATURE_SRAMECC_V3;
922
923 return EFlagsV3;
924}
925
926unsigned AMDGPUTargetELFStreamer::getEFlagsV4() {
927 unsigned EFlagsV4 = 0;
928
929 // mach.
930 EFlagsV4 |= getElfMach(GPU: STI.getCPU());
931
932 // xnack.
933 switch (getTargetID()->getXnackSetting()) {
934 case AMDGPU::TargetIDSetting::Unsupported:
935 EFlagsV4 |= ELF::EF_AMDGPU_FEATURE_XNACK_UNSUPPORTED_V4;
936 break;
937 case AMDGPU::TargetIDSetting::Any:
938 EFlagsV4 |= ELF::EF_AMDGPU_FEATURE_XNACK_ANY_V4;
939 break;
940 case AMDGPU::TargetIDSetting::Off:
941 EFlagsV4 |= ELF::EF_AMDGPU_FEATURE_XNACK_OFF_V4;
942 break;
943 case AMDGPU::TargetIDSetting::On:
944 EFlagsV4 |= ELF::EF_AMDGPU_FEATURE_XNACK_ON_V4;
945 break;
946 }
947 // sramecc.
948 switch (getTargetID()->getSramEccSetting()) {
949 case AMDGPU::TargetIDSetting::Unsupported:
950 EFlagsV4 |= ELF::EF_AMDGPU_FEATURE_SRAMECC_UNSUPPORTED_V4;
951 break;
952 case AMDGPU::TargetIDSetting::Any:
953 EFlagsV4 |= ELF::EF_AMDGPU_FEATURE_SRAMECC_ANY_V4;
954 break;
955 case AMDGPU::TargetIDSetting::Off:
956 EFlagsV4 |= ELF::EF_AMDGPU_FEATURE_SRAMECC_OFF_V4;
957 break;
958 case AMDGPU::TargetIDSetting::On:
959 EFlagsV4 |= ELF::EF_AMDGPU_FEATURE_SRAMECC_ON_V4;
960 break;
961 }
962
963 return EFlagsV4;
964}
965
966unsigned AMDGPUTargetELFStreamer::getEFlagsV6() {
967 unsigned Flags = getEFlagsV4();
968
969 unsigned Version = ForceGenericVersion;
970 if (!Version) {
971 switch (parseArchAMDGCN(CPU: STI.getCPU())) {
972 case AMDGPU::GK_GFX9_GENERIC:
973 Version = GenericVersion::GFX9;
974 break;
975 case AMDGPU::GK_GFX9_4_GENERIC:
976 Version = GenericVersion::GFX9_4;
977 break;
978 case AMDGPU::GK_GFX10_1_GENERIC:
979 Version = GenericVersion::GFX10_1;
980 break;
981 case AMDGPU::GK_GFX10_3_GENERIC:
982 Version = GenericVersion::GFX10_3;
983 break;
984 case AMDGPU::GK_GFX11_GENERIC:
985 Version = GenericVersion::GFX11;
986 break;
987 case AMDGPU::GK_GFX11_7_GENERIC:
988 Version = GenericVersion::GFX11_7;
989 break;
990 case AMDGPU::GK_GFX12_GENERIC:
991 Version = GenericVersion::GFX12;
992 break;
993 case AMDGPU::GK_GFX12_5_GENERIC:
994 Version = GenericVersion::GFX12_5;
995 break;
996 case AMDGPU::GK_GFX13_GENERIC:
997 Version = GenericVersion::GFX13;
998 break;
999 default:
1000 break;
1001 }
1002 }
1003
1004 // Versions start at 1.
1005 if (Version) {
1006 if (Version > ELF::EF_AMDGPU_GENERIC_VERSION_MAX)
1007 report_fatal_error(reason: "Cannot encode generic code object version " +
1008 Twine(Version) +
1009 " - no ELF flag can represent this version!");
1010 Flags |= (Version << ELF::EF_AMDGPU_GENERIC_VERSION_OFFSET);
1011 }
1012
1013 return Flags;
1014}
1015
1016void AMDGPUTargetELFStreamer::EmitDirectiveAMDGCNTarget() {}
1017
1018void AMDGPUTargetELFStreamer::EmitAMDKernelCodeT(AMDGPUMCKernelCodeT &Header) {
1019 MCStreamer &OS = getStreamer();
1020 OS.pushSection();
1021 Header.EmitKernelCodeT(OS, Ctx&: getContext());
1022 OS.popSection();
1023}
1024
1025void AMDGPUTargetELFStreamer::EmitAMDGPUSymbolType(StringRef SymbolName,
1026 unsigned Type) {
1027 auto *Symbol = static_cast<MCSymbolELF *>(
1028 getStreamer().getContext().getOrCreateSymbol(Name: SymbolName));
1029 Symbol->setType(Type);
1030}
1031
1032void AMDGPUTargetELFStreamer::emitAMDGPULDS(MCSymbol *Symbol, unsigned Size,
1033 Align Alignment) {
1034 auto *SymbolELF = static_cast<MCSymbolELF *>(Symbol);
1035 SymbolELF->setType(ELF::STT_OBJECT);
1036
1037 if (!SymbolELF->isBindingSet())
1038 SymbolELF->setBinding(ELF::STB_GLOBAL);
1039
1040 if (SymbolELF->declareCommon(Size, Alignment)) {
1041 report_fatal_error(reason: "Symbol: " + Symbol->getName() +
1042 " redeclared as different type");
1043 }
1044
1045 SymbolELF->setIndex(ELF::SHN_AMDGPU_LDS);
1046 SymbolELF->setSize(MCConstantExpr::create(Value: Size, Ctx&: getContext()));
1047}
1048
1049bool AMDGPUTargetELFStreamer::EmitISAVersion() {
1050 // Create two labels to mark the beginning and end of the desc field
1051 // and a MCExpr to calculate the size of the desc field.
1052 auto &Context = getContext();
1053 auto *DescBegin = Context.createTempSymbol();
1054 auto *DescEnd = Context.createTempSymbol();
1055 auto *DescSZ = MCBinaryExpr::createSub(
1056 LHS: MCSymbolRefExpr::create(Symbol: DescEnd, Ctx&: Context),
1057 RHS: MCSymbolRefExpr::create(Symbol: DescBegin, Ctx&: Context), Ctx&: Context);
1058
1059 EmitNote(Name: ElfNote::NoteNameV2, DescSZ, NoteType: ELF::NT_AMD_HSA_ISA_NAME,
1060 EmitDesc: [&](MCELFStreamer &OS) {
1061 OS.emitLabel(Symbol: DescBegin);
1062
1063 SmallString<32> Str;
1064 raw_svector_ostream StrOS(Str);
1065 StrOS << *getTargetID();
1066
1067 OS.emitBytes(Data: StrOS.str());
1068 OS.emitLabel(Symbol: DescEnd);
1069 });
1070 return true;
1071}
1072
1073bool AMDGPUTargetELFStreamer::EmitHSAMetadata(msgpack::Document &HSAMetadataDoc,
1074 bool Strict) {
1075 HSAMD::V3::MetadataVerifier Verifier(Strict);
1076 if (!Verifier.verify(HSAMetadataRoot&: HSAMetadataDoc.getRoot()))
1077 return false;
1078
1079 std::string HSAMetadataString;
1080 HSAMetadataDoc.writeToBlob(Blob&: HSAMetadataString);
1081
1082 // Create two labels to mark the beginning and end of the desc field
1083 // and a MCExpr to calculate the size of the desc field.
1084 auto &Context = getContext();
1085 auto *DescBegin = Context.createTempSymbol();
1086 auto *DescEnd = Context.createTempSymbol();
1087 auto *DescSZ = MCBinaryExpr::createSub(
1088 LHS: MCSymbolRefExpr::create(Symbol: DescEnd, Ctx&: Context),
1089 RHS: MCSymbolRefExpr::create(Symbol: DescBegin, Ctx&: Context), Ctx&: Context);
1090
1091 EmitNote(Name: ElfNote::NoteNameV3, DescSZ, NoteType: ELF::NT_AMDGPU_METADATA,
1092 EmitDesc: [&](MCELFStreamer &OS) {
1093 OS.emitLabel(Symbol: DescBegin);
1094 OS.emitBytes(Data: HSAMetadataString);
1095 OS.emitLabel(Symbol: DescEnd);
1096 });
1097 return true;
1098}
1099
1100bool AMDGPUTargetELFStreamer::EmitCodeEnd(const MCSubtargetInfo &STI) {
1101 const uint32_t Encoded_s_code_end = 0xbf9f0000;
1102 const uint32_t Encoded_s_nop = 0xbf800000;
1103 uint32_t Encoded_pad = Encoded_s_code_end;
1104
1105 // Instruction cache line size in bytes.
1106 const unsigned Log2CacheLineSize = AMDGPU::isGFX11Plus(STI) ? 7 : 6;
1107 const unsigned CacheLineSize = 1u << Log2CacheLineSize;
1108
1109 // Extra padding amount in bytes to support prefetch mode 3.
1110 unsigned FillSize = 3 * CacheLineSize;
1111
1112 if (AMDGPU::isGFX90A(STI)) {
1113 Encoded_pad = Encoded_s_nop;
1114 FillSize = 16 * CacheLineSize;
1115 }
1116
1117 MCStreamer &OS = getStreamer();
1118 OS.pushSection();
1119 OS.emitValueToAlignment(Alignment: Align(CacheLineSize), Fill: Encoded_pad, FillLen: 4);
1120 for (unsigned I = 0; I < FillSize; I += 4)
1121 OS.emitInt32(Value: Encoded_pad);
1122 OS.popSection();
1123 return true;
1124}
1125
1126void AMDGPUTargetELFStreamer::EmitAmdhsaKernelDescriptor(
1127 const MCSubtargetInfo &STI, StringRef KernelName,
1128 const MCKernelDescriptor &KernelDescriptor, const MCExpr *NextVGPR,
1129 const MCExpr *NextSGPR, const MCExpr *ReserveVCC,
1130 const MCExpr *ReserveFlatScr) {
1131 auto &Streamer = getStreamer();
1132 auto &Context = Streamer.getContext();
1133
1134 auto *KernelCodeSymbol =
1135 static_cast<MCSymbolELF *>(Context.getOrCreateSymbol(Name: Twine(KernelName)));
1136 auto *KernelDescriptorSymbol = static_cast<MCSymbolELF *>(
1137 Context.getOrCreateSymbol(Name: Twine(KernelName) + Twine(".kd")));
1138
1139 // Copy kernel descriptor symbol's binding, other and visibility from the
1140 // kernel code symbol.
1141 KernelDescriptorSymbol->setBinding(KernelCodeSymbol->getBinding());
1142 KernelDescriptorSymbol->setOther(KernelCodeSymbol->getOther());
1143 KernelDescriptorSymbol->setVisibility(KernelCodeSymbol->getVisibility());
1144 // Kernel descriptor symbol's type and size are fixed.
1145 KernelDescriptorSymbol->setType(ELF::STT_OBJECT);
1146 KernelDescriptorSymbol->setSize(
1147 MCConstantExpr::create(Value: sizeof(amdhsa::kernel_descriptor_t), Ctx&: Context));
1148
1149 // The visibility of the kernel code symbol must be protected or less to allow
1150 // static relocations from the kernel descriptor to be used.
1151 if (KernelCodeSymbol->getVisibility() == ELF::STV_DEFAULT)
1152 KernelCodeSymbol->setVisibility(ELF::STV_PROTECTED);
1153
1154 Streamer.emitLabel(Symbol: KernelDescriptorSymbol);
1155 Streamer.emitValue(
1156 Value: KernelDescriptor.group_segment_fixed_size,
1157 Size: sizeof(amdhsa::kernel_descriptor_t::group_segment_fixed_size));
1158 Streamer.emitValue(
1159 Value: KernelDescriptor.private_segment_fixed_size,
1160 Size: sizeof(amdhsa::kernel_descriptor_t::private_segment_fixed_size));
1161 Streamer.emitValue(Value: KernelDescriptor.kernarg_size,
1162 Size: sizeof(amdhsa::kernel_descriptor_t::kernarg_size));
1163
1164 for (uint32_t i = 0; i < sizeof(amdhsa::kernel_descriptor_t::reserved0); ++i)
1165 Streamer.emitInt8(Value: 0u);
1166
1167 // FIXME: Remove the use of VK_AMDGPU_REL64 in the expression below. The
1168 // expression being created is:
1169 // (start of kernel code) - (start of kernel descriptor)
1170 // It implies R_AMDGPU_REL64, but ends up being R_AMDGPU_ABS64.
1171 Streamer.emitValue(
1172 Value: MCBinaryExpr::createSub(
1173 LHS: MCSymbolRefExpr::create(Symbol: KernelCodeSymbol, specifier: AMDGPUMCExpr::S_REL64,
1174 Ctx&: Context),
1175 RHS: MCSymbolRefExpr::create(Symbol: KernelDescriptorSymbol, Ctx&: Context), Ctx&: Context),
1176 Size: sizeof(amdhsa::kernel_descriptor_t::kernel_code_entry_byte_offset));
1177 for (uint32_t i = 0; i < sizeof(amdhsa::kernel_descriptor_t::reserved1); ++i)
1178 Streamer.emitInt8(Value: 0u);
1179 Streamer.emitValue(Value: KernelDescriptor.compute_pgm_rsrc3,
1180 Size: sizeof(amdhsa::kernel_descriptor_t::compute_pgm_rsrc3));
1181 Streamer.emitValue(Value: KernelDescriptor.compute_pgm_rsrc1,
1182 Size: sizeof(amdhsa::kernel_descriptor_t::compute_pgm_rsrc1));
1183 Streamer.emitValue(Value: KernelDescriptor.compute_pgm_rsrc2,
1184 Size: sizeof(amdhsa::kernel_descriptor_t::compute_pgm_rsrc2));
1185 Streamer.emitValue(
1186 Value: KernelDescriptor.kernel_code_properties,
1187 Size: sizeof(amdhsa::kernel_descriptor_t::kernel_code_properties));
1188 Streamer.emitValue(Value: KernelDescriptor.kernarg_preload,
1189 Size: sizeof(amdhsa::kernel_descriptor_t::kernarg_preload));
1190 for (uint32_t i = 0; i < sizeof(amdhsa::kernel_descriptor_t::reserved3); ++i)
1191 Streamer.emitInt8(Value: 0u);
1192}
1193
1194void AMDGPUTargetELFStreamer::emitAMDGPUInfo(
1195 const AMDGPU::InfoSectionData &Data) {
1196 MCELFStreamer &S = getStreamer();
1197 MCContext &Context = S.getContext();
1198
1199 StringTableBuilder StrTab(StringTableBuilder::ELF);
1200 auto getOrAddString = [&](StringRef Str) -> uint32_t {
1201 if (Str.empty())
1202 return UINT32_MAX;
1203 return StrTab.add(S: Str);
1204 };
1205
1206 auto EmitU32Entry = [&](AMDGPU::InfoKind Kind, uint32_t Val) {
1207 S.emitInt8(Value: static_cast<uint8_t>(Kind));
1208 S.emitInt8(Value: 4);
1209 S.emitInt32(Value: Val);
1210 };
1211 auto EmitSymEntry = [&](AMDGPU::InfoKind Kind, MCSymbol *Sym) {
1212 S.emitInt8(Value: static_cast<uint8_t>(Kind));
1213 S.emitInt8(Value: 8);
1214 S.emitValue(Value: MCSymbolRefExpr::create(Symbol: Sym, Ctx&: Context), Size: 8);
1215 };
1216
1217 S.pushSection();
1218 MCSectionELF *InfoSec = Context.getELFSection(
1219 Section: ".amdgpu.info", Type: ELF::SHT_PROGBITS, Flags: ELF::SHF_EXCLUDE);
1220 S.switchSection(Section: InfoSec);
1221
1222 forEachInfoScope(Data, Emit: [&](MCSymbol *Sym, const AMDGPU::FuncInfo *Info,
1223 ArrayRef<MCSymbol *> Uses,
1224 ArrayRef<MCSymbol *> Calls,
1225 ArrayRef<StringRef> IndirectCallTypeIds,
1226 ArrayRef<StringRef> TypeIds) {
1227 EmitSymEntry(AMDGPU::InfoKind::INFO_FUNC, Sym);
1228
1229 if (Info) {
1230 AMDGPU::FuncInfoFlags Flags{};
1231 if (Info->UsesVCC)
1232 Flags |= AMDGPU::FuncInfoFlags::FUNC_USES_VCC;
1233 if (Info->UsesFlatScratch)
1234 Flags |= AMDGPU::FuncInfoFlags::FUNC_USES_FLAT_SCRATCH;
1235 if (Info->HasDynStack)
1236 Flags |= AMDGPU::FuncInfoFlags::FUNC_HAS_DYN_STACK;
1237 EmitU32Entry(AMDGPU::InfoKind::INFO_FLAGS, llvm::to_underlying(E: Flags));
1238 EmitU32Entry(AMDGPU::InfoKind::INFO_NUM_SGPR, Info->NumSGPR);
1239 EmitU32Entry(AMDGPU::InfoKind::INFO_NUM_VGPR, Info->NumArchVGPR);
1240 // INFO_NUM_AGPR is only emitted when the function actually uses AGPRs,
1241 // since AGPRs are not available on all architectures.
1242 if (Info->NumAccVGPR)
1243 EmitU32Entry(AMDGPU::InfoKind::INFO_NUM_AGPR, Info->NumAccVGPR);
1244 EmitU32Entry(AMDGPU::InfoKind::INFO_PRIVATE_SEGMENT_SIZE,
1245 Info->PrivateSegmentSize);
1246 }
1247
1248 for (MCSymbol *Res : Uses)
1249 EmitSymEntry(AMDGPU::InfoKind::INFO_USE, Res);
1250 for (MCSymbol *Dst : Calls)
1251 EmitSymEntry(AMDGPU::InfoKind::INFO_CALL, Dst);
1252 for (StringRef TypeId : IndirectCallTypeIds) {
1253 EmitU32Entry(AMDGPU::InfoKind::INFO_INDIRECT_CALL,
1254 getOrAddString(TypeId));
1255 }
1256 for (StringRef TypeId : TypeIds)
1257 EmitU32Entry(AMDGPU::InfoKind::INFO_TYPEID, getOrAddString(TypeId));
1258 });
1259
1260 if (!StrTab.empty()) {
1261 StrTab.finalizeInOrder();
1262 MCSectionELF *Sec = Context.getELFSection(Section: ".amdgpu.strtab", Type: ELF::SHT_STRTAB,
1263 Flags: ELF::SHF_EXCLUDE);
1264 S.switchSection(Section: Sec);
1265 SmallString<128> Buf;
1266 raw_svector_ostream OS(Buf);
1267 StrTab.write(OS);
1268 S.emitBytes(Data: Buf);
1269 }
1270
1271 S.popSection();
1272}
1273