From: Jack Xiao <[email protected]>

Use per context sdma csa address for mes sdma queue.

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_sdma.c | 24 ++++++++++++++++--------
 1 file changed, 16 insertions(+), 8 deletions(-)

diff --git a/drivers/gpu/drm/amd/amdgpu/amdgpu_sdma.c 
b/drivers/gpu/drm/amd/amdgpu/amdgpu_sdma.c
index e1835fd4b237..8e221a1ba937 100644
--- a/drivers/gpu/drm/amd/amdgpu/amdgpu_sdma.c
+++ b/drivers/gpu/drm/amd/amdgpu/amdgpu_sdma.c
@@ -74,14 +74,22 @@ uint64_t amdgpu_sdma_get_csa_mc_addr(struct amdgpu_ring 
*ring,
        if (amdgpu_sriov_vf(adev) || vmid == 0 || !amdgpu_mcbp)
                return 0;
 
-       r = amdgpu_sdma_get_index_from_ring(ring, &index);
-
-       if (r || index > 31)
-               csa_mc_addr = 0;
-       else
-               csa_mc_addr = amdgpu_csa_vaddr(adev) +
-                       AMDGPU_CSA_SDMA_OFFSET +
-                       index * AMDGPU_CSA_SDMA_SIZE;
+       if (ring->is_mes_queue) {
+               uint32_t offset = 0;
+
+               offset = offsetof(struct amdgpu_mes_ctx_meta_data,
+                                 sdma[ring->idx].sdma_meta_data);
+               csa_mc_addr = amdgpu_mes_ctx_get_offs_gpu_addr(ring, offset);
+       } else {
+               r = amdgpu_sdma_get_index_from_ring(ring, &index);
+
+               if (r || index > 31)
+                       csa_mc_addr = 0;
+               else
+                       csa_mc_addr = amdgpu_csa_vaddr(adev) +
+                               AMDGPU_CSA_SDMA_OFFSET +
+                               index * AMDGPU_CSA_SDMA_SIZE;
+       }
 
        return csa_mc_addr;
 }
-- 
2.35.1

Reply via email to