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