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