wip

Author: ZuseZ4

compiler/rustc_llvm/llvm-wrapper/RustWrapper.cpp

@@ -39,6 +39,17 @@
 #include "llvm/Transforms/Utils/ValueMapper.h"
 #include <iostream>
 
+#include "llvm/ADT/StringExtras.h"
+#include "llvm/Object/ArchiveWriter.h"
+#include "llvm/Object/OffloadBinary.h"
+#include "llvm/Support/CommandLine.h"
+#include "llvm/Support/FileOutputBuffer.h"
+#include "llvm/Support/MemoryBuffer.h"
+#include "llvm/Support/Path.h"
+#include "llvm/Support/Signals.h"
+//#include "llvm/Support/StringSaver.h"
+//#include "llvm/Support/WithColor.h"
+
 // for raw `write` in the bad-alloc handler
 #ifdef _MSC_VER
 #include <io.h>
@@ -57,6 +68,55 @@ using namespace llvm;
 using namespace llvm::sys;
 using namespace llvm::object;
 
+static Error writeFile(StringRef Filename, StringRef Data) {
+  Expected<std::unique_ptr<FileOutputBuffer>> OutputOrErr =
+      FileOutputBuffer::create(Filename, Data.size());
+  if (!OutputOrErr)
+    return OutputOrErr.takeError();
+  std::unique_ptr<FileOutputBuffer> Output = std::move(*OutputOrErr);
+  llvm::copy(Data, Output->getBufferStart());
+  if (Error E = Output->commit())
+    return E;
+  return Error::success();
+}
+
+extern "C" bool rustBundleImages() {
+  SmallVector<char, 1024> BinaryData;
+  raw_svector_ostream OS(BinaryData);
+  BumpPtrAllocator Alloc;
+  StringSaver Saver(Alloc);
+
+  DenseMap<StringRef, StringRef> Args;
+  Args["triple"] = "amdgcn-amd-amdhsa"; // required
+  Args["arch"]   = "gfx90a"; // optional, but from rustc?
+
+  {
+    OffloadBinary::OffloadingImage ImageBinary{};
+
+    llvm::ErrorOr<std::unique_ptr<llvm::MemoryBuffer>> ObjectOrErr =
+        llvm::MemoryBuffer::getFileOrSTDIN("device.bc");
+    if (std::error_code EC = ObjectOrErr.getError())
+      return false;
+
+    // Clang uses the '.o' suffix for LTO bitcode.
+    ImageBinary.TheImageKind = object::IMG_Bitcode;
+    ImageBinary.Image = std::move(*ObjectOrErr);
+    ImageBinary.TheOffloadKind = object::OFK_OpenMP;
+    for (const auto &[Key, Value] : Args) {
+      ImageBinary.StringData[Key] = Value;
+    }
+    llvm::SmallString<0> Buffer = OffloadBinary::write(ImageBinary);
+    if (Buffer.size() % OffloadBinary::getAlignment() != 0)
+      // "Offload binary has invalid size alignment");
+      return false;
+    OS << Buffer;
+  }
+  if (Error E = writeFile("host.out",
+                          StringRef(BinaryData.begin(), BinaryData.size())))
+    return false;
+  return true;
+}
+
 // This opcode is an LLVM detail that could hypothetically change (?), so
 // verify that the hard-coded value in `dwarf_const.rs` still agrees with LLVM.
 static_assert(dwarf::DW_OP_LLVM_fragment == 0x1000);