Hw context level changes to add set/get freuquency apis (#9180)

Signed-off-by: Manoj Takasi <mtakasi@amd.com>

Author: ManojTakasi

build/petalinux.build

@@ -1,3 +1,3 @@
 # When updating Petalinux build please file a SH ticket to retain the build
 # https://jira.xilinx.com/secure/CreateIssue!default.jspa
-PETALINUX=/proj/petalinux/2025.2/petalinux-v2025.2_05300031/tool/petalinux-v2025.2-final
+PETALINUX=/proj/petalinux/2025.2/petalinux-v2025.2_08071216/tool/petalinux-v2025.2-final

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

@@ -23,6 +23,10 @@
 #include <limits>
 #include <memory>
 
+namespace {
+static constexpr double hz_per_mhz = 1'000'000.0;
+}
+
 namespace xrt {
 
 // class hw_context_impl - insulated implemention of an xrt::hw_context
@@ -213,6 +217,43 @@ class hw_context_impl : public std::enable_shared_from_this<hw_context_impl>
 
     throw std::runtime_error("no module found with given kernel name in ctx");
   }
+
+  double
+  get_aie_freq() const
+  {
+    try {
+      auto freq_hz = m_hdl->get_aie_freq();
+      return static_cast<double>(freq_hz) / hz_per_mhz; // Convert Hz to MHz
+    }
+    catch (const xrt_core::error& e) {
+      if (e.code() == std::errc::not_supported)
+        throw std::runtime_error("get_aie_freq() API is not supported on this platform");
+
+      throw std::runtime_error("Failed to read AIE frequency: " + std::string(e.what()));
+    }
+    catch (const std::exception& e) {
+      throw std::runtime_error("Failed to read AIE frequency: " + std::string(e.what()));
+    }
+  }
+
+  void
+  set_aie_freq(double freq_mhz)
+  {
+    try {
+      auto freq_hz = static_cast<uint64_t>(freq_mhz * hz_per_mhz);
+      m_hdl->set_aie_freq(freq_hz);
+    }
+    catch (const xrt_core::error& e) {
+      if (e.code() == std::errc::not_supported)
+        throw std::runtime_error("set_aie_freq() API is not supported on this platform");
+
+      throw std::runtime_error("Failed to set AIE frequency: " + std::string(e.what()));
+    }
+    catch (const std::exception& e) {
+      throw std::runtime_error("Failed to set AIE frequency: " + std::string(e.what()));
+    }
+  }
+
 };
 
 } // xrt
@@ -395,6 +436,20 @@ hw_context::
 ////////////////////////////////////////////////////////////////
 namespace xrt::aie {
 
+double
+hw_context::
+get_aie_freq() const
+{
+  return get_handle()->get_aie_freq();
+}
+
+void
+hw_context::
+set_aie_freq(double freq_mhz)
+{
+  get_handle()->set_aie_freq(freq_mhz);
+}
+
 void
 hw_context::
 reset_array()

src/runtime_src/core/common/shim/hwctx_handle.h

@@ -119,6 +119,19 @@ class hwctx_handle
   {
     throw xrt_core::error(std::errc::not_supported, __func__);
   }
+
+  virtual uint64_t
+  get_aie_freq() const
+  {
+    throw xrt_core::error(std::errc::not_supported, __func__);
+  }
+
+  virtual void
+  set_aie_freq(uint64_t /*freq_hz */) const
+  {
+    throw xrt_core::error(std::errc::not_supported, __func__);
+  }
+
 };
 
 } // xrt_core

src/runtime_src/core/edge/user/hwctx_object.cpp

@@ -145,4 +145,17 @@ hwctx_object::reset_array() const
 #endif
 }
 
+uint64_t
+hwctx_object::
+get_aie_freq() const
+{
+ return m_shim->get_aie_freq(this);
+}
+
+void
+hwctx_object::
+set_aie_freq(uint64_t freq_hz) const
+{
+ return m_shim->set_aie_freq(this, freq_hz);
+}
 }

src/runtime_src/core/edge/user/hwctx_object.h

@@ -111,6 +111,11 @@ namespace zynqaie {
     void
     reset_array() const override;
 
+    uint64_t
+    get_aie_freq() const override;
+
+    void
+    set_aie_freq(uint64_t freq_hz)const override;
 #ifdef XRT_ENABLE_AIE
     std::shared_ptr<aie_array>
     get_aie_array_shared();

src/runtime_src/core/edge/user/shim.cpp

@@ -2165,6 +2165,62 @@ openAIEContext(xrt::aie::access_mode am)
   return ret ? -errno : ret;
 }
 
