From: Tvrtko Ursulin <[email protected]>

Convert `v3d_stats` from embedded structs to heap-allocated, refcounted
objects. This decouples the stats lifetime from the containing
structures (this is, `v3d_queue_state` and `v3d_file_priv`), allowing
jobs to safely hold their own references to stats objects even after the
file descriptor is closed.

Signed-off-by: Tvrtko Ursulin <[email protected]>
Co-developed-by: Maíra Canal <[email protected]>
Signed-off-by: Maíra Canal <[email protected]>
---
 drivers/gpu/drm/v3d/v3d_drv.c   | 19 ++++++++++++++-----
 drivers/gpu/drm/v3d/v3d_drv.h   | 19 +++++++++++++++++--
 drivers/gpu/drm/v3d/v3d_gem.c   | 42 ++++++++++++++++++++++++++---------------
 drivers/gpu/drm/v3d/v3d_sched.c | 29 ++++++++++++++++++++++++----
 drivers/gpu/drm/v3d/v3d_sysfs.c |  2 +-
 5 files changed, 84 insertions(+), 27 deletions(-)

diff --git a/drivers/gpu/drm/v3d/v3d_drv.c b/drivers/gpu/drm/v3d/v3d_drv.c
index 
16aef2a5ecd218b8c04bc0097263554e5d7ad954..aafb402c6ac3118a57df9fc0a0d21d35d48e3b2c
 100644
--- a/drivers/gpu/drm/v3d/v3d_drv.c
+++ b/drivers/gpu/drm/v3d/v3d_drv.c
@@ -140,15 +140,18 @@ v3d_open(struct drm_device *dev, struct drm_file *file)
        v3d_priv->v3d = v3d;
 
        for (i = 0; i < V3D_MAX_QUEUES; i++) {
+               v3d_priv->stats[i] = v3d_stats_alloc();
+               if (!v3d_priv->stats[i]) {
+                       ret = -ENOMEM;
+                       goto err_stats;
+               }
+
                sched = &v3d->queue[i].sched;
                ret = drm_sched_entity_init(&v3d_priv->sched_entity[i],
                                            DRM_SCHED_PRIORITY_NORMAL, &sched,
                                            1, NULL);
                if (ret)
                        goto err_sched;
-
-               memset(&v3d_priv->stats[i], 0, sizeof(v3d_priv->stats[i]));
-               seqcount_init(&v3d_priv->stats[i].lock);
        }
 
        v3d_perfmon_open_file(v3d_priv);
@@ -157,8 +160,12 @@ v3d_open(struct drm_device *dev, struct drm_file *file)
        return 0;
 
 err_sched:
-       for (i--; i >= 0; i--)
+       v3d_stats_put(v3d_priv->stats[i]);
+err_stats:
+       for (i--; i >= 0; i--) {
                drm_sched_entity_destroy(&v3d_priv->sched_entity[i]);
+               v3d_stats_put(v3d_priv->stats[i]);
+       }
        kfree(v3d_priv);
        return ret;
 }
@@ -182,6 +189,8 @@ v3d_postclose(struct drm_device *dev, struct drm_file *file)
                        job->file_priv = NULL;
                        spin_unlock_irqrestore(&queue->queue_lock, irqflags);
                }
+
+               v3d_stats_put(v3d_priv->stats[q]);
        }
 
        v3d_perfmon_close_file(v3d_priv);
