From: Alexey Skidanov <alexey.skida...@amd.com>

This patch splits the current kfd_get_process_device_data() to two
functions, one that specifically creates a pdd and another one which
just do lookup.

This is done to enhance the readability and maintainability of the code.

Signed-off-by: Alexey Skidanov <Alexey.Skidanov at amd.com>
Reviewed-by: Oded Gabbay <oded.gabbay at amd.com>
---
 .../gpu/drm/amd/amdkfd/kfd_device_queue_manager.c  |  1 -
 drivers/gpu/drm/amd/amdkfd/kfd_doorbell.c          |  4 ---
 drivers/gpu/drm/amd/amdkfd/kfd_flat_memory.c       | 13 ++++---
 drivers/gpu/drm/amd/amdkfd/kfd_priv.h              |  5 +--
 drivers/gpu/drm/amd/amdkfd/kfd_process.c           | 40 +++++++++++++---------
 .../gpu/drm/amd/amdkfd/kfd_process_queue_manager.c | 14 +++++---
 6 files changed, 46 insertions(+), 31 deletions(-)

diff --git a/drivers/gpu/drm/amd/amdkfd/kfd_device_queue_manager.c 
b/drivers/gpu/drm/amd/amdkfd/kfd_device_queue_manager.c
index f44d673..7b6df51 100644
--- a/drivers/gpu/drm/amd/amdkfd/kfd_device_queue_manager.c
+++ b/drivers/gpu/drm/amd/amdkfd/kfd_device_queue_manager.c
@@ -75,7 +75,6 @@ get_sh_mem_bases_nybble_64(struct kfd_process_device *pdd)
        nybble = (pdd->lds_base >> 60) & 0x0E;

        return nybble;
-
 }

 static inline unsigned int get_sh_mem_bases_32(struct kfd_process_device *pdd)
diff --git a/drivers/gpu/drm/amd/amdkfd/kfd_doorbell.c 
b/drivers/gpu/drm/amd/amdkfd/kfd_doorbell.c
index b5791a5..1a9b355 100644
--- a/drivers/gpu/drm/amd/amdkfd/kfd_doorbell.c
+++ b/drivers/gpu/drm/amd/amdkfd/kfd_doorbell.c
@@ -137,10 +137,6 @@ int kfd_doorbell_mmap(struct kfd_process *process, struct 
vm_area_struct *vma)
        if (dev == NULL)
                return -EINVAL;

-       /* Find if pdd exists for combination of process and gpu id */
-       if (!kfd_get_process_device_data(dev, process, 0))
-               return -EINVAL;
-
        /* Calculate physical address of doorbell */
        address = kfd_get_process_doorbells(dev, process);

