From 4fb2ed2c5a9fd8c1018e11fa6d182201a14b7a4f Mon Sep 17 00:00:00 2001 From: Flora Cui Date: Thu, 20 Nov 2025 10:14:19 +0800 Subject: [PATCH] librocdxg: fix vgpr count refer to https://github.com/ROCm/rocm-systems/pull/1807 Signed-off-by: Flora Cui --- queues.cpp | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/queues.cpp b/queues.cpp index db93a42c6b..edaaea9d1a 100644 --- a/queues.cpp +++ b/queues.cpp @@ -30,7 +30,10 @@ uint32_t get_vgpr_size_per_cu(HSA_ENGINE_ID id) { uint32_t vgpr_size = 0x40000; - if (id.ui32.Major >= 11) { + uint32_t gfxv = HSA_GET_GFX_VERSION_FULL(id.ui32); + if( gfxv == 0x1100 || gfxv == 0x1101 || + gfxv == 0x1151 || + gfxv == 0x1200 || gfxv ==0x1201) { vgpr_size = 0x60000; }