[PATCH v2 1/7] habanalabs: add handle field to context structure

[Oded Gabbay]

This patch adds a field to the context's structure that will hold a unique
handle for the context.

This will be needed when the user will create the context.

Signed-off-by: Oded Gabbay <oded.gab...@gmail.com>
---
 drivers/misc/habanalabs/context.c    | 27 ++++++++++++++++-----------
 drivers/misc/habanalabs/habanalabs.h |  2 ++
 2 files changed, 18 insertions(+), 11 deletions(-)

diff --git a/drivers/misc/habanalabs/context.c 
b/drivers/misc/habanalabs/context.c
index 8682590e3f6e..1d8390418234 100644
--- a/drivers/misc/habanalabs/context.c
+++ b/drivers/misc/habanalabs/context.c
@@ -67,9 +67,20 @@ int hl_ctx_create(struct hl_device *hdev, struct hl_fpriv 
*hpriv)
                goto out_err;
        }
 
+       mutex_lock(&mgr->ctx_lock);
+       rc = idr_alloc(&mgr->ctx_handles, ctx, 1, 0, GFP_KERNEL);
+       mutex_unlock(&mgr->ctx_lock);
+
+       if (rc < 0) {
+               dev_err(hdev->dev, "Failed to allocate IDR for a new CTX\n");
+               goto free_ctx;
+       }
+
+       ctx->handle = rc;
+
        rc = hl_ctx_init(hdev, ctx, false);
        if (rc)
-               goto free_ctx;
+               goto remove_from_idr;
 
        hl_hpriv_get(hpriv);
        ctx->hpriv = hpriv;
@@ -78,18 +89,12 @@ int hl_ctx_create(struct hl_device *hdev, struct hl_fpriv 
*hpriv)
        hpriv->ctx = ctx;
        hdev->user_ctx = ctx;
 
-       mutex_lock(&mgr->ctx_lock);
-       rc = idr_alloc(&mgr->ctx_handles, ctx, 1, 0, GFP_KERNEL);
-       mutex_unlock(&mgr->ctx_lock);
-
-       if (rc < 0) {
-               dev_err(hdev->dev, "Failed to allocate IDR for a new CTX\n");
-               hl_ctx_free(hdev, ctx);
-               goto out_err;
-       }
-
        return 0;
 
+remove_from_idr:
+       mutex_lock(&mgr->ctx_lock);
+       idr_remove(&mgr->ctx_handles, ctx->handle);
+       mutex_unlock(&mgr->ctx_lock);
 free_ctx:
        kfree(ctx);
 out_err:
diff --git a/drivers/misc/habanalabs/habanalabs.h 
b/drivers/misc/habanalabs/habanalabs.h
index 57183ae9b95d..e41800e68578 100644
--- a/drivers/misc/habanalabs/habanalabs.h
+++ b/drivers/misc/habanalabs/habanalabs.h
@@ -631,6 +631,7 @@ struct hl_va_range {
  *                             execution phase before the context switch phase
  *                             has finished.
  * @asid: context's unique address space ID in the device's MMU.
+ * @handle: context's opaque handle for user
  */
 struct hl_ctx {
        DECLARE_HASHTABLE(mem_hash, MEM_HASH_TABLE_BITS);
@@ -652,6 +653,7 @@ struct hl_ctx {
        atomic_t                thread_ctx_switch_token;
        u32                     thread_ctx_switch_wait_token;
        u32                     asid;
+       u32                     handle;
 };
 
 /**
-- 
2.17.1