drm/amdkfd: Use the correct wptr size

Write pointer could be 32-bit or 64-bit. Use the correct size during
initialization.

Signed-off-by: Lijo Lazar <lijo.lazar@amd.com>
Acked-by: Alex Deucher <alexander.deucher@amd.com>
Signed-off-by: Alex Deucher <alexander.deucher@amd.com>
Cc: stable@vger.kernel.org

diff --git a/drivers/gpu/drm/amd/amdkfd/kfd_kernel_queue.c b/drivers/gpu/drm/amd/amdkfd/kfd_kernel_queue.c
index 55d18aed257bc2cf669722302021713987a1143a..2b0a830f5b294653bb809dbb9e1dab43eebdb8ad 100644 (file)
--- a/drivers/gpu/drm/amd/amdkfd/kfd_kernel_queue.c
+++ b/drivers/gpu/drm/amd/amdkfd/kfd_kernel_queue.c
@@ -125,7 +125,7 @@ static bool kq_initialize(struct kernel_queue *kq, struct kfd_node *dev,
 
        memset(kq->pq_kernel_addr, 0, queue_size);
        memset(kq->rptr_kernel, 0, sizeof(*kq->rptr_kernel));
-       memset(kq->wptr_kernel, 0, sizeof(*kq->wptr_kernel));
+       memset(kq->wptr_kernel, 0, dev->kfd->device_info.doorbell_size);
 
        prop.queue_size = queue_size;
        prop.is_interop = false;