Line data Source code
1 : //===--- AMDGPUMetadata.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 : /// AMDGPU metadata definitions and in-memory representations.
12 : ///
13 : //
14 : //===----------------------------------------------------------------------===//
15 :
16 : #include "llvm/ADT/Twine.h"
17 : #include "llvm/Support/AMDGPUMetadata.h"
18 : #include "llvm/Support/YAMLTraits.h"
19 :
20 : using namespace llvm::AMDGPU;
21 : using namespace llvm::AMDGPU::HSAMD;
22 :
23 : LLVM_YAML_IS_SEQUENCE_VECTOR(Kernel::Arg::Metadata)
24 : LLVM_YAML_IS_SEQUENCE_VECTOR(Kernel::Metadata)
25 :
26 : namespace llvm {
27 : namespace yaml {
28 :
29 : template <>
30 : struct ScalarEnumerationTraits<AccessQualifier> {
31 1014 : static void enumeration(IO &YIO, AccessQualifier &EN) {
32 1014 : YIO.enumCase(EN, "Default", AccessQualifier::Default);
33 1014 : YIO.enumCase(EN, "ReadOnly", AccessQualifier::ReadOnly);
34 1014 : YIO.enumCase(EN, "WriteOnly", AccessQualifier::WriteOnly);
35 1014 : YIO.enumCase(EN, "ReadWrite", AccessQualifier::ReadWrite);
36 1014 : }
37 : };
38 :
39 : template <>
40 : struct ScalarEnumerationTraits<AddressSpaceQualifier> {
41 8094 : static void enumeration(IO &YIO, AddressSpaceQualifier &EN) {
42 8094 : YIO.enumCase(EN, "Private", AddressSpaceQualifier::Private);
43 8094 : YIO.enumCase(EN, "Global", AddressSpaceQualifier::Global);
44 8094 : YIO.enumCase(EN, "Constant", AddressSpaceQualifier::Constant);
45 8094 : YIO.enumCase(EN, "Local", AddressSpaceQualifier::Local);
46 8094 : YIO.enumCase(EN, "Generic", AddressSpaceQualifier::Generic);
47 8094 : YIO.enumCase(EN, "Region", AddressSpaceQualifier::Region);
48 8094 : }
49 : };
50 :
51 : template <>
52 : struct ScalarEnumerationTraits<ValueKind> {
53 13736 : static void enumeration(IO &YIO, ValueKind &EN) {
54 13736 : YIO.enumCase(EN, "ByValue", ValueKind::ByValue);
55 13736 : YIO.enumCase(EN, "GlobalBuffer", ValueKind::GlobalBuffer);
56 13736 : YIO.enumCase(EN, "DynamicSharedPointer", ValueKind::DynamicSharedPointer);
57 13736 : YIO.enumCase(EN, "Sampler", ValueKind::Sampler);
58 13736 : YIO.enumCase(EN, "Image", ValueKind::Image);
59 13736 : YIO.enumCase(EN, "Pipe", ValueKind::Pipe);
60 13736 : YIO.enumCase(EN, "Queue", ValueKind::Queue);
61 13736 : YIO.enumCase(EN, "HiddenGlobalOffsetX", ValueKind::HiddenGlobalOffsetX);
62 13736 : YIO.enumCase(EN, "HiddenGlobalOffsetY", ValueKind::HiddenGlobalOffsetY);
63 13736 : YIO.enumCase(EN, "HiddenGlobalOffsetZ", ValueKind::HiddenGlobalOffsetZ);
64 13736 : YIO.enumCase(EN, "HiddenNone", ValueKind::HiddenNone);
65 13736 : YIO.enumCase(EN, "HiddenPrintfBuffer", ValueKind::HiddenPrintfBuffer);
66 13736 : YIO.enumCase(EN, "HiddenDefaultQueue", ValueKind::HiddenDefaultQueue);
67 13736 : YIO.enumCase(EN, "HiddenCompletionAction",
68 : ValueKind::HiddenCompletionAction);
69 13736 : }
70 : };
71 :
72 : template <>
73 : struct ScalarEnumerationTraits<ValueType> {
74 13736 : static void enumeration(IO &YIO, ValueType &EN) {
75 13736 : YIO.enumCase(EN, "Struct", ValueType::Struct);
76 13736 : YIO.enumCase(EN, "I8", ValueType::I8);
77 13736 : YIO.enumCase(EN, "U8", ValueType::U8);
78 13736 : YIO.enumCase(EN, "I16", ValueType::I16);
79 13736 : YIO.enumCase(EN, "U16", ValueType::U16);
80 13736 : YIO.enumCase(EN, "F16", ValueType::F16);
81 13736 : YIO.enumCase(EN, "I32", ValueType::I32);
82 13736 : YIO.enumCase(EN, "U32", ValueType::U32);
83 13736 : YIO.enumCase(EN, "F32", ValueType::F32);
84 13736 : YIO.enumCase(EN, "I64", ValueType::I64);
85 13736 : YIO.enumCase(EN, "U64", ValueType::U64);
86 13736 : YIO.enumCase(EN, "F64", ValueType::F64);
87 13736 : }
88 : };
89 :
90 : template <>
91 : struct MappingTraits<Kernel::Attrs::Metadata> {
92 186 : static void mapping(IO &YIO, Kernel::Attrs::Metadata &MD) {
93 : YIO.mapOptional(Kernel::Attrs::Key::ReqdWorkGroupSize,
94 372 : MD.mReqdWorkGroupSize, std::vector<uint32_t>());
95 : YIO.mapOptional(Kernel::Attrs::Key::WorkGroupSizeHint,
96 186 : MD.mWorkGroupSizeHint, std::vector<uint32_t>());
97 : YIO.mapOptional(Kernel::Attrs::Key::VecTypeHint,
98 372 : MD.mVecTypeHint, std::string());
99 186 : YIO.mapOptional(Kernel::Attrs::Key::RuntimeHandle, MD.mRuntimeHandle,
100 186 : std::string());
101 186 : }
102 : };
103 :
104 : template <>
105 : struct MappingTraits<Kernel::Arg::Metadata> {
106 13736 : static void mapping(IO &YIO, Kernel::Arg::Metadata &MD) {
107 27472 : YIO.mapOptional(Kernel::Arg::Key::Name, MD.mName, std::string());
108 13736 : YIO.mapOptional(Kernel::Arg::Key::TypeName, MD.mTypeName, std::string());
109 13736 : YIO.mapRequired(Kernel::Arg::Key::Size, MD.mSize);
110 13736 : YIO.mapRequired(Kernel::Arg::Key::Align, MD.mAlign);
111 13736 : YIO.mapRequired(Kernel::Arg::Key::ValueKind, MD.mValueKind);
112 13736 : YIO.mapRequired(Kernel::Arg::Key::ValueType, MD.mValueType);
113 13736 : YIO.mapOptional(Kernel::Arg::Key::PointeeAlign, MD.mPointeeAlign,
114 13736 : uint32_t(0));
115 13736 : YIO.mapOptional(Kernel::Arg::Key::AddrSpaceQual, MD.mAddrSpaceQual,
116 13736 : AddressSpaceQualifier::Unknown);
117 13736 : YIO.mapOptional(Kernel::Arg::Key::AccQual, MD.mAccQual,
118 13736 : AccessQualifier::Unknown);
119 13736 : YIO.mapOptional(Kernel::Arg::Key::ActualAccQual, MD.mActualAccQual,
120 13736 : AccessQualifier::Unknown);
121 13736 : YIO.mapOptional(Kernel::Arg::Key::IsConst, MD.mIsConst, false);
122 13736 : YIO.mapOptional(Kernel::Arg::Key::IsRestrict, MD.mIsRestrict, false);
123 13736 : YIO.mapOptional(Kernel::Arg::Key::IsVolatile, MD.mIsVolatile, false);
124 13736 : YIO.mapOptional(Kernel::Arg::Key::IsPipe, MD.mIsPipe, false);
125 13736 : }
126 : };
127 :
128 : template <>
129 : struct MappingTraits<Kernel::CodeProps::Metadata> {
130 5105 : static void mapping(IO &YIO, Kernel::CodeProps::Metadata &MD) {
131 : YIO.mapRequired(Kernel::CodeProps::Key::KernargSegmentSize,
132 5105 : MD.mKernargSegmentSize);
133 : YIO.mapRequired(Kernel::CodeProps::Key::GroupSegmentFixedSize,
134 5105 : MD.mGroupSegmentFixedSize);
135 : YIO.mapRequired(Kernel::CodeProps::Key::PrivateSegmentFixedSize,
136 5105 : MD.mPrivateSegmentFixedSize);
137 : YIO.mapRequired(Kernel::CodeProps::Key::KernargSegmentAlign,
138 5105 : MD.mKernargSegmentAlign);
139 : YIO.mapRequired(Kernel::CodeProps::Key::WavefrontSize,
140 5105 : MD.mWavefrontSize);
141 : YIO.mapOptional(Kernel::CodeProps::Key::NumSGPRs,
142 5105 : MD.mNumSGPRs, uint16_t(0));
143 : YIO.mapOptional(Kernel::CodeProps::Key::NumVGPRs,
144 5105 : MD.mNumVGPRs, uint16_t(0));
145 : YIO.mapOptional(Kernel::CodeProps::Key::MaxFlatWorkGroupSize,
146 5105 : MD.mMaxFlatWorkGroupSize, uint32_t(0));
147 : YIO.mapOptional(Kernel::CodeProps::Key::IsDynamicCallStack,
148 5105 : MD.mIsDynamicCallStack, false);
149 : YIO.mapOptional(Kernel::CodeProps::Key::IsXNACKEnabled,
150 5105 : MD.mIsXNACKEnabled, false);
151 : YIO.mapOptional(Kernel::CodeProps::Key::NumSpilledSGPRs,
152 5105 : MD.mNumSpilledSGPRs, uint16_t(0));
153 : YIO.mapOptional(Kernel::CodeProps::Key::NumSpilledVGPRs,
154 5105 : MD.mNumSpilledVGPRs, uint16_t(0));
155 5105 : }
156 : };
157 :
158 : template <>
159 : struct MappingTraits<Kernel::DebugProps::Metadata> {
160 12 : static void mapping(IO &YIO, Kernel::DebugProps::Metadata &MD) {
161 : YIO.mapOptional(Kernel::DebugProps::Key::DebuggerABIVersion,
162 12 : MD.mDebuggerABIVersion, std::vector<uint32_t>());
163 : YIO.mapOptional(Kernel::DebugProps::Key::ReservedNumVGPRs,
164 12 : MD.mReservedNumVGPRs, uint16_t(0));
165 : YIO.mapOptional(Kernel::DebugProps::Key::ReservedFirstVGPR,
166 12 : MD.mReservedFirstVGPR, uint16_t(-1));
167 : YIO.mapOptional(Kernel::DebugProps::Key::PrivateSegmentBufferSGPR,
168 12 : MD.mPrivateSegmentBufferSGPR, uint16_t(-1));
169 : YIO.mapOptional(Kernel::DebugProps::Key::WavefrontPrivateSegmentOffsetSGPR,
170 12 : MD.mWavefrontPrivateSegmentOffsetSGPR, uint16_t(-1));
171 12 : }
172 : };
173 :
174 : template <>
175 : struct MappingTraits<Kernel::Metadata> {
176 5124 : static void mapping(IO &YIO, Kernel::Metadata &MD) {
177 5124 : YIO.mapRequired(Kernel::Key::Name, MD.mName);
178 5124 : YIO.mapRequired(Kernel::Key::SymbolName, MD.mSymbolName);
179 10248 : YIO.mapOptional(Kernel::Key::Language, MD.mLanguage, std::string());
180 5124 : YIO.mapOptional(Kernel::Key::LanguageVersion, MD.mLanguageVersion,
181 5124 : std::vector<uint32_t>());
182 4971 : if (!MD.mAttrs.empty() || !YIO.outputting())
183 284 : YIO.mapOptional(Kernel::Key::Attrs, MD.mAttrs);
184 5124 : if (!MD.mArgs.empty() || !YIO.outputting())
185 4392 : YIO.mapOptional(Kernel::Key::Args, MD.mArgs);
186 5124 : if (!MD.mCodeProps.empty() || !YIO.outputting())
187 : YIO.mapOptional(Kernel::Key::CodeProps, MD.mCodeProps);
188 5124 : if (!MD.mDebugProps.empty() || !YIO.outputting())
189 140 : YIO.mapOptional(Kernel::Key::DebugProps, MD.mDebugProps);
190 5124 : }
191 : };
192 :
193 : template <>
194 : struct MappingTraits<HSAMD::Metadata> {
195 692 : static void mapping(IO &YIO, HSAMD::Metadata &MD) {
196 692 : YIO.mapRequired(Key::Version, MD.mVersion);
197 692 : YIO.mapOptional(Key::Printf, MD.mPrintf, std::vector<std::string>());
198 692 : if (!MD.mKernels.empty() || !YIO.outputting())
199 585 : YIO.mapOptional(Key::Kernels, MD.mKernels);
200 692 : }
201 : };
202 :
203 : } // end namespace yaml
204 :
205 : namespace AMDGPU {
206 : namespace HSAMD {
207 :
208 26 : std::error_code fromString(std::string String, Metadata &HSAMetadata) {
209 52 : yaml::Input YamlInput(String);
210 : YamlInput >> HSAMetadata;
211 26 : return YamlInput.error();
212 : }
213 :
214 666 : std::error_code toString(Metadata HSAMetadata, std::string &String) {
215 666 : raw_string_ostream YamlStream(String);
216 666 : yaml::Output YamlOutput(YamlStream, nullptr, std::numeric_limits<int>::max());
217 666 : YamlOutput << HSAMetadata;
218 666 : return std::error_code();
219 : }
220 :
221 : } // end namespace HSAMD
222 :
223 : namespace PALMD {
224 :
225 43 : std::error_code toString(const Metadata &PALMetadata, std::string &String) {
226 43 : raw_string_ostream Stream(String);
227 515 : for (auto I = PALMetadata.begin(), E = PALMetadata.end(); I != E; ++I) {
228 472 : Stream << Twine(I == PALMetadata.begin() ? " 0x" : ",0x");
229 472 : Stream << Twine::utohexstr(*I);
230 : }
231 : Stream.flush();
232 43 : return std::error_code();
233 : }
234 :
235 : } // end namespace PALMD
236 : } // end namespace AMDGPU
237 : } // end namespace llvm
|
