LCOV - code coverage report
Current view: top level - lib/Support - AMDGPUMetadata.cpp (source / functions) Hit Total Coverage
Test: llvm-toolchain.info Lines: 126 126 100.0 %
Date: 2018-09-23 13:06:45 Functions: 13 13 100.0 %
Legend: Lines: hit not hit

          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        8088 :   static void enumeration(IO &YIO, AddressSpaceQualifier &EN) {
      42        8088 :     YIO.enumCase(EN, "Private", AddressSpaceQualifier::Private);
      43        8088 :     YIO.enumCase(EN, "Global", AddressSpaceQualifier::Global);
      44        8088 :     YIO.enumCase(EN, "Constant", AddressSpaceQualifier::Constant);
      45        8088 :     YIO.enumCase(EN, "Local", AddressSpaceQualifier::Local);
      46        8088 :     YIO.enumCase(EN, "Generic", AddressSpaceQualifier::Generic);
      47        8088 :     YIO.enumCase(EN, "Region", AddressSpaceQualifier::Region);
      48        8088 :   }
      49             : };
      50             : 
      51             : template <>
      52             : struct ScalarEnumerationTraits<ValueKind> {
      53       13730 :   static void enumeration(IO &YIO, ValueKind &EN) {
      54       13730 :     YIO.enumCase(EN, "ByValue", ValueKind::ByValue);
      55       13730 :     YIO.enumCase(EN, "GlobalBuffer", ValueKind::GlobalBuffer);
      56       13730 :     YIO.enumCase(EN, "DynamicSharedPointer", ValueKind::DynamicSharedPointer);
      57       13730 :     YIO.enumCase(EN, "Sampler", ValueKind::Sampler);
      58       13730 :     YIO.enumCase(EN, "Image", ValueKind::Image);
      59       13730 :     YIO.enumCase(EN, "Pipe", ValueKind::Pipe);
      60       13730 :     YIO.enumCase(EN, "Queue", ValueKind::Queue);
      61       13730 :     YIO.enumCase(EN, "HiddenGlobalOffsetX", ValueKind::HiddenGlobalOffsetX);
      62       13730 :     YIO.enumCase(EN, "HiddenGlobalOffsetY", ValueKind::HiddenGlobalOffsetY);
      63       13730 :     YIO.enumCase(EN, "HiddenGlobalOffsetZ", ValueKind::HiddenGlobalOffsetZ);
      64       13730 :     YIO.enumCase(EN, "HiddenNone", ValueKind::HiddenNone);
      65       13730 :     YIO.enumCase(EN, "HiddenPrintfBuffer", ValueKind::HiddenPrintfBuffer);
      66       13730 :     YIO.enumCase(EN, "HiddenDefaultQueue", ValueKind::HiddenDefaultQueue);
      67       13730 :     YIO.enumCase(EN, "HiddenCompletionAction",
      68             :                  ValueKind::HiddenCompletionAction);
      69       13730 :   }
      70             : };
      71             : 
      72             : template <>
      73             : struct ScalarEnumerationTraits<ValueType> {
      74       13730 :   static void enumeration(IO &YIO, ValueType &EN) {
      75       13730 :     YIO.enumCase(EN, "Struct", ValueType::Struct);
      76       13730 :     YIO.enumCase(EN, "I8", ValueType::I8);
      77       13730 :     YIO.enumCase(EN, "U8", ValueType::U8);
      78       13730 :     YIO.enumCase(EN, "I16", ValueType::I16);
      79       13730 :     YIO.enumCase(EN, "U16", ValueType::U16);
      80       13730 :     YIO.enumCase(EN, "F16", ValueType::F16);
      81       13730 :     YIO.enumCase(EN, "I32", ValueType::I32);
      82       13730 :     YIO.enumCase(EN, "U32", ValueType::U32);
      83       13730 :     YIO.enumCase(EN, "F32", ValueType::F32);
      84       13730 :     YIO.enumCase(EN, "I64", ValueType::I64);
      85       13730 :     YIO.enumCase(EN, "U64", ValueType::U64);
      86       13730 :     YIO.enumCase(EN, "F64", ValueType::F64);
      87       13730 :   }
      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       13730 :   static void mapping(IO &YIO, Kernel::Arg::Metadata &MD) {
     107       27460 :     YIO.mapOptional(Kernel::Arg::Key::Name, MD.mName, std::string());
     108       13730 :     YIO.mapOptional(Kernel::Arg::Key::TypeName, MD.mTypeName, std::string());
     109       13730 :     YIO.mapRequired(Kernel::Arg::Key::Size, MD.mSize);
     110       13730 :     YIO.mapRequired(Kernel::Arg::Key::Align, MD.mAlign);
     111       13730 :     YIO.mapRequired(Kernel::Arg::Key::ValueKind, MD.mValueKind);
     112       13730 :     YIO.mapRequired(Kernel::Arg::Key::ValueType, MD.mValueType);
     113       13730 :     YIO.mapOptional(Kernel::Arg::Key::PointeeAlign, MD.mPointeeAlign,
     114       13730 :                     uint32_t(0));
     115       13730 :     YIO.mapOptional(Kernel::Arg::Key::AddrSpaceQual, MD.mAddrSpaceQual,
     116       13730 :                     AddressSpaceQualifier::Unknown);
     117       13730 :     YIO.mapOptional(Kernel::Arg::Key::AccQual, MD.mAccQual,
     118       13730 :                     AccessQualifier::Unknown);
     119       13730 :     YIO.mapOptional(Kernel::Arg::Key::ActualAccQual, MD.mActualAccQual,
     120       13730 :                     AccessQualifier::Unknown);
     121       13730 :     YIO.mapOptional(Kernel::Arg::Key::IsConst, MD.mIsConst, false);
     122       13730 :     YIO.mapOptional(Kernel::Arg::Key::IsRestrict, MD.mIsRestrict, false);
     123       13730 :     YIO.mapOptional(Kernel::Arg::Key::IsVolatile, MD.mIsVolatile, false);
     124       13730 :     YIO.mapOptional(Kernel::Arg::Key::IsPipe, MD.mIsPipe, false);
     125       13730 :   }
     126             : };
     127             : 
     128             : template <>
     129             : struct MappingTraits<Kernel::CodeProps::Metadata> {
     130        5099 :   static void mapping(IO &YIO, Kernel::CodeProps::Metadata &MD) {
     131             :     YIO.mapRequired(Kernel::CodeProps::Key::KernargSegmentSize,
     132        5099 :                     MD.mKernargSegmentSize);
     133             :     YIO.mapRequired(Kernel::CodeProps::Key::GroupSegmentFixedSize,
     134        5099 :                     MD.mGroupSegmentFixedSize);
     135             :     YIO.mapRequired(Kernel::CodeProps::Key::PrivateSegmentFixedSize,
     136        5099 :                     MD.mPrivateSegmentFixedSize);
     137             :     YIO.mapRequired(Kernel::CodeProps::Key::KernargSegmentAlign,
     138        5099 :                     MD.mKernargSegmentAlign);
     139             :     YIO.mapRequired(Kernel::CodeProps::Key::WavefrontSize,
     140        5099 :                     MD.mWavefrontSize);
     141             :     YIO.mapOptional(Kernel::CodeProps::Key::NumSGPRs,
     142        5099 :                     MD.mNumSGPRs, uint16_t(0));
     143             :     YIO.mapOptional(Kernel::CodeProps::Key::NumVGPRs,
     144        5099 :                     MD.mNumVGPRs, uint16_t(0));
     145             :     YIO.mapOptional(Kernel::CodeProps::Key::MaxFlatWorkGroupSize,
     146        5099 :                     MD.mMaxFlatWorkGroupSize, uint32_t(0));
     147             :     YIO.mapOptional(Kernel::CodeProps::Key::IsDynamicCallStack,
     148        5099 :                     MD.mIsDynamicCallStack, false);
     149             :     YIO.mapOptional(Kernel::CodeProps::Key::IsXNACKEnabled,
     150        5099 :                     MD.mIsXNACKEnabled, false);
     151             :     YIO.mapOptional(Kernel::CodeProps::Key::NumSpilledSGPRs,
     152        5099 :                     MD.mNumSpilledSGPRs, uint16_t(0));
     153             :     YIO.mapOptional(Kernel::CodeProps::Key::NumSpilledVGPRs,
     154        5099 :                     MD.mNumSpilledVGPRs, uint16_t(0));
     155        5099 :   }
     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        5118 :   static void mapping(IO &YIO, Kernel::Metadata &MD) {
     177        5118 :     YIO.mapRequired(Kernel::Key::Name, MD.mName);
     178        5118 :     YIO.mapRequired(Kernel::Key::SymbolName, MD.mSymbolName);
     179       10236 :     YIO.mapOptional(Kernel::Key::Language, MD.mLanguage, std::string());
     180        5118 :     YIO.mapOptional(Kernel::Key::LanguageVersion, MD.mLanguageVersion,
     181        5118 :                     std::vector<uint32_t>());
     182        4965 :     if (!MD.mAttrs.empty() || !YIO.outputting())
     183         284 :       YIO.mapOptional(Kernel::Key::Attrs, MD.mAttrs);
     184        5118 :     if (!MD.mArgs.empty() || !YIO.outputting())
     185        4386 :       YIO.mapOptional(Kernel::Key::Args, MD.mArgs);
     186        5118 :     if (!MD.mCodeProps.empty() || !YIO.outputting())
     187             :       YIO.mapOptional(Kernel::Key::CodeProps, MD.mCodeProps);
     188        5118 :     if (!MD.mDebugProps.empty() || !YIO.outputting())
     189         140 :       YIO.mapOptional(Kernel::Key::DebugProps, MD.mDebugProps);
     190        5118 :   }
     191             : };
     192             : 
     193             : template <>
     194             : struct MappingTraits<HSAMD::Metadata> {
     195         684 :   static void mapping(IO &YIO, HSAMD::Metadata &MD) {
     196         684 :     YIO.mapRequired(Key::Version, MD.mVersion);
     197         684 :     YIO.mapOptional(Key::Printf, MD.mPrintf, std::vector<std::string>());
     198         684 :     if (!MD.mKernels.empty() || !YIO.outputting())
     199         579 :       YIO.mapOptional(Key::Kernels, MD.mKernels);
     200         684 :   }
     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         658 : std::error_code toString(Metadata HSAMetadata, std::string &String) {
     215         658 :   raw_string_ostream YamlStream(String);
     216         658 :   yaml::Output YamlOutput(YamlStream, nullptr, std::numeric_limits<int>::max());
     217         658 :   YamlOutput << HSAMetadata;
     218         658 :   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

Generated by: LCOV version 1.13