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

[Joseph Huber via llvm-commits]

================
@@ -64,6 +69,58 @@ 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,
+                                      // SALINAS std::string ArchName) {
+                                      StringRef ArchName) {
+  assert((O.isELF() || O.isCOFF()) && "Invalid file type");
+  // Collect all Bundles and their Entries ....
+  SmallVector<llvm::object::OffloadBundleFatBin> FoundBundles;
+  SmallVector<llvm::object::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();
+       bundle_num++) {
+    if (!ArchName.empty())
+      FoundEntries = BundleIter->entryIDContains(ArchName);
+    else
+      FoundEntries = BundleIter->getEntries();
+
+    // now we have a list of Found Entries .... dump them
+    SmallVectorImpl<llvm::object::OffloadBundleEntry>::iterator FoundIter =
+        FoundEntries.begin();
+    for (int64_t entry_num = 0; entry_num < FoundEntries.size(); entry_num++) {
+      // create file name for this object file:  <source-filename>:<Bundle
+      // Number>.<EntryID>
+      std::string str = BundleIter->getFileName().str() + ":" +
+                        itostr(bundle_num) + "." + FoundIter->ID.str();
+      StringRef OutputFilename = StringRef(str);
+      if (Error Err = object::extractCodeObject(
+              O, FoundIter->Offset, FoundIter->Size, OutputFilename))
+        reportError(O.getFileName(),
+                    "while extracting offload Bundle Entries: " +
+                        toString(std::move(Err)));
+
+      // TODO: If -d was specified, disasseble the Code Object too
+
+      ++FoundIter;
+    } // end of for found_entries loop
----------------
jhuber6 wrote:

Useless comments.

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