mirror of
https://github.com/Fishwaldo/Star64_linux.git
synced 2025-07-05 22:12:16 +00:00
drm/amdkfd: Free gang_ctx_bo and wptr_bo in pqm_uninit
[ Upstream commit 72838777aa38352e20301e123b97110c456cd38e ] [Why] Memory leaks of gang_ctx_bo and wptr_bo. [How] Free gang_ctx_bo and wptr_bo in pqm_uninit. v2: add a common function pqm_clean_queue_resource to free queue's resources. v3: reset pdd->pqd.num_gws when destorying GWS queue. Reviewed-by: Felix Kuehling <Felix.Kuehling@amd.com> Signed-off-by: ZhenGuo Yin <zhenguo.yin@amd.com> Signed-off-by: Alex Deucher <alexander.deucher@amd.com> Signed-off-by: Sasha Levin <sashal@kernel.org>
This commit is contained in:
parent
61dfdb7790
commit
ccba042816
1 changed files with 33 additions and 21 deletions
|
@ -169,16 +169,43 @@ int pqm_init(struct process_queue_manager *pqm, struct kfd_process *p)
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
static void pqm_clean_queue_resource(struct process_queue_manager *pqm,
|
||||||
|
struct process_queue_node *pqn)
|
||||||
|
{
|
||||||
|
struct kfd_node *dev;
|
||||||
|
struct kfd_process_device *pdd;
|
||||||
|
|
||||||
|
dev = pqn->q->device;
|
||||||
|
|
||||||
|
pdd = kfd_get_process_device_data(dev, pqm->process);
|
||||||
|
if (!pdd) {
|
||||||
|
pr_err("Process device data doesn't exist\n");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (pqn->q->gws) {
|
||||||
|
if (KFD_GC_VERSION(pqn->q->device) != IP_VERSION(9, 4, 3) &&
|
||||||
|
!dev->kfd->shared_resources.enable_mes)
|
||||||
|
amdgpu_amdkfd_remove_gws_from_process(
|
||||||
|
pqm->process->kgd_process_info, pqn->q->gws);
|
||||||
|
pdd->qpd.num_gws = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (dev->kfd->shared_resources.enable_mes) {
|
||||||
|
amdgpu_amdkfd_free_gtt_mem(dev->adev, pqn->q->gang_ctx_bo);
|
||||||
|
if (pqn->q->wptr_bo)
|
||||||
|
amdgpu_amdkfd_free_gtt_mem(dev->adev, pqn->q->wptr_bo);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
void pqm_uninit(struct process_queue_manager *pqm)
|
void pqm_uninit(struct process_queue_manager *pqm)
|
||||||
{
|
{
|
||||||
struct process_queue_node *pqn, *next;
|
struct process_queue_node *pqn, *next;
|
||||||
|
|
||||||
list_for_each_entry_safe(pqn, next, &pqm->queues, process_queue_list) {
|
list_for_each_entry_safe(pqn, next, &pqm->queues, process_queue_list) {
|
||||||
if (pqn->q && pqn->q->gws &&
|
if (pqn->q)
|
||||||
KFD_GC_VERSION(pqn->q->device) != IP_VERSION(9, 4, 3) &&
|
pqm_clean_queue_resource(pqm, pqn);
|
||||||
!pqn->q->device->kfd->shared_resources.enable_mes)
|
|
||||||
amdgpu_amdkfd_remove_gws_from_process(pqm->process->kgd_process_info,
|
|
||||||
pqn->q->gws);
|
|
||||||
kfd_procfs_del_queue(pqn->q);
|
kfd_procfs_del_queue(pqn->q);
|
||||||
uninit_queue(pqn->q);
|
uninit_queue(pqn->q);
|
||||||
list_del(&pqn->process_queue_list);
|
list_del(&pqn->process_queue_list);
|
||||||
|
@ -461,22 +488,7 @@ int pqm_destroy_queue(struct process_queue_manager *pqm, unsigned int qid)
|
||||||
goto err_destroy_queue;
|
goto err_destroy_queue;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (pqn->q->gws) {
|
pqm_clean_queue_resource(pqm, pqn);
|
||||||
if (KFD_GC_VERSION(pqn->q->device) != IP_VERSION(9, 4, 3) &&
|
|
||||||
!dev->kfd->shared_resources.enable_mes)
|
|
||||||
amdgpu_amdkfd_remove_gws_from_process(
|
|
||||||
pqm->process->kgd_process_info,
|
|
||||||
pqn->q->gws);
|
|
||||||
pdd->qpd.num_gws = 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (dev->kfd->shared_resources.enable_mes) {
|
|
||||||
amdgpu_amdkfd_free_gtt_mem(dev->adev,
|
|
||||||
pqn->q->gang_ctx_bo);
|
|
||||||
if (pqn->q->wptr_bo)
|
|
||||||
amdgpu_amdkfd_free_gtt_mem(dev->adev, pqn->q->wptr_bo);
|
|
||||||
|
|
||||||
}
|
|
||||||
uninit_queue(pqn->q);
|
uninit_queue(pqn->q);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue