[Openmp-commits] [openmp] 56ec7ce - [openmp][amdgpu] Let fine grain and kernarg pools differ

[Jon Chesterfield via Openmp-commits]

Author: Jon Chesterfield
Date: 2022-12-14T02:04:21Z
New Revision: 56ec7ce80d5dac03cadfc4dc8c85132942ef5ca3

URL: https://github.com/llvm/llvm-project/commit/56ec7ce80d5dac03cadfc4dc8c85132942ef5ca3
DIFF: https://github.com/llvm/llvm-project/commit/56ec7ce80d5dac03cadfc4dc8c85132942ef5ca3.diff

LOG: [openmp][amdgpu] Let fine grain and kernarg pools differ

Added: 
    

Modified: 
    openmp/libomptarget/plugins/amdgpu/src/rtl.cpp

Removed: 
    


################################################################################
diff  --git a/openmp/libomptarget/plugins/amdgpu/src/rtl.cpp b/openmp/libomptarget/plugins/amdgpu/src/rtl.cpp
index b0e29cb6e4e96..ecf4a1661b0f7 100644
--- a/openmp/libomptarget/plugins/amdgpu/src/rtl.cpp
+++ b/openmp/libomptarget/plugins/amdgpu/src/rtl.cpp
@@ -939,9 +939,10 @@ class RTLDeviceInfoTy : HSALifetime {
         if (GlobalFlags & HSA_AMD_MEMORY_POOL_GLOBAL_FLAG_KERNARG_INIT) {
           KernArgPool = MemoryPool;
           KernArgPoolSet = true;
+        } else {
+          HostFineGrainedMemoryPool = MemoryPool;
+          FineGrainedMemoryPoolSet = true;
         }
-        HostFineGrainedMemoryPool = MemoryPool;
-        FineGrainedMemoryPoolSet = true;
       }
     }