From: Jack Xiao <[email protected]>

Free the mes gang and its resources.

Signed-off-by: Jack Xiao <[email protected]>
Acked-by: Christian König <[email protected]>
Reviewed-by: Hawking Zhang <[email protected]>
Signed-off-by: Alex Deucher <[email protected]>
---
 drivers/gpu/drm/amd/amdgpu/amdgpu_mes.c | 30 +++++++++++++++++++++++++
 drivers/gpu/drm/amd/amdgpu/amdgpu_mes.h |  1 +
 2 files changed, 31 insertions(+)

diff --git a/drivers/gpu/drm/amd/amdgpu/amdgpu_mes.c 
b/drivers/gpu/drm/amd/amdgpu/amdgpu_mes.c
index 74385e4b45c4..07ddf7bf6a3b 100644
--- a/drivers/gpu/drm/amd/amdgpu/amdgpu_mes.c
+++ b/drivers/gpu/drm/amd/amdgpu/amdgpu_mes.c
@@ -416,3 +416,33 @@ int amdgpu_mes_add_gang(struct amdgpu_device *adev, int 
pasid,
        mutex_unlock(&adev->mes.mutex);
        return r;
 }
+
+int amdgpu_mes_remove_gang(struct amdgpu_device *adev, int gang_id)
+{
+       struct amdgpu_mes_gang *gang;
+
+       mutex_lock(&adev->mes.mutex);
+
+       gang = idr_find(&adev->mes.gang_id_idr, gang_id);
+       if (!gang) {
+               DRM_ERROR("gang id %d doesn't exist\n", gang_id);
+               mutex_unlock(&adev->mes.mutex);
+               return -EINVAL;
+       }
+
+       if (!list_empty(&gang->queue_list)) {
+               DRM_ERROR("queue list is not empty\n");
+               mutex_unlock(&adev->mes.mutex);
+               return -EBUSY;
+       }
+
+       idr_remove(&adev->mes.gang_id_idr, gang->gang_id);
+       amdgpu_bo_free_kernel(&gang->gang_ctx_bo,
+                             &gang->gang_ctx_gpu_addr,
+                             &gang->gang_ctx_cpu_ptr);
+       list_del(&gang->list);
+       kfree(gang);
+
+       mutex_unlock(&adev->mes.mutex);
+       return 0;
+}
diff --git a/drivers/gpu/drm/amd/amdgpu/amdgpu_mes.h 
b/drivers/gpu/drm/amd/amdgpu/amdgpu_mes.h
index 3109bd1db6bc..f401a0a3eebd 100644
--- a/drivers/gpu/drm/amd/amdgpu/amdgpu_mes.h
+++ b/drivers/gpu/drm/amd/amdgpu/amdgpu_mes.h
@@ -228,5 +228,6 @@ void amdgpu_mes_destroy_process(struct amdgpu_device *adev, 
int pasid);
 int amdgpu_mes_add_gang(struct amdgpu_device *adev, int pasid,
                        struct amdgpu_mes_gang_properties *gprops,
                        int *gang_id);
+int amdgpu_mes_remove_gang(struct amdgpu_device *adev, int gang_id);
 
 #endif /* __AMDGPU_MES_H__ */
-- 
2.35.1

Reply via email to