Expose NPU device engine busy time through DRM fdinfo to enable
standardized userspace monitoring and profiling.

Add support for reporting per-engine busy time in nanoseconds
via fdinfo using the "drm-engine-<driver-name>" key. The value
reflects the accumulated NPU device execution time for a given
process, allowing userspace tools to observe NPU device
utilization without driver-specific ioctls.

Signed-off-by: Max Zhen <[email protected]>
---
 Documentation/accel/amdxdna/amdnpu.rst  |  1 +
 drivers/accel/amdxdna/aie.c             | 39 +++++++++++++++++++++++++
 drivers/accel/amdxdna/aie.h             |  4 +++
 drivers/accel/amdxdna/aie2_ctx.c        |  6 ++++
 drivers/accel/amdxdna/amdxdna_pci_drv.c |  6 ++++
 drivers/accel/amdxdna/amdxdna_pci_drv.h |  9 ++++++
 6 files changed, 65 insertions(+)

diff --git a/Documentation/accel/amdxdna/amdnpu.rst 
b/Documentation/accel/amdxdna/amdnpu.rst
index 064973bf4893..d72af1b51ccf 100644
--- a/Documentation/accel/amdxdna/amdnpu.rst
+++ b/Documentation/accel/amdxdna/amdnpu.rst
@@ -289,6 +289,7 @@ Example of the output showing the implemented key value 
pairs:
         drm-driver:    amdxdna_accel_driver
         drm-client-id: 3219
         drm-pdev:      0000:c5:00.1
+        drm-engine-npu: 14003870 ns
         amdxdna_accel_driver-heap-alloc:       60 KiB
         amdxdna_accel_driver-internal-alloc:   67588 KiB
         amdxdna_accel_driver-external-alloc:   0
diff --git a/drivers/accel/amdxdna/aie.c b/drivers/accel/amdxdna/aie.c
index dd6f36f222c7..bf8eb66baab8 100644
--- a/drivers/accel/amdxdna/aie.c
+++ b/drivers/accel/amdxdna/aie.c
@@ -118,6 +118,45 @@ void amdxdna_vbnv_init(struct amdxdna_dev *xdna)
        amdxdna_update_vbnv(xdna, info->rev_vbnv_tbl, rev);
 }
 
+void amdxdna_io_stats_job_start(struct amdxdna_client *client)
+{
+       int depth;
+
+       guard(spinlock)(&client->io_stats.lock);
+
+       depth = client->io_stats.job_depth++;
+       if (!depth)
+               client->io_stats.start_time = ktime_get_ns();
+}
+
+void amdxdna_io_stats_job_done(struct amdxdna_client *client)
+{
+       u64 busy_ns;
+       int depth;
+
+       guard(spinlock)(&client->io_stats.lock);
+
+       depth = --client->io_stats.job_depth;
+       if (!depth) {
+               busy_ns = ktime_get_ns() - client->io_stats.start_time;
+               client->io_stats.start_time = 0;
+               client->io_stats.busy_time += busy_ns;
+       }
+}
+
+u64 amdxdna_io_stats_busy_time_ns(struct amdxdna_client *client)
+{
+       u64 busy_ns;
+
+       guard(spinlock)(&client->io_stats.lock);
+
+       busy_ns = client->io_stats.busy_time;
+       if (client->io_stats.job_depth)
+               busy_ns += ktime_get_ns() - client->io_stats.start_time;
+
+       return busy_ns;
+}
+
 int amdxdna_get_metadata(struct aie_device *aie,
                         struct amdxdna_client *client,
                         struct amdxdna_drm_get_info *args)
diff --git a/drivers/accel/amdxdna/aie.h b/drivers/accel/amdxdna/aie.h
index 0483d582b7f8..1fc5ddfe696a 100644
--- a/drivers/accel/amdxdna/aie.h
+++ b/drivers/accel/amdxdna/aie.h
@@ -117,4 +117,8 @@ void aie_smu_fini(struct smu_device *smu);
 int aie_smu_set_clocks(struct smu_device *smu, u32 *npuclk, u32 *hclk);
 int aie_smu_set_dpm(struct smu_device *smu, u32 dpm_level);
 
+void amdxdna_io_stats_job_start(struct amdxdna_client *client);
+void amdxdna_io_stats_job_done(struct amdxdna_client *client);
+u64 amdxdna_io_stats_busy_time_ns(struct amdxdna_client *client);
+
 #endif /* _AIE_H_ */
