kfdtest: Skip ContiguousVRAMAllocation if no enough VRAM

RDMATest.ContiguousVRAMAllocation test uses 4GB buffer, skip the test if
total VRAM size is less than 5GB, considering page table and other
reserved VRAM usage.

Change-Id: I0342417501cdd3477c2bf1b2f7d1e6bef61d1871
Signed-off-by: Philip Yang <Philip.Yang@amd.com>
Signed-off-by: Chris Freehill <cfreehil@amd.com>


[ROCm/ROCR-Runtime commit: 42df8b2b34]
This commit is contained in:
Philip Yang
2024-06-17 13:30:19 -04:00
کامیت شده توسط Chris Freehill
والد 405b60af73
کامیت b6e18ceeae
@@ -125,6 +125,11 @@ TEST_F(RDMATest, ContiguousVRAMAllocation) {
int defaultGPUNode = m_NodeInfo.HsaDefaultGPUNode();
ASSERT_GE(defaultGPUNode, 0) << "failed to get default GPU Node";
if (GetVramSize(defaultGPUNode) < BufferSize + (1UL << 30)) {
LOG() << "no enough VRAM, skipping the test" << std::endl;
return;
}
HsaMemoryBuffer isaBuffer(PAGE_SIZE, defaultGPUNode);
HsaMemoryBuffer srcSysBuffer(PAGE_SIZE, defaultGPUNode, false);
void *LocalBuffer;