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]
这个提交包含在:
Philip Yang 编写, Chris Freehill 于 2024-07-18 10:20:05 -05:00 提交
1 父节点 405b60af73
当前提交 b6e18ceeae
1 个文件已更改
+5
@@ -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;