diff --git a/drivers/accel/amdxdna/aie2_ctx.c b/drivers/accel/amdxdna/aie2_ctx.c
index d927c8c9d557..9c6b7353cbe9 100644
--- a/drivers/accel/amdxdna/aie2_ctx.c
+++ b/drivers/accel/amdxdna/aie2_ctx.c
@@ -290,6 +290,7 @@ aie2_sched_resp_handler(void *handle, void __iomem *data, 
size_t size)
        int ret = 0;
        u32 status;
 
+       amdxdna_io_stats_job_done(job->hwctx->client);
        cmd_abo = job->cmd_bo;
 
        if (unlikely(job->job_timeout)) {
@@ -322,6 +323,8 @@ aie2_sched_drvcmd_resp_handler(void *handle, void __iomem 
*data, size_t size)
        struct amdxdna_sched_job *job = handle;
        int ret = 0;
 
+       amdxdna_io_stats_job_done(job->hwctx->client);
+
        if (unlikely(!data || size != sizeof(u32))) {
                job->drv_cmd->result = U32_MAX;
                ret = -EINVAL;
@@ -344,6 +347,7 @@ aie2_sched_cmdlist_resp_handler(void *handle, void __iomem 
*data, size_t size)
        u32 cmd_status;
        int ret = 0;
 
+       amdxdna_io_stats_job_done(job->hwctx->client);
        cmd_abo = job->cmd_bo;
 
        if (unlikely(job->job_timeout)) {
@@ -405,6 +409,7 @@ aie2_sched_job_run(struct drm_sched_job *sched_job)
        kref_get(&job->refcnt);
        fence = dma_fence_get(job->fence);
 
+       amdxdna_io_stats_job_start(job->hwctx->client);
        if (job->drv_cmd) {
                switch (job->drv_cmd->opcode) {
                case SYNC_DEBUG_BO:
@@ -432,6 +437,7 @@ aie2_sched_job_run(struct drm_sched_job *sched_job)
 
 out:
        if (ret) {
+               amdxdna_io_stats_job_done(job->hwctx->client);
                dma_fence_put(job->fence);
                aie2_job_put(job);
                mmput(job->mm);
diff --git a/drivers/accel/amdxdna/amdxdna_pci_drv.c 
b/drivers/accel/amdxdna/amdxdna_pci_drv.c
index bb339e641416..67a02d6b7f97 100644
--- a/drivers/accel/amdxdna/amdxdna_pci_drv.c
+++ b/drivers/accel/amdxdna/amdxdna_pci_drv.c
@@ -14,6 +14,7 @@
 #include <linux/iommu.h>
 #include <linux/pci.h>
 
+#include "aie.h"
 #include "amdxdna_cbuf.h"
 #include "amdxdna_ctx.h"
 #include "amdxdna_debugfs.h"
@@ -151,6 +152,8 @@ static int amdxdna_drm_open(struct drm_device *ddev, struct 
drm_file *filp)
        filp->driver_priv = client;
        client->filp = filp;
 
+       spin_lock_init(&client->io_stats.lock);
+
        XDNA_DBG(xdna, "pid %d opened", client->pid);
        return 0;
 
@@ -287,6 +290,9 @@ static void amdxdna_show_fdinfo(struct drm_printer *p, 
struct drm_file *filp)
        size_t heap_usage, external_usage, internal_usage;
        char *drv_name = filp->minor->dev->driver->name;
 
+       drm_printf(p, "drm-engine-npu:\t%llu ns\n",
+                  amdxdna_io_stats_busy_time_ns(client));
+
        mutex_lock(&client->mm_lock);
 
        heap_usage = client->heap_usage;
diff --git a/drivers/accel/amdxdna/amdxdna_pci_drv.h 
b/drivers/accel/amdxdna/amdxdna_pci_drv.h
index a997d27a504d..510356a31532 100644
--- a/drivers/accel/amdxdna/amdxdna_pci_drv.h
+++ b/drivers/accel/amdxdna/amdxdna_pci_drv.h
@@ -143,6 +143,13 @@ struct amdxdna_device_id {
        const struct amdxdna_dev_info *dev_info;
 };
 
+struct amdxdna_io_stats {
+       spinlock_t                      lock; /* protect io stats */
+       int                             job_depth;
+       u64                             start_time;
+       u64                             busy_time;
+};
+
 /*
  * struct amdxdna_client - amdxdna client
  * A per fd data structure for managing context and other user process stuffs.
@@ -169,6 +176,8 @@ struct amdxdna_client {
        size_t                          heap_usage;
        size_t                          total_bo_usage;
        size_t                          total_int_bo_usage;
+
+       struct amdxdna_io_stats         io_stats;
 };
 
 #define amdxdna_for_each_hwctx(client, hwctx_id, entry)                \
-- 
2.34.1

Reply via email to