Add API for resetting user queues.

Signed-off-by: Alex Deucher <alexander.deuc...@amd.com>
---
 drivers/gpu/drm/amd/amdgpu/amdgpu_mes.c | 43 +++++++++++++++++++++++++
 drivers/gpu/drm/amd/amdgpu/amdgpu_mes.h |  9 ++++++
 2 files changed, 52 insertions(+)

diff --git a/drivers/gpu/drm/amd/amdgpu/amdgpu_mes.c 
b/drivers/gpu/drm/amd/amdgpu/amdgpu_mes.c
index 1739aa11cbd2..b3d6a9fa6775 100644
--- a/drivers/gpu/drm/amd/amdgpu/amdgpu_mes.c
+++ b/drivers/gpu/drm/amd/amdgpu/amdgpu_mes.c
@@ -774,6 +774,49 @@ int amdgpu_mes_remove_hw_queue(struct amdgpu_device *adev, 
int queue_id)
        return 0;
 }
 
+int amdgpu_mes_reset_hw_queue(struct amdgpu_device *adev, int queue_id)
+{
+       unsigned long flags;
+       struct amdgpu_mes_queue *queue;
+       struct amdgpu_mes_gang *gang;
+       struct mes_reset_queue_input queue_input;
+       int r;
+
+       /*
+        * Avoid taking any other locks under MES lock to avoid circular
+        * lock dependencies.
+        */
+       amdgpu_mes_lock(&adev->mes);
+
+       /* remove the mes gang from idr list */
+       spin_lock_irqsave(&adev->mes.queue_id_lock, flags);
+
+       queue = idr_find(&adev->mes.queue_id_idr, queue_id);
+       if (!queue) {
+               spin_unlock_irqrestore(&adev->mes.queue_id_lock, flags);
+               amdgpu_mes_unlock(&adev->mes);
+               DRM_ERROR("queue id %d doesn't exist\n", queue_id);
+               return -EINVAL;
+       }
+       spin_unlock_irqrestore(&adev->mes.queue_id_lock, flags);
+
+       DRM_DEBUG("try to reset queue, doorbell off = 0x%llx\n",
+                 queue->doorbell_off);
+
+       gang = queue->gang;
+       queue_input.doorbell_offset = queue->doorbell_off;
+       queue_input.gang_context_addr = gang->gang_ctx_gpu_addr;
+
+       r = adev->mes.funcs->reset_hw_queue(&adev->mes, &queue_input);
+       if (r)
+               DRM_ERROR("failed to reset hardware queue, queue id = %d\n",
+                         queue_id);
+
+       amdgpu_mes_unlock(&adev->mes);
+
+       return 0;
+}
+
 int amdgpu_mes_map_legacy_queue(struct amdgpu_device *adev,
                                struct amdgpu_ring *ring)
 {
diff --git a/drivers/gpu/drm/amd/amdgpu/amdgpu_mes.h 
b/drivers/gpu/drm/amd/amdgpu/amdgpu_mes.h
index 4456956c325b..771b63db1846 100644
--- a/drivers/gpu/drm/amd/amdgpu/amdgpu_mes.h
+++ b/drivers/gpu/drm/amd/amdgpu/amdgpu_mes.h
@@ -248,6 +248,11 @@ struct mes_remove_queue_input {
        uint64_t        gang_context_addr;
 };
 
+struct mes_reset_queue_input {
+       uint32_t        doorbell_offset;
+       uint64_t        gang_context_addr;
+};
+
 struct mes_map_legacy_queue_input {
        uint32_t                           queue_type;
        uint32_t                           doorbell_offset;
@@ -360,6 +365,9 @@ struct amdgpu_mes_funcs {
 
        int (*reset_legacy_queue)(struct amdgpu_mes *mes,
                                  struct mes_reset_legacy_queue_input *input);
+
+       int (*reset_hw_queue)(struct amdgpu_mes *mes,
+                             struct mes_reset_queue_input *input);
 };
 
 #define amdgpu_mes_kiq_hw_init(adev) (adev)->mes.kiq_hw_init((adev))
@@ -387,6 +395,7 @@ int amdgpu_mes_add_hw_queue(struct amdgpu_device *adev, int 
gang_id,
                            struct amdgpu_mes_queue_properties *qprops,
                            int *queue_id);
 int amdgpu_mes_remove_hw_queue(struct amdgpu_device *adev, int queue_id);
+int amdgpu_mes_reset_hw_queue(struct amdgpu_device *adev, int queue_id);
 
 int amdgpu_mes_map_legacy_queue(struct amdgpu_device *adev,
                                struct amdgpu_ring *ring);
-- 
2.45.2

Reply via email to