diff options
author | Felix Kuehling <Felix.Kuehling@amd.com> | 2017-11-28 02:29:50 +0300 |
---|---|---|
committer | Oded Gabbay <oded.gabbay@gmail.com> | 2017-11-28 02:29:50 +0300 |
commit | c7b1243eef128c49aabdfab71cd4f5ecb6602d97 (patch) | |
tree | 3f87a9bd0228506b0366328aad0c68297e39fb6b /drivers/gpu/drm/amd/amdkfd | |
parent | 851a645efd0fee7f27819ab30f3b353bace3cd98 (diff) | |
download | linux-c7b1243eef128c49aabdfab71cd4f5ecb6602d97.tar.xz |
drm/amdkfd: Get reference to lead_thread task struct
Increment the kfd_process.lead_thread's reference counter to make
it safe to dereference. This is needed for getting a safe reference
to the process' mm_struct.
Signed-off-by: Felix Kuehling <Felix.Kuehling@amd.com>
Reviewed-by: Oded Gabbay <oded.gabbay@gmail.com>
Signed-off-by: Oded Gabbay <oded.gabbay@gmail.com>
Diffstat (limited to 'drivers/gpu/drm/amd/amdkfd')
-rw-r--r-- | drivers/gpu/drm/amd/amdkfd/kfd_process.c | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/drivers/gpu/drm/amd/amdkfd/kfd_process.c b/drivers/gpu/drm/amd/amdkfd/kfd_process.c index 99c18ee65a76..660d8bcb93e0 100644 --- a/drivers/gpu/drm/amd/amdkfd/kfd_process.c +++ b/drivers/gpu/drm/amd/amdkfd/kfd_process.c @@ -24,6 +24,7 @@ #include <linux/log2.h> #include <linux/sched.h> #include <linux/sched/mm.h> +#include <linux/sched/task.h> #include <linux/slab.h> #include <linux/amd-iommu.h> #include <linux/notifier.h> @@ -191,6 +192,8 @@ static void kfd_process_wq_release(struct work_struct *work) mutex_destroy(&p->mutex); + put_task_struct(p->lead_thread); + kfree(p); kfree(work); @@ -342,6 +345,7 @@ static struct kfd_process *create_process(const struct task_struct *thread) (uintptr_t)process->mm); process->lead_thread = thread->group_leader; + get_task_struct(process->lead_thread); INIT_LIST_HEAD(&process->per_device_data); |