Add realtime pmf npu metrics.

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

Author: VinitAmd

src/driver/amdxdna/aie2_pci.h

@@ -260,6 +260,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
@@ -325,6 +334,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

@@ -4,6 +4,12 @@
  */
 
 #include "aie2_pci.h"
+#include <linux/version.h>
+#include <linux/amd-pmf-io.h>
+
+#if KERNEL_VERSION(6, 18, 0) < LINUX_VERSION_CODE
+#define PMF_NPU_METRICS
+#endif
 
 #define SMU_RESULT_OK		1
 
@@ -51,6 +57,22 @@ static int aie2_smu_exec(struct amdxdna_dev_hdl *ndev, u32 reg_cmd,
 	return 0;
 }
 
+#ifdef PMF_NPU_METRICS
+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 +131,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)
 {
+#ifdef PMF_NPU_METRICS
+	int ret = aie2_smu_get_npu_metrics(ndev);
+
+	if (ret)
+		return ret;
+
+	return ndev->npu_metrics_info.mpnpuclk_freq;
+#endif
 	return ndev->npuclk_freq;
 }
 
 int aie2_smu_get_hclock_freq(struct amdxdna_dev_hdl *ndev)
 {
+#ifdef PMF_NPU_METRICS
+	int ret = aie2_smu_get_npu_metrics(ndev);
+
+	if (ret)
+		return ret;
+
+	return ndev->npu_metrics_info.npuclk_freq;
+#endif
+
 	return ndev->hclk_freq;
 }