From: Shyam Sundar S K <[email protected]>

The PMF driver retrieves NPU metrics data from the PMFW. Introduce a new
interface to make NPU metrics accessible to other drivers like AMDXDNA
driver, which can access and utilize this information as needed.

Reviewed-by: Mario Limonciello <[email protected]>
Co-developed-by: Patil Rajesh Reddy <[email protected]>
Signed-off-by: Patil Rajesh Reddy <[email protected]>
Signed-off-by: Shyam Sundar S K <[email protected]>
Signed-off-by: Lizhi Hou <[email protected]>
---
 drivers/platform/x86/amd/pmf/core.c | 75 +++++++++++++++++++++++++++++
 drivers/platform/x86/amd/pmf/pmf.h  |  2 +
 include/linux/amd-pmf-io.h          | 21 ++++++++
 3 files changed, 98 insertions(+)

diff --git a/drivers/platform/x86/amd/pmf/core.c 
b/drivers/platform/x86/amd/pmf/core.c
index a6a5d416edf9..8e4ce91b3527 100644
--- a/drivers/platform/x86/amd/pmf/core.c
+++ b/drivers/platform/x86/amd/pmf/core.c
@@ -8,12 +8,15 @@
  * Author: Shyam Sundar S K <[email protected]>
  */
 
+#include <linux/array_size.h>
+#include <linux/cleanup.h>
 #include <linux/debugfs.h>
 #include <linux/iopoll.h>
 #include <linux/module.h>
 #include <linux/pci.h>
 #include <linux/platform_device.h>
 #include <linux/power_supply.h>
+#include <linux/string.h>
 #include <asm/amd/node.h>
 #include "pmf.h"
 