diff --git a/drivers/gpu/drm/amd/amdkfd/kfd_flat_memory.c 
b/drivers/gpu/drm/amd/amdkfd/kfd_flat_memory.c
index e64aa99..5c91029 100644
--- a/drivers/gpu/drm/amd/amdkfd/kfd_flat_memory.c
+++ b/drivers/gpu/drm/amd/amdkfd/kfd_flat_memory.c
@@ -303,10 +303,11 @@ int kfd_init_apertures(struct kfd_process *process)
        while ((dev = kfd_topology_enum_kfd_devices(id)) != NULL &&
                id < NUM_OF_SUPPORTED_GPUS) {

-               pdd = kfd_get_process_device_data(dev, process, 1);
-               if (!pdd)
-                       return -1;
-
+               pdd = kfd_create_process_device_data(dev, process);
+               if (pdd == NULL) {
+                       pr_err("Failed to create process device data\n");
+                       goto err;
+               }
                /*
                 * For 64 bit process aperture will be statically reserved in
                 * the x86_64 non canonical process address space
@@ -349,6 +350,10 @@ int kfd_init_apertures(struct kfd_process *process)
        }

        return 0;
+
+err:
+       mutex_unlock(&process->mutex);
+       return -1;
 }


diff --git a/drivers/gpu/drm/amd/amdkfd/kfd_priv.h 
b/drivers/gpu/drm/amd/amdkfd/kfd_priv.h
index ba2bba8..a2e053c 100644
--- a/drivers/gpu/drm/amd/amdkfd/kfd_priv.h
+++ b/drivers/gpu/drm/amd/amdkfd/kfd_priv.h
@@ -473,8 +473,9 @@ struct kfd_process_device 
*kfd_bind_process_to_device(struct kfd_dev *dev,
                                                        struct kfd_process *p);
 void kfd_unbind_process_from_device(struct kfd_dev *dev, unsigned int pasid);
 struct kfd_process_device *kfd_get_process_device_data(struct kfd_dev *dev,
-                                                       struct kfd_process *p,
-                                                       int create_pdd);
+                                                       struct kfd_process *p);
+struct kfd_process_device *kfd_create_process_device_data(struct kfd_dev *dev,
+                                                       struct kfd_process *p);

 /* Process device data iterator */
 struct kfd_process_device *kfd_get_first_process_device_data(struct 
kfd_process *p);
diff --git a/drivers/gpu/drm/amd/amdkfd/kfd_process.c 
b/drivers/gpu/drm/amd/amdkfd/kfd_process.c
index 3c76ef0..a369c14 100644
--- a/drivers/gpu/drm/amd/amdkfd/kfd_process.c
+++ b/drivers/gpu/drm/amd/amdkfd/kfd_process.c
@@ -311,24 +311,29 @@ err_alloc_process:
 }

 struct kfd_process_device *kfd_get_process_device_data(struct kfd_dev *dev,
-                                                       struct kfd_process *p,
-                                                       int create_pdd)
+                                                       struct kfd_process *p)
 {
        struct kfd_process_device *pdd = NULL;

        list_for_each_entry(pdd, &p->per_device_data, per_device_list)
                if (pdd->dev == dev)
-                       return pdd;
-
-       if (create_pdd) {
-               pdd = kzalloc(sizeof(*pdd), GFP_KERNEL);
-               if (pdd != NULL) {
-                       pdd->dev = dev;
-                       INIT_LIST_HEAD(&pdd->qpd.queues_list);
-                       INIT_LIST_HEAD(&pdd->qpd.priv_queue_list);
-                       pdd->qpd.dqm = dev->dqm;
-                       list_add(&pdd->per_device_list, &p->per_device_data);
-               }
+                       break;
+
+       return pdd;
+}
+
+struct kfd_process_device *kfd_create_process_device_data(struct kfd_dev *dev,
+                                                       struct kfd_process *p)
+{
+       struct kfd_process_device *pdd = NULL;
+
+       pdd = kzalloc(sizeof(*pdd), GFP_KERNEL);
+       if (pdd != NULL) {
+               pdd->dev = dev;
+               INIT_LIST_HEAD(&pdd->qpd.queues_list);
+               INIT_LIST_HEAD(&pdd->qpd.priv_queue_list);
+               pdd->qpd.dqm = dev->dqm;
+               list_add(&pdd->per_device_list, &p->per_device_data);
        }

        return pdd;
@@ -344,11 +349,14 @@ struct kfd_process_device 
*kfd_get_process_device_data(struct kfd_dev *dev,
 struct kfd_process_device *kfd_bind_process_to_device(struct kfd_dev *dev,
                                                        struct kfd_process *p)
 {
-       struct kfd_process_device *pdd = kfd_get_process_device_data(dev, p, 1);
+       struct kfd_process_device *pdd;
        int err;

-       if (pdd == NULL)
+       pdd = kfd_get_process_device_data(dev, p);
+       if (!pdd) {
+               pr_err("Process device data doesn't exist\n");
                return ERR_PTR(-ENOMEM);
+       }

        if (pdd->bound)
                return pdd;
@@ -384,7 +392,7 @@ void kfd_unbind_process_from_device(struct kfd_dev *dev, 
unsigned int pasid)

        pqm_uninit(&p->pqm);

-       pdd = kfd_get_process_device_data(dev, p, 0);
+       pdd = kfd_get_process_device_data(dev, p);

        /*
         * Just mark pdd as unbound, because we still need it to call
diff --git a/drivers/gpu/drm/amd/amdkfd/kfd_process_queue_manager.c 
b/drivers/gpu/drm/amd/amdkfd/kfd_process_queue_manager.c
index 4752678..d12f9d3 100644
--- a/drivers/gpu/drm/amd/amdkfd/kfd_process_queue_manager.c
+++ b/drivers/gpu/drm/amd/amdkfd/kfd_process_queue_manager.c
@@ -167,8 +167,11 @@ int pqm_create_queue(struct process_queue_manager *pqm,
        q = NULL;
        kq = NULL;

-       pdd = kfd_get_process_device_data(dev, pqm->process, 1);
-       BUG_ON(!pdd);
+       pdd = kfd_get_process_device_data(dev, pqm->process);
+       if (!pdd) {
+               pr_err("Process device data doesn't exist\n");
+               return -1;
+       }

        retval = find_available_queue_slot(pqm, qid);
        if (retval != 0)
@@ -273,8 +276,11 @@ int pqm_destroy_queue(struct process_queue_manager *pqm, 
unsigned int qid)
                dev = pqn->q->device;
        BUG_ON(!dev);

-       pdd = kfd_get_process_device_data(dev, pqm->process, 1);
-       BUG_ON(!pdd);
+       pdd = kfd_get_process_device_data(dev, pqm->process);
+       if (!pdd) {
+               pr_err("Process device data doesn't exist\n");
+               return -1;
+       }

        if (pqn->kq) {
                /* destroy kernel queue (DIQ) */
-- 
1.9.1

Reply via email to