@@ -209,7 +218,7 @@ static void v3d_show_fdinfo(struct drm_printer *p, struct 
drm_file *file)
        enum v3d_queue queue;
 
        for (queue = 0; queue < V3D_MAX_QUEUES; queue++) {
-               struct v3d_stats *stats = &file_priv->stats[queue];
+               struct v3d_stats *stats = file_priv->stats[queue];
                u64 active_runtime, jobs_completed;
 
                v3d_get_stats(stats, timestamp, &active_runtime, 
&jobs_completed);
diff --git a/drivers/gpu/drm/v3d/v3d_drv.h b/drivers/gpu/drm/v3d/v3d_drv.h
index 
2e5520015e08c47fef4bfbf185eda15027992032..03fa2d174b1ca8b5a98a72c4addaa1f977d11174
 100644
--- a/drivers/gpu/drm/v3d/v3d_drv.h
+++ b/drivers/gpu/drm/v3d/v3d_drv.h
@@ -38,6 +38,8 @@ static inline char *v3d_queue_to_string(enum v3d_queue queue)
 }
 
 struct v3d_stats {
+       struct kref refcount;
+
        u64 start_ns;
        u64 enabled_ns;
        u64 jobs_completed;
@@ -62,7 +64,7 @@ struct v3d_queue_state {
        u64 emit_seqno;
 
        /* Stores the GPU stats for this queue in the global context. */
-       struct v3d_stats stats;
+       struct v3d_stats *stats;
 
        /* Currently active job for this queue */
        struct v3d_job *active_job;
@@ -230,7 +232,7 @@ struct v3d_file_priv {
        struct drm_sched_entity sched_entity[V3D_MAX_QUEUES];
 
        /* Stores the GPU stats for a specific queue for this fd. */
-       struct v3d_stats stats[V3D_MAX_QUEUES];
+       struct v3d_stats *stats[V3D_MAX_QUEUES];
 
        /* Per-fd reset counter, must be incremented when a job submitted
         * by this fd causes a GPU reset. It must be protected by
@@ -603,10 +605,23 @@ void v3d_timestamp_query_info_free(struct 
v3d_timestamp_query_info *query_info,
                                   unsigned int count);
 void v3d_performance_query_info_free(struct v3d_performance_query_info 
*query_info,
                                     unsigned int count);
+struct v3d_stats *v3d_stats_alloc(void);
+void v3d_stats_release(struct kref *refcount);
 void v3d_job_update_stats(struct v3d_job *job, enum v3d_queue q);
 int v3d_sched_init(struct v3d_dev *v3d);
 void v3d_sched_fini(struct v3d_dev *v3d);
 
+static inline struct v3d_stats *v3d_stats_get(struct v3d_stats *stats)
+{
+       kref_get(&stats->refcount);
+       return stats;
+}
+
+static inline void v3d_stats_put(struct v3d_stats *stats)
+{
+       kref_put(&stats->refcount, v3d_stats_release);
+}
+
 /* v3d_perfmon.c */
 void v3d_perfmon_init(struct v3d_dev *v3d);
 void v3d_perfmon_get(struct v3d_perfmon *perfmon);
diff --git a/drivers/gpu/drm/v3d/v3d_gem.c b/drivers/gpu/drm/v3d/v3d_gem.c
index 
57965c0d6f6efea0019fb0b1a47addf2f586d138..859e63dd7e9738e3a3702edfb857ec3e844b052b
 100644
--- a/drivers/gpu/drm/v3d/v3d_gem.c
+++ b/drivers/gpu/drm/v3d/v3d_gem.c
@@ -287,9 +287,13 @@ v3d_gem_init(struct drm_device *dev)
        for (i = 0; i < V3D_MAX_QUEUES; i++) {
                struct v3d_queue_state *queue = &v3d->queue[i];
 
+               queue->stats = v3d_stats_alloc();
+               if (!queue->stats) {
+                       ret = -ENOMEM;
+                       goto err_stats;
+               }
+
                queue->fence_context = dma_fence_context_alloc(1);
-               memset(&queue->stats, 0, sizeof(queue->stats));
-               seqcount_init(&queue->stats.lock);
 
                spin_lock_init(&queue->queue_lock);
                spin_lock_init(&queue->fence_lock);
@@ -298,16 +302,16 @@ v3d_gem_init(struct drm_device *dev)
        spin_lock_init(&v3d->mm_lock);
        ret = drmm_mutex_init(dev, &v3d->bo_lock);
        if (ret)
-               return ret;
+               goto err_stats;
        ret = drmm_mutex_init(dev, &v3d->reset_lock);
        if (ret)
-               return ret;
+               goto err_stats;
        ret = drmm_mutex_init(dev, &v3d->sched_lock);
        if (ret)
-               return ret;
+               goto err_stats;
        ret = drmm_mutex_init(dev, &v3d->cache_clean_lock);
        if (ret)
-               return ret;
+               goto err_stats;
 
        /* Note: We don't allocate address 0.  Various bits of HW
         * treat 0 as special, such as the occlusion query counters
@@ -319,10 +323,10 @@ v3d_gem_init(struct drm_device *dev)
                               &v3d->pt_paddr,
                               GFP_KERNEL | __GFP_NOWARN | __GFP_ZERO);
        if (!v3d->pt) {
-               drm_mm_takedown(&v3d->mm);
                dev_err(v3d->drm.dev,
                        "Failed to allocate page tables. Please ensure you have 
DMA enabled.\n");
-               return -ENOMEM;
+               ret = -ENOMEM;
+               goto err_dma_alloc;
        }
 
        v3d_init_hw_state(v3d);
@@ -331,14 +335,20 @@ v3d_gem_init(struct drm_device *dev)
        v3d_huge_mnt_init(v3d);
 
        ret = v3d_sched_init(v3d);
-       if (ret) {
-               drm_mm_takedown(&v3d->mm);
-               dma_free_coherent(v3d->drm.dev, pt_size, (void *)v3d->pt,
-                                 v3d->pt_paddr);
-               return ret;
-       }
+       if (ret)
+               goto err_sched;
 
        return 0;
+
+err_sched:
+       dma_free_coherent(v3d->drm.dev, pt_size, (void *)v3d->pt, 
v3d->pt_paddr);
+err_dma_alloc:
+       drm_mm_takedown(&v3d->mm);
+err_stats:
+       for (i--; i >= 0; i--)
+               v3d_stats_put(v3d->queue[i].stats);
+
+       return ret;
 }
 
 void
@@ -352,8 +362,10 @@ v3d_gem_destroy(struct drm_device *dev)
        /* Waiting for jobs to finish would need to be done before
         * unregistering V3D.
         */
-       for (q = 0; q < V3D_MAX_QUEUES; q++)
+       for (q = 0; q < V3D_MAX_QUEUES; q++) {
                WARN_ON(v3d->queue[q].active_job);
+               v3d_stats_put(v3d->queue[q].stats);
+       }
 
        drm_mm_takedown(&v3d->mm);
 
diff --git a/drivers/gpu/drm/v3d/v3d_sched.c b/drivers/gpu/drm/v3d/v3d_sched.c
index 
18265721c1d32158fa6f7e68fa3e70a77d265b9d..46b776fa883729cec67959efa100347da4cfed86
 100644
--- a/drivers/gpu/drm/v3d/v3d_sched.c
+++ b/drivers/gpu/drm/v3d/v3d_sched.c
@@ -66,6 +66,27 @@ to_cpu_job(struct drm_sched_job *sched_job)
        return container_of(sched_job, struct v3d_cpu_job, base.base);
 }
 
+void v3d_stats_release(struct kref *refcount)
+{
+       struct v3d_stats *stats = container_of(refcount, typeof(*stats), 
refcount);
+
+       kfree(stats);
+}
+
+struct v3d_stats *v3d_stats_alloc(void)
+{
+       struct v3d_stats *stats;
+
+       stats = kzalloc(sizeof(*stats), GFP_KERNEL);
+       if (!stats)
+               return NULL;
+
+       kref_init(&stats->refcount);
+       seqcount_init(&stats->lock);
+
+       return stats;
+}
+
 static void
 v3d_sched_job_free(struct drm_sched_job *sched_job)
 {
@@ -141,8 +162,8 @@ v3d_job_start_stats(struct v3d_job *job, enum v3d_queue 
queue)
 {
        struct v3d_dev *v3d = job->v3d;
        struct v3d_file_priv *file = job->file_priv;
-       struct v3d_stats *global_stats = &v3d->queue[queue].stats;
-       struct v3d_stats *local_stats = &file->stats[queue];
+       struct v3d_stats *global_stats = v3d->queue[queue].stats;
+       struct v3d_stats *local_stats = file->stats[queue];
        u64 now = local_clock();
 
        preempt_disable();
@@ -173,7 +194,7 @@ v3d_job_update_stats(struct v3d_job *job, enum v3d_queue q)
 {
        struct v3d_dev *v3d = job->v3d;
        struct v3d_queue_state *queue = &v3d->queue[q];
-       struct v3d_stats *global_stats = &queue->stats;
+       struct v3d_stats *global_stats = queue->stats;
        u64 now = local_clock();
 
        preempt_disable();
@@ -181,7 +202,7 @@ v3d_job_update_stats(struct v3d_job *job, enum v3d_queue q)
        /* Don't update the local stats if the file context has already closed 
*/
        spin_lock(&queue->queue_lock);
        if (job->file_priv)
-               v3d_stats_update(&job->file_priv->stats[q], now);
+               v3d_stats_update(job->file_priv->stats[q], now);
        spin_unlock(&queue->queue_lock);
 
        v3d_stats_update(global_stats, now);
diff --git a/drivers/gpu/drm/v3d/v3d_sysfs.c b/drivers/gpu/drm/v3d/v3d_sysfs.c
index 
d610e355964ffaf45f7d44e5c667369cedc205dc..b45a9b3db42d698856873145e4948aad27eeb28c
 100644
--- a/drivers/gpu/drm/v3d/v3d_sysfs.c
+++ b/drivers/gpu/drm/v3d/v3d_sysfs.c
@@ -20,7 +20,7 @@ gpu_stats_show(struct device *dev, struct device_attribute 
*attr, char *buf)
        len += sysfs_emit(buf, "queue\ttimestamp\tjobs\truntime\n");
 
        for (queue = 0; queue < V3D_MAX_QUEUES; queue++) {
-               struct v3d_stats *stats = &v3d->queue[queue].stats;
+               struct v3d_stats *stats = v3d->queue[queue].stats;
                u64 active_runtime, jobs_completed;
 
                v3d_get_stats(stats, timestamp, &active_runtime, 
&jobs_completed);

-- 
2.52.0

Reply via email to