Add aie coredump support using hw ctx (#9368)

* Add aie coredump support using hw ctx

Signed-off-by: Rahul Bramandlapalli <rbramand@amd.com>

* Fix clang-tidy warnings

Signed-off-by: Rahul Bramandlapalli <rbramand@amd.com>

---------

Signed-off-by: Rahul Bramandlapalli <rbramand@amd.com>

Author: rbramand-xilinx

src/runtime_src/core/common/api/xrt_hw_context.cpp

@@ -9,6 +9,7 @@
 
 #include "core/common/config_reader.h"
 #include "core/common/message.h"
+#include "core/common/query_requests.h"
 #include "core/common/time.h"
 #include "core/common/utils.h"
 #include "core/include/xrt/xrt_hw_context.h"
@@ -494,6 +495,21 @@ class hw_context_impl : public std::enable_shared_from_this<hw_context_impl>
     msg.append(dump_file_name);
     xrt_core::message::send(xrt_core::message::severity_level::debug, "xrt_hw_context", msg);
   }
+
+  std::vector<char>
+  get_aie_coredump() const
+  {
+    try {
+      return xrt_core::device_query<xrt_core::query::aie_coredump>(m_core_device,
+                                                                   m_hdl->get_slotidx());
+    }
+    catch (const xrt_core::query::no_such_key&) {
+      throw std::runtime_error("AIE coredump is not supported on this platform");
+    }
+    catch (const std::exception& e) {
+      throw std::runtime_error("Failed to get AIE coredump: " + std::string(e.what()));
+    }
+  }
 };
 
 } // xrt
@@ -693,6 +709,13 @@ hw_context::
 ~hw_context()
 {}
 
+std::vector<char>
+hw_context::
+get_aie_coredump() const
+{
+  return get_handle()->get_aie_coredump();
+}
+
 } // xrt
 
 ////////////////////////////////////////////////////////////////

src/runtime_src/core/common/query_requests.h

@@ -348,7 +348,9 @@ enum class key_type
   noop,
 
   xocl_errors_ex,
-  xocl_ex_error_code2string
+  xocl_ex_error_code2string,
+
+  aie_coredump
 };
 
 struct pcie_vendor : request
@@ -4374,6 +4376,16 @@ struct read_trace_data : request
   virtual std::any
   get(const device*, const std::any&) const override = 0;
 };
+
+// Used for getting AIE coredump of all tiles within a ctx
+struct aie_coredump : request
+{
+  using result_type = std::vector<char>;
+  static const key_type key = key_type::aie_coredump;
+
+  std::any
+  get(const device*, const std::any&) const override = 0;
+};
 } // query
 
 } // xrt_core

src/runtime_src/core/include/xrt/xrt_hw_context.h

@@ -262,6 +262,15 @@ class hw_context : public detail::pimpl<hw_context_impl>
   access_mode
   get_mode() const;
 
+  /**
+   * get_aie_coredump() - Returns the coredump of AIE Array if available.
+   * Coredump represents memory/register dump of AIE tiles in this ctx.
+   * This function can throw.
+   */
+  XRT_API_EXPORT
+  std::vector<char>
+  get_aie_coredump() const;
+
 public:
   /// @cond
   // Undocumented internal access to low level context handle