+uint64_t
+shim::
+get_aie_freq(const zynqaie::hwctx_object* hwctx_obj)
+{
+#ifdef XRT_ENABLE_AIE
+  if (!hwctx_obj)
+    throw xrt_core::error(-EINVAL, "Invalid hardware context object");
+
+  auto hw_ctx_id = hwctx_obj->get_slotidx();
+  auto partition_info = hwctx_obj->get_partition_info();
+
+
+  struct drm_zocl_aie_freq_scale aie_arg;
+  aie_arg.hw_ctx_id = hw_ctx_id;
+  aie_arg.partition_id = partition_info.partition_id;
+  aie_arg.freq = 0;
+  aie_arg.dir = 0;
+
+  int ret = ioctl(mKernelFD, DRM_IOCTL_ZOCL_AIE_FREQSCALE, &aie_arg);
+
+  if (ret)
+    throw xrt_core::error(-errno, boost::str(boost::format("Reading AIE frequency from hw_context(%d) partition(%d) failed") % hw_ctx_id % partition_info.partition_id));
+
+  return aie_arg.freq;  // Return frequency in Hz
+#else
+  throw xrt_core::error(std::errc::not_supported, "AIE is not enabled for this device");
+#endif
+}
+
+void
+shim::
+set_aie_freq(const zynqaie::hwctx_object* hwctx_obj, uint64_t freq_hz)
+{
+#ifdef XRT_ENABLE_AIE
+  if (!hwctx_obj)
+    throw xrt_core::error(-EINVAL, "Invalid hardware context object");
+
+  auto hw_ctx_id = hwctx_obj->get_slotidx();
+  auto partition_info = hwctx_obj->get_partition_info();
+
+  struct drm_zocl_aie_freq_scale aie_arg;
+  aie_arg.hw_ctx_id = hw_ctx_id;
+  aie_arg.partition_id = partition_info.partition_id;
+  aie_arg.freq = freq_hz;
+  aie_arg.dir = 1;
+
+  int ret = ioctl(mKernelFD, DRM_IOCTL_ZOCL_AIE_FREQSCALE, &aie_arg);
+
+  if (ret)
+    throw xrt_core::error(-errno, boost::str(boost::format("Setting AIE frequency for hw_context(%d) partition(%d) to %lu Hz failed") % hw_ctx_id % partition_info.partition_id % freq_hz));
+
+#else
+  throw xrt_core::error(std::errc::not_supported, "AIE is not enabled for this device");
+#endif
+}
+
 xrt::aie::access_mode
 shim::
 getAIEAccessMode()

src/runtime_src/core/edge/user/shim.h

@@ -363,6 +363,8 @@ class shim {
   void open_graph_context(const zynqaie::hwctx_object* hwctx, const uuid_t xclbinId, unsigned int graph_id, xrt::graph::access_mode am);
   void close_graph_context(const zynqaie::hwctx_object* hwctx, unsigned int graph_id);
   int openAIEContext(xrt::aie::access_mode am);
+  uint64_t get_aie_freq(const zynqaie::hwctx_object* hwctx_obj);
+  void set_aie_freq(const zynqaie::hwctx_object* hwctx_obj, uint64_t freq_hz);
   xrt::aie::access_mode getAIEAccessMode();
   void setAIEAccessMode(xrt::aie::access_mode am);
 #endif

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

@@ -283,6 +283,28 @@ class hw_context : public xrt::hw_context
    */
   void
   reset_array();
+     /**
+   * get_aie_freq() - Get current AIE frequency for this hardware context
+   *
+   * Return: Current frequency in MHz
+   *
+   * This function retrieves the current AIE frequency for the partition
+   * associated with this hardware context.
+   */
+  double
+  get_aie_freq() const;
+
+    /**
+   * set_aie_freq() - Set AIE frequency for this hardware context
+   *
+   * @freq_mhz: Frequency to set in MHz
+   *
+   * This function sets the AIE frequency for the partition
+   * associated with this hardware context.
+   */
+  void
+  set_aie_freq(double /* freq_mhz*/);
+
 };
 
 class profiling_impl;