rocrtst: Limit pool sizes to 2GB on emulator

Individual simple tests such as CPUAccessToGPUMemoryTest are taking several hours on emulators as the total amount of VRAM keeps increasing. Limit the pool sizes to 2GB, only on emulator. Change-Id: I4b33e8549f89413da255731e6748f606ca64a663 [ROCm/ROCR-Runtime commit: 588a5a2fd3]
2024-09-10 13:57:03 +00:00
parent 01c1e91c70
commit a0ccfce615
1 changed files with 6 additions and 0 deletions
@@ -400,6 +400,12 @@ hsa_status_t AcquirePoolInfo(hsa_amd_memory_pool_t pool,
                                                          &pool_i->size);
  RET_IF_HSA_COMMON_ERR(err);

+#ifdef ROCRTST_EMULATOR_BUILD
+  // Limit pool sizes to 2 GB on emulator
+  const size_t max_pool_size = 2*1024*1024*1024UL;
+  pool_i->size = std::min(pool_i->size, max_pool_size);
+#endif
+
  err = hsa_amd_memory_pool_get_info(pool,
             HSA_AMD_MEMORY_POOL_INFO_RUNTIME_ALLOC_ALLOWED,
                                                      &pool_i->alloc_allowed);