[Intel-gfx] [PATCH] drm: Convert prime dma-buf <-> handle to rhashtable
Sean Paul
seanpaul at chromium.org
Fri Sep 23 10:27:34 UTC 2016
On Thu, Sep 22, 2016 at 7:43 AM, Chris Wilson <chris at chris-wilson.co.uk> wrote:
> Currently we use a linear walk to lookup a handle and return a dma-buf,
> and vice versa. A long overdue TODO task is to convert that to a
> hashtable. Since the initial implementation of dma-buf/prime, we now
> have resizeable hashtables we can use (and now a future task is to RCU
> enable the lookup!).
>
> Bugzilla: https://bugs.freedesktop.org/show_bug.cgi?id=94631
> Signed-off-by: Chris Wilson <chris at chris-wilson.co.uk>
Looks like a nice improvement
Reviewed-by: Sean Paul <seanpaul at chromium.org>
> ---
> drivers/gpu/drm/drm_prime.c | 94 +++++++++++++++++++++++++++++++++++----------
> include/drm/drmP.h | 5 ++-
> 2 files changed, 77 insertions(+), 22 deletions(-)
>
> diff --git a/drivers/gpu/drm/drm_prime.c b/drivers/gpu/drm/drm_prime.c
> index 780589b420a4..ad077def660d 100644
> --- a/drivers/gpu/drm/drm_prime.c
> +++ b/drivers/gpu/drm/drm_prime.c
> @@ -28,6 +28,7 @@
>
> #include <linux/export.h>
> #include <linux/dma-buf.h>
> +#include <linux/rhashtable.h>
> #include <drm/drmP.h>
> #include <drm/drm_gem.h>
>
> @@ -61,9 +62,11 @@
> */
>
> struct drm_prime_member {
> - struct list_head entry;
> struct dma_buf *dma_buf;
> uint32_t handle;
> +
> + struct rhash_head dma_buf_rht;
> + struct rhash_head handle_rht;
> };
>
> struct drm_prime_attachment {
> @@ -71,10 +74,31 @@ struct drm_prime_attachment {
> enum dma_data_direction dir;
> };
>
> +static const struct rhashtable_params dma_buf_params = {
> + .head_offset = offsetof(struct drm_prime_member, dma_buf_rht),
> + .key_len = sizeof(struct dma_buf *),
> + .key_offset = offsetof(struct drm_prime_member, dma_buf),
> + .hashfn = jhash,
> + .nulls_base = 1u << RHT_BASE_SHIFT,
> + .automatic_shrinking = true,
> + .nelem_hint = 2,
> +};
> +
> +static const struct rhashtable_params handle_params = {
> + .head_offset = offsetof(struct drm_prime_member, handle_rht),
> + .key_len = sizeof(uint32_t),
> + .key_offset = offsetof(struct drm_prime_member, handle),
> + .hashfn = jhash,
> + .nulls_base = 1u << RHT_BASE_SHIFT,
> + .automatic_shrinking = true,
> + .nelem_hint = 2,
> +};
> +
> static int drm_prime_add_buf_handle(struct drm_prime_file_private *prime_fpriv,
> struct dma_buf *dma_buf, uint32_t handle)
> {
> struct drm_prime_member *member;
> + int err;
>
> member = kmalloc(sizeof(*member), GFP_KERNEL);
> if (!member)
> @@ -83,8 +107,28 @@ static int drm_prime_add_buf_handle(struct drm_prime_file_private *prime_fpriv,
> get_dma_buf(dma_buf);
> member->dma_buf = dma_buf;
> member->handle = handle;
> - list_add(&member->entry, &prime_fpriv->head);
> +
> + err = rhashtable_insert_fast(&prime_fpriv->dma_bufs,
> + &member->dma_buf_rht,
> + dma_buf_params);
> + if (err)
> + goto err_dma_buf;
> +
> + err = rhashtable_insert_fast(&prime_fpriv->handles,
> + &member->handle_rht,
> + handle_params);
> + if (err)
> + goto err_dma_rht;
> +
> return 0;
> +
> +err_dma_rht:
> + rhashtable_remove_fast(&prime_fpriv->dma_bufs,
> + &member->dma_buf_rht,
> + dma_buf_params);
> +err_dma_buf:
> + dma_buf_put(dma_buf);
> + return err;
> }
>
> static struct dma_buf *drm_prime_lookup_buf_by_handle(struct drm_prime_file_private *prime_fpriv,
> @@ -92,10 +136,10 @@ static struct dma_buf *drm_prime_lookup_buf_by_handle(struct drm_prime_file_priv
> {
> struct drm_prime_member *member;
>
> - list_for_each_entry(member, &prime_fpriv->head, entry) {
> - if (member->handle == handle)
> - return member->dma_buf;
> - }
> + member = rhashtable_lookup_fast(&prime_fpriv->handles,
> + &handle, handle_params);
> + if (member)
> + return member->dma_buf;
>
> return NULL;
> }
> @@ -106,12 +150,13 @@ static int drm_prime_lookup_buf_handle(struct drm_prime_file_private *prime_fpri
> {
> struct drm_prime_member *member;
>
> - list_for_each_entry(member, &prime_fpriv->head, entry) {
> - if (member->dma_buf == dma_buf) {
> - *handle = member->handle;
> - return 0;
> - }
> + member = rhashtable_lookup_fast(&prime_fpriv->dma_bufs,
> + &dma_buf, dma_buf_params);
> + if (member) {
> + *handle = member->handle;
> + return 0;
> }
> +
> return -ENOENT;
> }
>
> @@ -166,14 +211,21 @@ static void drm_gem_map_detach(struct dma_buf *dma_buf,
> void drm_prime_remove_buf_handle_locked(struct drm_prime_file_private *prime_fpriv,
> struct dma_buf *dma_buf)
> {
> - struct drm_prime_member *member, *safe;
> + struct drm_prime_member *member;
>
> - list_for_each_entry_safe(member, safe, &prime_fpriv->head, entry) {
> - if (member->dma_buf == dma_buf) {
> - dma_buf_put(dma_buf);
> - list_del(&member->entry);
> - kfree(member);
> - }
> + member = rhashtable_lookup_fast(&prime_fpriv->dma_bufs,
> + &dma_buf, dma_buf_params);
> + if (member) {
> + rhashtable_remove_fast(&prime_fpriv->dma_bufs,
> + &member->dma_buf_rht,
> + dma_buf_params);
> +
> + rhashtable_remove_fast(&prime_fpriv->handles,
> + &member->handle_rht,
> + handle_params);
> +
> + dma_buf_put(dma_buf);
> + kfree(member);
> }
> }
>
> @@ -759,12 +811,14 @@ EXPORT_SYMBOL(drm_prime_gem_destroy);
>
> void drm_prime_init_file_private(struct drm_prime_file_private *prime_fpriv)
> {
> - INIT_LIST_HEAD(&prime_fpriv->head);
> mutex_init(&prime_fpriv->lock);
> + rhashtable_init(&prime_fpriv->dma_bufs, &dma_buf_params);
> + rhashtable_init(&prime_fpriv->handles, &handle_params);
> }
>
> void drm_prime_destroy_file_private(struct drm_prime_file_private *prime_fpriv)
> {
> /* by now drm_gem_release should've made sure the list is empty */
> - WARN_ON(!list_empty(&prime_fpriv->head));
> + rhashtable_destroy(&prime_fpriv->dma_bufs);
> + rhashtable_destroy(&prime_fpriv->handles);
> }
> diff --git a/include/drm/drmP.h b/include/drm/drmP.h
> index c53dc90942e0..6966fb030d0f 100644
> --- a/include/drm/drmP.h
> +++ b/include/drm/drmP.h
> @@ -51,6 +51,7 @@
> #include <linux/platform_device.h>
> #include <linux/poll.h>
> #include <linux/ratelimit.h>
> +#include <linux/rhashtable.h>
> #include <linux/sched.h>
> #include <linux/slab.h>
> #include <linux/types.h>
> @@ -371,10 +372,10 @@ struct drm_pending_event {
> we deliver the event, for tracing only */
> };
>
> -/* initial implementaton using a linked list - todo hashtab */
> struct drm_prime_file_private {
> - struct list_head head;
> struct mutex lock;
> + struct rhashtable dma_bufs;
> + struct rhashtable handles;
> };
>
> /** File private data */
> --
> 2.9.3
>
> _______________________________________________
> Intel-gfx mailing list
> Intel-gfx at lists.freedesktop.org
> https://lists.freedesktop.org/mailman/listinfo/intel-gfx
More information about the dri-devel
mailing list