1 files changed, 1 insertions, 2 deletions

diff --git a/drivers/gpu/drm/amd/amdkfd/kfd_packet_manager.c b/drivers/gpu/drm/amd/amdkfd/kfd_packet_manager.c
index 0ecbd1f9b606..0c3bc00978f7 100644
--- a/drivers/gpu/drm/amd/amdkfd/kfd_packet_manager.c
+++ b/drivers/gpu/drm/amd/amdkfd/kfd_packet_manager.c
@@ -188,8 +188,7 @@ static int pm_create_map_process(struct packet_manager *pm, uint32_t *buffer,
         packet->sh_mem_ape1_base = qpd->sh_mem_ape1_base;
         packet->sh_mem_ape1_limit = qpd->sh_mem_ape1_limit;
 
-        /* TODO: scratch support */
-        packet->sh_hidden_private_base_vmid = 0;
+        packet->sh_hidden_private_base_vmid = qpd->sh_hidden_private_base;
 
         packet->gds_addr_lo = lower_32_bits(qpd->gds_context_area);
         packet->gds_addr_hi = upper_32_bits(qpd->gds_context_area);