@@ -53,6 +56,8 @@ static bool force_load;
 module_param(force_load, bool, 0444);
 MODULE_PARM_DESC(force_load, "Force load this driver on supported older 
platforms (experimental)");
 
+static struct device *pmf_device;
+
 static int amd_pmf_pwr_src_notify_call(struct notifier_block *nb, unsigned 
long event, void *data)
 {
        struct amd_pmf_dev *pmf = container_of(nb, struct amd_pmf_dev, 
pwr_src_notifier);
@@ -314,6 +319,70 @@ int amd_pmf_init_metrics_table(struct amd_pmf_dev *dev)
        return 0;
 }
 
+static int is_npu_metrics_supported(struct amd_pmf_dev *pdev)
+{
+       switch (pdev->cpu_id) {
+       case PCI_DEVICE_ID_AMD_1AH_M20H_ROOT:
+       case PCI_DEVICE_ID_AMD_1AH_M60H_ROOT:
+               return 0;
+       default:
+               return -EOPNOTSUPP;
+       }
+}
+
+static int amd_pmf_get_smu_metrics(struct amd_pmf_dev *dev, struct 
amd_pmf_npu_metrics *data)
+{
+       int ret, i;
+
+       guard(mutex)(&dev->metrics_mutex);
+
+       if (is_npu_metrics_supported(dev))
+               return -EOPNOTSUPP;
+
+       ret = amd_pmf_set_dram_addr(dev, true);
+       if (ret)
+               return ret;
+
+       memset(dev->buf, 0, dev->mtable_size);
+
+       /* Send SMU command to get NPU metrics */
+       ret = amd_pmf_send_cmd(dev, SET_TRANSFER_TABLE, SET_CMD, 
METRICS_TABLE_ID, NULL);
+       if (ret) {
+               dev_err(dev->dev, "SMU command failed to get NPU metrics: 
%d\n", ret);
+               return ret;
+       }
+
+       memcpy(&dev->m_table_v2, dev->buf, dev->mtable_size);
+
+       data->npuclk_freq = dev->m_table_v2.npuclk_freq;
+       for (i = 0; i < ARRAY_SIZE(data->npu_busy); i++)
+               data->npu_busy[i] = dev->m_table_v2.npu_busy[i];
+       data->npu_power = dev->m_table_v2.npu_power;
+       data->mpnpuclk_freq = dev->m_table_v2.mpnpuclk_freq;
+       data->npu_reads = dev->m_table_v2.npu_reads;
+       data->npu_writes = dev->m_table_v2.npu_writes;
+
+       return 0;
+}
+
+int amd_pmf_get_npu_data(struct amd_pmf_npu_metrics *info)
+{
+       struct amd_pmf_dev *pdev;
+
+       if (!info)
+               return -EINVAL;
+
+       if (!pmf_device)
+               return -ENODEV;
+
+       pdev = dev_get_drvdata(pmf_device);
+       if (!pdev)
+               return -ENODEV;
+
+       return amd_pmf_get_smu_metrics(pdev, info);
+}
+EXPORT_SYMBOL_GPL(amd_pmf_get_npu_data);
+
 static int amd_pmf_suspend_handler(struct device *dev)
 {
        struct amd_pmf_dev *pdev = dev_get_drvdata(dev);
@@ -469,6 +538,10 @@ static int amd_pmf_probe(struct platform_device *pdev)
        mutex_init(&dev->update_mutex);
        mutex_init(&dev->cb_mutex);
 
+       err = devm_mutex_init(dev->dev, &dev->metrics_mutex);
+       if (err)
+               return err;
+
        apmf_acpi_init(dev);
        platform_set_drvdata(pdev, dev);
        amd_pmf_dbgfs_register(dev);
@@ -477,6 +550,8 @@ static int amd_pmf_probe(struct platform_device *pdev)
        if (is_apmf_func_supported(dev, APMF_FUNC_SBIOS_HEARTBEAT_V2))
                amd_pmf_notify_sbios_heartbeat_event_v2(dev, ON_LOAD);
 
+       pmf_device = dev->dev;
+
        dev_info(dev->dev, "registered PMF device successfully\n");
 
        return 0;
diff --git a/drivers/platform/x86/amd/pmf/pmf.h 
b/drivers/platform/x86/amd/pmf/pmf.h
index f07e9f4c660a..0354cc5dc79e 100644
--- a/drivers/platform/x86/amd/pmf/pmf.h
+++ b/drivers/platform/x86/amd/pmf/pmf.h
@@ -12,6 +12,7 @@
 #define PMF_H
 
 #include <linux/acpi.h>
+#include <linux/amd-pmf-io.h>
 #include <linux/input.h>
 #include <linux/platform_device.h>
 #include <linux/platform_profile.h>
@@ -412,6 +413,7 @@ struct amd_pmf_dev {
        struct apmf_sbios_req_v1 req1;
        struct pmf_bios_inputs_prev cb_prev; /* To preserve custom BIOS inputs 
*/
        bool cb_flag;                        /* To handle first custom BIOS 
input */
+       struct mutex metrics_mutex;
 };
 
 struct apmf_sps_prop_granular_v2 {
diff --git a/include/linux/amd-pmf-io.h b/include/linux/amd-pmf-io.h
index 6fa510f419c0..55198d2875cc 100644
--- a/include/linux/amd-pmf-io.h
+++ b/include/linux/amd-pmf-io.h
@@ -61,5 +61,26 @@ enum laptop_placement {
        LP_UNDEFINED,
 };
 
+/**
+ * struct amd_pmf_npu_metrics: Get NPU metrics data from PMF driver
+ * @npuclk_freq: NPU clock frequency [MHz]
+ * @npu_busy: NPU busy % [0-100]
+ * @npu_power: NPU power [mW]
+ * @mpnpuclk_freq: MPNPU [MHz]
+ * @npu_reads: NPU read bandwidth [MB/sec]
+ * @npu_writes: NPU write bandwidth [MB/sec]
+ */
+struct amd_pmf_npu_metrics {
+       u16 npuclk_freq;
+       u16 npu_busy[8];
+       u16 npu_power;
+       u16 mpnpuclk_freq;
+       u16 npu_reads;
+       u16 npu_writes;
+};
+
 int amd_get_sfh_info(struct amd_sfh_info *sfh_info, enum sfh_message_type op);
+
+/* AMD PMF and NPU interface */
+int amd_pmf_get_npu_data(struct amd_pmf_npu_metrics *info);
 #endif
-- 
2.34.1

Reply via email to