blob: 304cf5252ae12d6489e66ca497eb9a9ed5d5cfbc [file] [log] [blame]
Konstantin Zhuravlyov7498cd62017-03-22 22:32:22 +00001//===--- AMDGPUCodeObjectMetadataStreamer.cpp -------------------*- C++ -*-===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10/// \file
11/// \brief AMDGPU Code Object Metadata Streamer.
12///
13//
14//===----------------------------------------------------------------------===//
15
16#include "AMDGPU.h"
17#include "AMDGPUCodeObjectMetadataStreamer.h"
18#include "Utils/AMDGPUBaseInfo.h"
19#include "llvm/ADT/StringSwitch.h"
20#include "llvm/IR/Constants.h"
21#include "llvm/IR/Module.h"
22#include "llvm/Support/YAMLTraits.h"
23
24using namespace llvm::AMDGPU;
25using namespace llvm::AMDGPU::CodeObject;
26using namespace llvm::AMDGPU::IsaInfo;
27
28LLVM_YAML_IS_FLOW_SEQUENCE_VECTOR(uint32_t)
29LLVM_YAML_IS_FLOW_SEQUENCE_VECTOR(std::string)
30LLVM_YAML_IS_SEQUENCE_VECTOR(Kernel::Arg::Metadata)
31LLVM_YAML_IS_SEQUENCE_VECTOR(Kernel::Metadata)
32
33namespace llvm {
34
35static cl::opt<bool> DumpCodeObjectMetadata(
36 "amdgpu-dump-comd",
37 cl::desc("Dump AMDGPU Code Object Metadata"));
38static cl::opt<bool> VerifyCodeObjectMetadata(
39 "amdgpu-verify-comd",
40 cl::desc("Verify AMDGPU Code Object Metadata"));
41
42namespace yaml {
43
44template <>
45struct ScalarEnumerationTraits<AccessQualifier> {
46 static void enumeration(IO &YIO, AccessQualifier &EN) {
47 YIO.enumCase(EN, "Default", AccessQualifier::Default);
48 YIO.enumCase(EN, "ReadOnly", AccessQualifier::ReadOnly);
49 YIO.enumCase(EN, "WriteOnly", AccessQualifier::WriteOnly);
50 YIO.enumCase(EN, "ReadWrite", AccessQualifier::ReadWrite);
51 }
52};
53
54template <>
55struct ScalarEnumerationTraits<AddressSpaceQualifier> {
56 static void enumeration(IO &YIO, AddressSpaceQualifier &EN) {
57 YIO.enumCase(EN, "Private", AddressSpaceQualifier::Private);
58 YIO.enumCase(EN, "Global", AddressSpaceQualifier::Global);
59 YIO.enumCase(EN, "Constant", AddressSpaceQualifier::Constant);
60 YIO.enumCase(EN, "Local", AddressSpaceQualifier::Local);
61 YIO.enumCase(EN, "Generic", AddressSpaceQualifier::Generic);
62 YIO.enumCase(EN, "Region", AddressSpaceQualifier::Region);
63 }
64};
65
66template <>
67struct ScalarEnumerationTraits<ValueKind> {
68 static void enumeration(IO &YIO, ValueKind &EN) {
69 YIO.enumCase(EN, "ByValue", ValueKind::ByValue);
70 YIO.enumCase(EN, "GlobalBuffer", ValueKind::GlobalBuffer);
71 YIO.enumCase(EN, "DynamicSharedPointer", ValueKind::DynamicSharedPointer);
72 YIO.enumCase(EN, "Sampler", ValueKind::Sampler);
73 YIO.enumCase(EN, "Image", ValueKind::Image);
74 YIO.enumCase(EN, "Pipe", ValueKind::Pipe);
75 YIO.enumCase(EN, "Queue", ValueKind::Queue);
76 YIO.enumCase(EN, "HiddenGlobalOffsetX", ValueKind::HiddenGlobalOffsetX);
77 YIO.enumCase(EN, "HiddenGlobalOffsetY", ValueKind::HiddenGlobalOffsetY);
78 YIO.enumCase(EN, "HiddenGlobalOffsetZ", ValueKind::HiddenGlobalOffsetZ);
79 YIO.enumCase(EN, "HiddenNone", ValueKind::HiddenNone);
80 YIO.enumCase(EN, "HiddenPrintfBuffer", ValueKind::HiddenPrintfBuffer);
81 YIO.enumCase(EN, "HiddenDefaultQueue", ValueKind::HiddenDefaultQueue);
82 YIO.enumCase(EN, "HiddenCompletionAction",
83 ValueKind::HiddenCompletionAction);
84 }
85};
86
87template <>
88struct ScalarEnumerationTraits<ValueType> {
89 static void enumeration(IO &YIO, ValueType &EN) {
90 YIO.enumCase(EN, "Struct", ValueType::Struct);
91 YIO.enumCase(EN, "I8", ValueType::I8);
92 YIO.enumCase(EN, "U8", ValueType::U8);
93 YIO.enumCase(EN, "I16", ValueType::I16);
94 YIO.enumCase(EN, "U16", ValueType::U16);
95 YIO.enumCase(EN, "F16", ValueType::F16);
96 YIO.enumCase(EN, "I32", ValueType::I32);
97 YIO.enumCase(EN, "U32", ValueType::U32);
98 YIO.enumCase(EN, "F32", ValueType::F32);
99 YIO.enumCase(EN, "I64", ValueType::I64);
100 YIO.enumCase(EN, "U64", ValueType::U64);
101 YIO.enumCase(EN, "F64", ValueType::F64);
102 }
103};
104
105template <>
106struct MappingTraits<Isa::Metadata> {
107 static void mapping(IO &YIO, Isa::Metadata &MD) {
108 YIO.mapRequired(Isa::Key::WavefrontSize, MD.mWavefrontSize);
109 YIO.mapRequired(Isa::Key::LocalMemorySize, MD.mLocalMemorySize);
110 YIO.mapRequired(Isa::Key::EUsPerCU, MD.mEUsPerCU);
111 YIO.mapRequired(Isa::Key::MaxWavesPerEU, MD.mMaxWavesPerEU);
112 YIO.mapRequired(Isa::Key::MaxFlatWorkGroupSize, MD.mMaxFlatWorkGroupSize);
113 YIO.mapRequired(Isa::Key::SGPRAllocGranule, MD.mSGPRAllocGranule);
114 YIO.mapRequired(Isa::Key::TotalNumSGPRs, MD.mTotalNumSGPRs);
115 YIO.mapRequired(Isa::Key::AddressableNumSGPRs, MD.mAddressableNumSGPRs);
116 YIO.mapRequired(Isa::Key::VGPRAllocGranule, MD.mVGPRAllocGranule);
117 YIO.mapRequired(Isa::Key::TotalNumVGPRs, MD.mTotalNumVGPRs);
118 YIO.mapRequired(Isa::Key::AddressableNumVGPRs, MD.mAddressableNumVGPRs);
119 }
120};
121
122template <>
123struct MappingTraits<Kernel::Attrs::Metadata> {
124 static void mapping(IO &YIO, Kernel::Attrs::Metadata &MD) {
125 YIO.mapOptional(Kernel::Attrs::Key::ReqdWorkGroupSize,
126 MD.mReqdWorkGroupSize, std::vector<uint32_t>());
127 YIO.mapOptional(Kernel::Attrs::Key::WorkGroupSizeHint,
128 MD.mWorkGroupSizeHint, std::vector<uint32_t>());
129 YIO.mapOptional(Kernel::Attrs::Key::VecTypeHint,
130 MD.mVecTypeHint, std::string());
131 }
132};
133
134template <>
135struct MappingTraits<Kernel::Arg::Metadata> {
136 static void mapping(IO &YIO, Kernel::Arg::Metadata &MD) {
137 YIO.mapRequired(Kernel::Arg::Key::Size, MD.mSize);
138 YIO.mapRequired(Kernel::Arg::Key::Align, MD.mAlign);
139 YIO.mapRequired(Kernel::Arg::Key::Kind, MD.mValueKind);
140 YIO.mapRequired(Kernel::Arg::Key::ValueType, MD.mValueType);
141 YIO.mapOptional(Kernel::Arg::Key::PointeeAlign, MD.mPointeeAlign,
142 uint32_t(0));
143 YIO.mapOptional(Kernel::Arg::Key::AccQual, MD.mAccQual,
144 AccessQualifier::Unknown);
145 YIO.mapOptional(Kernel::Arg::Key::AddrSpaceQual, MD.mAddrSpaceQual,
146 AddressSpaceQualifier::Unknown);
147 YIO.mapOptional(Kernel::Arg::Key::IsConst, MD.mIsConst, false);
148 YIO.mapOptional(Kernel::Arg::Key::IsPipe, MD.mIsPipe, false);
149 YIO.mapOptional(Kernel::Arg::Key::IsRestrict, MD.mIsRestrict, false);
150 YIO.mapOptional(Kernel::Arg::Key::IsVolatile, MD.mIsVolatile, false);
151 YIO.mapOptional(Kernel::Arg::Key::Name, MD.mName, std::string());
152 YIO.mapOptional(Kernel::Arg::Key::TypeName, MD.mTypeName, std::string());
153 }
154};
155
156template <>
157struct MappingTraits<Kernel::Metadata> {
158 static void mapping(IO &YIO, Kernel::Metadata &MD) {
159 YIO.mapRequired(Kernel::Key::Name, MD.mName);
160 YIO.mapOptional(Kernel::Key::Language, MD.mLanguage, std::string());
161 YIO.mapOptional(Kernel::Key::LanguageVersion, MD.mLanguageVersion,
162 std::vector<uint32_t>());
163 if (!MD.mAttrs.empty() || !YIO.outputting())
164 YIO.mapOptional(Kernel::Key::Attrs, MD.mAttrs);
165 if (!MD.mArgs.empty() || !YIO.outputting())
166 YIO.mapOptional(Kernel::Key::Args, MD.mArgs);
167 }
168};
169
170template <>
171struct MappingTraits<CodeObject::Metadata> {
172 static void mapping(IO &YIO, CodeObject::Metadata &MD) {
173 YIO.mapRequired(Key::Version, MD.mVersion);
174 YIO.mapOptional(Key::Isa, MD.mIsa);
175 YIO.mapOptional(Key::Printf, MD.mPrintf, std::vector<std::string>());
176 if (!MD.mKernels.empty() || !YIO.outputting())
177 YIO.mapOptional(Key::Kernels, MD.mKernels);
178 }
179};
180
181} // end namespace yaml
182
183namespace AMDGPU {
184
185/* static */
186std::error_code CodeObject::Metadata::fromYamlString(
187 std::string YamlString, CodeObject::Metadata &CodeObjectMetadata) {
188 yaml::Input YamlInput(YamlString);
189 YamlInput >> CodeObjectMetadata;
190 return YamlInput.error();
191}
192
193/* static */
194std::error_code CodeObject::Metadata::toYamlString(
195 CodeObject::Metadata CodeObjectMetadata, std::string &YamlString) {
196 raw_string_ostream YamlStream(YamlString);
197 yaml::Output YamlOutput(YamlStream, nullptr, std::numeric_limits<int>::max());
198 YamlOutput << CodeObjectMetadata;
199 return std::error_code();
200}
201
202namespace CodeObject {
203
204void MetadataStreamer::dump(StringRef YamlString) const {
205 errs() << "AMDGPU Code Object Metadata:\n" << YamlString << '\n';
206}
207
208void MetadataStreamer::verify(StringRef YamlString) const {
209 errs() << "AMDGPU Code Object Metadata Parser Test: ";
210
211 CodeObject::Metadata FromYamlString;
212 if (Metadata::fromYamlString(YamlString, FromYamlString)) {
213 errs() << "FAIL\n";
214 return;
215 }
216
217 std::string ToYamlString;
218 if (Metadata::toYamlString(FromYamlString, ToYamlString)) {
219 errs() << "FAIL\n";
220 return;
221 }
222
223 errs() << (YamlString == ToYamlString ? "PASS" : "FAIL") << '\n';
224 if (YamlString != ToYamlString) {
225 errs() << "Original input: " << YamlString << '\n'
226 << "Produced output: " << ToYamlString << '\n';
227 }
228}
229
230AccessQualifier MetadataStreamer::getAccessQualifier(StringRef AccQual) const {
231 if (AccQual.empty())
232 return AccessQualifier::Unknown;
233
234 return StringSwitch<AccessQualifier>(AccQual)
235 .Case("read_only", AccessQualifier::ReadOnly)
236 .Case("write_only", AccessQualifier::WriteOnly)
237 .Case("read_write", AccessQualifier::ReadWrite)
238 .Default(AccessQualifier::Default);
239}
240
241AddressSpaceQualifier MetadataStreamer::getAddressSpaceQualifer(
242 unsigned AddressSpace) const {
243 switch (AddressSpace) {
244 case AMDGPUAS::PRIVATE_ADDRESS:
245 return AddressSpaceQualifier::Private;
246 case AMDGPUAS::GLOBAL_ADDRESS:
247 return AddressSpaceQualifier::Global;
248 case AMDGPUAS::CONSTANT_ADDRESS:
249 return AddressSpaceQualifier::Constant;
250 case AMDGPUAS::LOCAL_ADDRESS:
251 return AddressSpaceQualifier::Local;
252 case AMDGPUAS::FLAT_ADDRESS:
253 return AddressSpaceQualifier::Generic;
254 case AMDGPUAS::REGION_ADDRESS:
255 return AddressSpaceQualifier::Region;
256 }
257
258 llvm_unreachable("Unknown address space qualifier");
259}
260
261ValueKind MetadataStreamer::getValueKind(Type *Ty, StringRef TypeQual,
262 StringRef BaseTypeName) const {
263 if (TypeQual.find("pipe") != StringRef::npos)
264 return ValueKind::Pipe;
265
266 return StringSwitch<ValueKind>(BaseTypeName)
267 .Case("sampler_t", ValueKind::Sampler)
268 .Case("queue_t", ValueKind::Queue)
269 .Cases("image1d_t",
270 "image1d_array_t",
271 "image1d_buffer_t",
272 "image2d_t" ,
273 "image2d_array_t",
274 "image2d_array_depth_t",
275 "image2d_array_msaa_t"
276 "image2d_array_msaa_depth_t"
277 "image2d_depth_t",
278 "image2d_msaa_t",
279 "image2d_msaa_depth_t",
280 "image3d_t", ValueKind::Image)
281 .Default(isa<PointerType>(Ty) ?
282 (Ty->getPointerAddressSpace() ==
283 AMDGPUAS::LOCAL_ADDRESS ?
284 ValueKind::DynamicSharedPointer :
285 ValueKind::GlobalBuffer) :
286 ValueKind::ByValue);
287}
288
289ValueType MetadataStreamer::getValueType(Type *Ty, StringRef TypeName) const {
290 switch (Ty->getTypeID()) {
291 case Type::IntegerTyID: {
292 auto Signed = !TypeName.startswith("u");
293 switch (Ty->getIntegerBitWidth()) {
294 case 8:
295 return Signed ? ValueType::I8 : ValueType::U8;
296 case 16:
297 return Signed ? ValueType::I16 : ValueType::U16;
298 case 32:
299 return Signed ? ValueType::I32 : ValueType::U32;
300 case 64:
301 return Signed ? ValueType::I64 : ValueType::U64;
302 default:
303 return ValueType::Struct;
304 }
305 }
306 case Type::HalfTyID:
307 return ValueType::F16;
308 case Type::FloatTyID:
309 return ValueType::F32;
310 case Type::DoubleTyID:
311 return ValueType::F64;
312 case Type::PointerTyID:
313 return getValueType(Ty->getPointerElementType(), TypeName);
314 case Type::VectorTyID:
315 return getValueType(Ty->getVectorElementType(), TypeName);
316 default:
317 return ValueType::Struct;
318 }
319}
320
321std::string MetadataStreamer::getTypeName(Type *Ty, bool Signed) const {
322 switch (Ty->getTypeID()) {
323 case Type::IntegerTyID: {
324 if (!Signed)
325 return (Twine('u') + getTypeName(Ty, true)).str();
326
327 auto BitWidth = Ty->getIntegerBitWidth();
328 switch (BitWidth) {
329 case 8:
330 return "char";
331 case 16:
332 return "short";
333 case 32:
334 return "int";
335 case 64:
336 return "long";
337 default:
338 return (Twine('i') + Twine(BitWidth)).str();
339 }
340 }
341 case Type::HalfTyID:
342 return "half";
343 case Type::FloatTyID:
344 return "float";
345 case Type::DoubleTyID:
346 return "double";
347 case Type::VectorTyID: {
348 auto VecTy = cast<VectorType>(Ty);
349 auto ElTy = VecTy->getElementType();
350 auto NumElements = VecTy->getVectorNumElements();
351 return (Twine(getTypeName(ElTy, Signed)) + Twine(NumElements)).str();
352 }
353 default:
354 return "unknown";
355 }
356}
357
358std::vector<uint32_t> MetadataStreamer::getWorkGroupDimensions(
359 MDNode *Node) const {
360 std::vector<uint32_t> Dims;
361 if (Node->getNumOperands() != 3)
362 return Dims;
363
364 for (auto &Op : Node->operands())
365 Dims.push_back(mdconst::extract<ConstantInt>(Op)->getZExtValue());
366 return Dims;
367}
368
369void MetadataStreamer::emitVersion() {
370 auto &Version = CodeObjectMetadata.mVersion;
371
372 Version.push_back(MetadataVersionMajor);
373 Version.push_back(MetadataVersionMinor);
374}
375
376void MetadataStreamer::emitIsa(const FeatureBitset &Features) {
377 auto &Isa = CodeObjectMetadata.mIsa;
378
379 Isa.mWavefrontSize = getWavefrontSize(Features);
380 Isa.mLocalMemorySize = getLocalMemorySize(Features);
381 Isa.mEUsPerCU = getEUsPerCU(Features);
382 Isa.mMaxWavesPerEU = getMaxWavesPerEU(Features);
383 Isa.mMaxFlatWorkGroupSize = getMaxFlatWorkGroupSize(Features);
384 Isa.mSGPRAllocGranule = getSGPRAllocGranule(Features);
385 Isa.mTotalNumSGPRs = getTotalNumSGPRs(Features);
386 Isa.mAddressableNumSGPRs = getAddressableNumSGPRs(Features);
387 Isa.mVGPRAllocGranule = getVGPRAllocGranule(Features);
388 Isa.mTotalNumVGPRs = getTotalNumVGPRs(Features);
389 Isa.mAddressableNumVGPRs = getAddressableNumVGPRs(Features);
390}
391
392void MetadataStreamer::emitPrintf(const Module &Mod) {
393 auto &Printf = CodeObjectMetadata.mPrintf;
394
395 auto Node = Mod.getNamedMetadata("llvm.printf.fmts");
396 if (!Node)
397 return;
398
399 for (auto Op : Node->operands())
400 if (Op->getNumOperands())
401 Printf.push_back(cast<MDString>(Op->getOperand(0))->getString());
402}
403
404void MetadataStreamer::emitKernelLanguage(const Function &Func) {
405 auto &Kernel = CodeObjectMetadata.mKernels.back();
406
407 // TODO: What about other languages?
408 auto Node = Func.getParent()->getNamedMetadata("opencl.ocl.version");
409 if (!Node || !Node->getNumOperands())
410 return;
411 auto Op0 = Node->getOperand(0);
412 if (Op0->getNumOperands() <= 1)
413 return;
414
415 Kernel.mLanguage = "OpenCL C";
416 Kernel.mLanguageVersion.push_back(
417 mdconst::extract<ConstantInt>(Op0->getOperand(0))->getZExtValue());
418 Kernel.mLanguageVersion.push_back(
419 mdconst::extract<ConstantInt>(Op0->getOperand(1))->getZExtValue());
420}
421
422void MetadataStreamer::emitKernelAttrs(const Function &Func) {
423 auto &Attrs = CodeObjectMetadata.mKernels.back().mAttrs;
424
425 if (auto Node = Func.getMetadata("reqd_work_group_size"))
426 Attrs.mReqdWorkGroupSize = getWorkGroupDimensions(Node);
427 if (auto Node = Func.getMetadata("work_group_size_hint"))
428 Attrs.mWorkGroupSizeHint = getWorkGroupDimensions(Node);
429 if (auto Node = Func.getMetadata("vec_type_hint")) {
430 Attrs.mVecTypeHint = getTypeName(
431 cast<ValueAsMetadata>(Node->getOperand(0))->getType(),
432 mdconst::extract<ConstantInt>(Node->getOperand(1))->getZExtValue());
433 }
434}
435
436void MetadataStreamer::emitKernelArgs(const Function &Func) {
437 for (auto &Arg : Func.args())
438 emitKernelArg(Arg);
439
440 // TODO: What about other languages?
441 if (!Func.getParent()->getNamedMetadata("opencl.ocl.version"))
442 return;
443
444 auto &DL = Func.getParent()->getDataLayout();
445 auto Int64Ty = Type::getInt64Ty(Func.getContext());
446
447 emitKernelArg(DL, Int64Ty, ValueKind::HiddenGlobalOffsetX);
448 emitKernelArg(DL, Int64Ty, ValueKind::HiddenGlobalOffsetY);
449 emitKernelArg(DL, Int64Ty, ValueKind::HiddenGlobalOffsetZ);
450
451 if (!Func.getParent()->getNamedMetadata("llvm.printf.fmts"))
452 return;
453
454 auto Int8PtrTy = Type::getInt8PtrTy(Func.getContext(),
455 AMDGPUAS::GLOBAL_ADDRESS);
456 emitKernelArg(DL, Int8PtrTy, ValueKind::HiddenPrintfBuffer);
457}
458
459void MetadataStreamer::emitKernelArg(const Argument &Arg) {
460 auto Func = Arg.getParent();
461 auto ArgNo = Arg.getArgNo();
462 const MDNode *Node;
463
464 StringRef TypeQual;
465 Node = Func->getMetadata("kernel_arg_type_qual");
466 if (Node && ArgNo < Node->getNumOperands())
467 TypeQual = cast<MDString>(Node->getOperand(ArgNo))->getString();
468
469 StringRef BaseTypeName;
470 Node = Func->getMetadata("kernel_arg_base_type");
471 if (Node && ArgNo < Node->getNumOperands())
472 BaseTypeName = cast<MDString>(Node->getOperand(ArgNo))->getString();
473
474 StringRef AccQual;
475 Node = Func->getMetadata("kernel_arg_access_qual");
476 if (Node && ArgNo < Node->getNumOperands())
477 AccQual = cast<MDString>(Node->getOperand(ArgNo))->getString();
478
479 StringRef Name;
480 Node = Func->getMetadata("kernel_arg_name");
481 if (Node && ArgNo < Node->getNumOperands())
482 Name = cast<MDString>(Node->getOperand(ArgNo))->getString();
483
484 StringRef TypeName;
485 Node = Func->getMetadata("kernel_arg_type");
486 if (Node && ArgNo < Node->getNumOperands())
487 TypeName = cast<MDString>(Node->getOperand(ArgNo))->getString();
488
489 emitKernelArg(Func->getParent()->getDataLayout(), Arg.getType(),
490 getValueKind(Arg.getType(), TypeQual, BaseTypeName), TypeQual,
491 BaseTypeName, AccQual, Name, TypeName);
492}
493
494void MetadataStreamer::emitKernelArg(const DataLayout &DL, Type *Ty,
495 ValueKind ValueKind, StringRef TypeQual,
496 StringRef BaseTypeName, StringRef AccQual,
497 StringRef Name, StringRef TypeName) {
498 CodeObjectMetadata.mKernels.back().mArgs.push_back(Kernel::Arg::Metadata());
499 auto &Arg = CodeObjectMetadata.mKernels.back().mArgs.back();
500
501 Arg.mSize = DL.getTypeAllocSize(Ty);
502 Arg.mAlign = DL.getABITypeAlignment(Ty);
503 Arg.mValueKind = ValueKind;
504 Arg.mValueType = getValueType(Ty, BaseTypeName);
505
506 if (auto PtrTy = dyn_cast<PointerType>(Ty)) {
507 auto ElTy = PtrTy->getElementType();
508 if (PtrTy->getAddressSpace() == AMDGPUAS::LOCAL_ADDRESS && ElTy->isSized())
509 Arg.mPointeeAlign = DL.getABITypeAlignment(ElTy);
510 }
511
512 Arg.mAccQual = getAccessQualifier(AccQual);
513
514 if (auto PtrTy = dyn_cast<PointerType>(Ty))
515 Arg.mAddrSpaceQual = getAddressSpaceQualifer(PtrTy->getAddressSpace());
516
517 SmallVector<StringRef, 1> SplitTypeQuals;
518 TypeQual.split(SplitTypeQuals, " ", -1, false);
519 for (StringRef Key : SplitTypeQuals) {
520 auto P = StringSwitch<bool*>(Key)
521 .Case("const", &Arg.mIsConst)
522 .Case("pipe", &Arg.mIsPipe)
523 .Case("restrict", &Arg.mIsRestrict)
524 .Case("volatile", &Arg.mIsVolatile)
525 .Default(nullptr);
526 if (P)
527 *P = true;
528 }
529
530 Arg.mName = Name;
531 Arg.mTypeName = TypeName;
532}
533
534void MetadataStreamer::begin(const FeatureBitset &Features, const Module &Mod) {
535 emitVersion();
536 emitIsa(Features);
537 emitPrintf(Mod);
538}
539
540void MetadataStreamer::emitKernel(const Function &Func) {
541 if (Func.getCallingConv() != CallingConv::AMDGPU_KERNEL)
542 return;
543
544 CodeObjectMetadata.mKernels.push_back(Kernel::Metadata());
545 auto &Kernel = CodeObjectMetadata.mKernels.back();
546
547 Kernel.mName = Func.getName();
548 emitKernelLanguage(Func);
549 emitKernelAttrs(Func);
550 emitKernelArgs(Func);
551}
552
553ErrorOr<std::string> MetadataStreamer::toYamlString() {
554 std::string YamlString;
555 if (auto Error = Metadata::toYamlString(CodeObjectMetadata, YamlString))
556 return Error;
557
558 if (DumpCodeObjectMetadata)
559 dump(YamlString);
560 if (VerifyCodeObjectMetadata)
561 verify(YamlString);
562
563 return YamlString;
564}
565
566ErrorOr<std::string> MetadataStreamer::toYamlString(
567 const FeatureBitset &Features, StringRef YamlString) {
568 if (auto Error = Metadata::fromYamlString(YamlString, CodeObjectMetadata))
569 return Error;
570
571 emitIsa(Features);
572 return toYamlString();
573}
574
575} // end namespace CodeObject
576} // end namespace AMDGPU
577} // end namespace llvm