From 186b0843caaaa259e86a88ee5a3dec989315673e Mon Sep 17 00:00:00 2001
From: lyndonli <Lyndon.Li@amd.com>
Date: Tue, 10 Sep 2024 10:36:42 +0800
Subject: [PATCH] wsl/hsakmt: Set PRIV for gfx11

To fix the hang issue when enabling CWSR on gfx11.

Signed-off-by: lyndonli <Lyndon.Li@amd.com>
Reviewed-by: Longlong Yao <Longlong.Yao@amd.com>
Part-of: <http://10.67.69.192/wsl/libhsakmt/-/merge_requests/4>
---
 wddm/cmd_util.cpp | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/wddm/cmd_util.cpp b/wddm/cmd_util.cpp
index bffbe408ee..bfd29049d6 100644
--- a/wddm/cmd_util.cpp
+++ b/wddm/cmd_util.cpp
@@ -164,6 +164,10 @@ size_t CmdUtil::BuildDispatch(
 
   GenerateSetShRegHeader(&dispatch.program_resource_regs, mmCOMPUTE_PGM_RSRC1);
   dispatch.program_resource_regs.compute_pgm_rsrc1 = pInfo->pKernelObject->compute_pgm_rsrc1;
+  if (pInfo->major == 11) {
+    AMD_HSA_BITS_SET(dispatch.program_resource_regs.compute_pgm_rsrc1,
+        AMD_COMPUTE_PGM_RSRC_ONE_PRIV, 1);
+  }
   dispatch.program_resource_regs.compute_pgm_rsrc2 =
     (pInfo->ldsBlks << 15) | pInfo->pKernelObject->compute_pgm_rsrc2;