1 //===- OffloadEmitter.cpp -------------------------------------------------===// 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 #include "llvm/Object/OffloadBinary.h" 10 #include "llvm/ObjectYAML/OffloadYAML.h" 11 #include "llvm/ObjectYAML/yaml2obj.h" 12 #include "llvm/Support/Error.h" 13 #include "llvm/Support/raw_ostream.h" 14 15 using namespace llvm; 16 using namespace OffloadYAML; 17 18 namespace llvm { 19 namespace yaml { 20 21 bool yaml2offload(Binary &Doc, raw_ostream &Out, ErrorHandler EH) { 22 for (const auto &Member : Doc.Members) { 23 object::OffloadBinary::OffloadingImage Image{}; 24 if (Member.ImageKind) 25 Image.TheImageKind = *Member.ImageKind; 26 if (Member.OffloadKind) 27 Image.TheOffloadKind = *Member.OffloadKind; 28 if (Member.Flags) 29 Image.Flags = *Member.Flags; 30 31 StringMap<StringRef> &StringData = Image.StringData; 32 if (Member.StringEntries) { 33 for (const auto &Entry : *Member.StringEntries) { 34 StringData[Entry.Key] = Entry.Value; 35 } 36 } 37 38 SmallVector<char, 1024> Data; 39 raw_svector_ostream OS(Data); 40 if (Member.Content) 41 Member.Content->writeAsBinary(OS); 42 Image.Image = MemoryBuffer::getMemBufferCopy(OS.str()); 43 44 std::unique_ptr<MemoryBuffer> Binary = object::OffloadBinary::write(Image); 45 46 // Copy the data to a new buffer so we can modify the bytes directly. 47 SmallVector<char> NewBuffer; 48 std::copy(Binary->getBufferStart(), Binary->getBufferEnd(), 49 std::back_inserter(NewBuffer)); 50 auto *TheHeader = 51 reinterpret_cast<object::OffloadBinary::Header *>(&NewBuffer[0]); 52 if (Doc.Version) 53 TheHeader->Version = *Doc.Version; 54 if (Doc.Size) 55 TheHeader->Size = *Doc.Size; 56 if (Doc.EntryOffset) 57 TheHeader->EntryOffset = *Doc.EntryOffset; 58 if (Doc.EntrySize) 59 TheHeader->EntrySize = *Doc.EntrySize; 60 61 Out.write(NewBuffer.begin(), NewBuffer.size()); 62 } 63 64 return true; 65 } 66 67 } // namespace yaml 68 } // namespace llvm 69