Add realtime pmf npu metrics.

Signed-off-by: vinit shukla <[email protected]>

Author: VinitAmd

src/driver/amdxdna/aie2_pci.h

@@ -259,6 +259,15 @@ enum aie2_power_state {
 	SMU_POWER_ON,
 };
 
+struct npu_metrics {
+        u16 npuclk_freq;
+        u16 npu_busy[8];
+        u16 npu_power;
+        u16 mpnpuclk_freq;
+        u16 npu_reads;
+        u16 npu_writes;
+};
+
 struct aie2_partition {
 #define CTX_RQ_REALTIME		0
 #define CTX_RQ_HIGH		1
@@ -324,6 +333,7 @@ struct amdxdna_dev_hdl {
 	u32				total_col;
 	struct aie_version		version;
 	struct aie_metadata		metadata;
+	struct npu_metrics		npu_metrics_info;
 
 	/*power management and clock */
 	int				pw_mode;

src/driver/amdxdna/aie2_smu.c

@@ -5,6 +5,10 @@
 
 #include "aie2_pci.h"
 
+#if KERNEL_VERSION(6, 18, 0) > LINUX_VERSION_CODE
+#include <linux/amd-pmf-io.h>
+#endif
+
 #define SMU_RESULT_OK		1
 
 /* SMU commands */
@@ -51,6 +55,22 @@ static int aie2_smu_exec(struct amdxdna_dev_hdl *ndev, u32 reg_cmd,
 	return 0;
 }
 
+#if KERNEL_VERSION(6, 18, 0) > LINUX_VERSION_CODE
+static int aie2_smu_get_npu_metrics(struct amdxdna_dev_hdl *ndev)
+{
+	struct amd_pmf_npu_metrics data;
+	int ret;
+
+	/* Get NPU metrics from the PMF interface */
+	ret = amd_pmf_get_npu_data(&data);
+	if (ret)
+		return ret;
+
+	memcpy(&ndev->npu_metrics_info, &data, sizeof(data));
+	return 0;
+}
+#endif
+
 int npu1_set_dpm(struct amdxdna_dev_hdl *ndev, u32 dpm_level)
 {
 	u32 freq;
@@ -109,11 +129,28 @@ int npu4_set_dpm(struct amdxdna_dev_hdl *ndev, u32 dpm_level)
 
 int aie2_smu_get_mpnpu_clock_freq(struct amdxdna_dev_hdl *ndev)
 {
+
+#if KERNEL_VERSION(6, 18, 0) > LINUX_VERSION_CODE
+	int ret = aie2_smu_get_npu_metrics(ndev);
+	if (ret)
+		return -EINVAL;
+
+	return ndev->npu_metrics_info.mpnpuclk_freq;
+#endif
 	return ndev->npuclk_freq;
 }
 
 int aie2_smu_get_hclock_freq(struct amdxdna_dev_hdl *ndev)
 {
+
+#if KERNEL_VERSION(6, 18, 0) > LINUX_VERSION_CODE
+        int ret = aie2_smu_get_npu_metrics(ndev);
+        if (ret)
+                return -EINVAL;
+
+        return ndev->npu_metrics_info.npuclk_freq;
+#endif
+
 	return ndev->hclk_freq;
 }