[llvm] [llvm-objdump] Add support for HIP offload bundles (PR #114834)

[Joseph Huber via llvm-commits]

================
@@ -64,6 +67,57 @@ void llvm::dumpOffloadBinary(const ObjectFile &O) {
   // Print out all the binaries that are contained in this buffer.
   for (uint64_t I = 0, E = Binaries.size(); I != E; ++I)
     printBinary(*Binaries[I].getBinary(), I);
+
+  dumpOffloadBundleFatBinary(O, ArchName);
+}
+
+// Given an Object file, collect all Bundles of FatBin Binaries
+// and dump them into Code Object files
+// if -d is specified, disassemble the Code Object Files
+// if -arch=-name is specified, only dump the Entries that match the target arch
+void llvm::dumpOffloadBundleFatBinary(const ObjectFile &O,
+                                      std::string ArchName) {
+  assert((O.isELF() || O.isCOFF()) && "Invalid file type");
+  // Collect all Bundles and their Entries ....
+  SmallVector<llvm::object::OffloadBundleFatBin> FoundBundles;
+  SmallVector<OffloadBundleEntry> FoundEntries;
+
+  if (Error Err = llvm::object::extractOffloadBundleFatBinary(O, FoundBundles))
+    reportError(O.getFileName(), "while extracting offload FatBin bundles: " +
+                                     toString(std::move(Err)));
+
+  // Now filter based on if arch-name is specified
+  SmallVectorImpl<llvm::object::OffloadBundleFatBin>::iterator BundleIter =
+      FoundBundles.begin();
+  for (uint64_t bundle_num = 0; bundle_num < FoundBundles.size();
----------------
jhuber6 wrote:

Here's a similar usage https://github.com/llvm/llvm-project/blob/9ffab5637c803d1ff8aaf94972b9d40108d450c9/clang/tools/clang-linker-wrapper/ClangLinkerWrapper.cpp#L1260. It's just an easier way to do a range-based for loop while still having access to the index.

https://github.com/llvm/llvm-project/pull/114834