From fd131e875eec437ac1a43fa73c355fa6eed641a7 Mon Sep 17 00:00:00 2001 From: Aaron Liu Date: Wed, 30 Jun 2021 19:44:53 +0800 Subject: [PATCH] kfdtest: MigrateLargeBufTest support APU Limit test buffer size to 3/4 total VRAM size, and max 1GB. Signed-off-by: Aaron Liu Change-Id: I937e10b0a6bd8215e3865b50f22ce75b3982a6f7 --- tests/kfdtest/src/KFDSVMRangeTest.cpp | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/kfdtest/src/KFDSVMRangeTest.cpp b/tests/kfdtest/src/KFDSVMRangeTest.cpp index 3bf7d95143..ecec01d4ac 100644 --- a/tests/kfdtest/src/KFDSVMRangeTest.cpp +++ b/tests/kfdtest/src/KFDSVMRangeTest.cpp @@ -803,11 +803,15 @@ TEST_F(KFDSVMRangeTest, MigrateLargeBufTest) { int defaultGPUNode = m_NodeInfo.HsaDefaultGPUNode(); ASSERT_GE(defaultGPUNode, 0) << "failed to get default GPU Node"; - if (!GetVramSize(defaultGPUNode)) { + HSAuint64 vramSize; + vramSize = GetVramSize(defaultGPUNode); + if (!vramSize) { LOG() << "Skipping test: No VRAM found." << std::endl; return; } + BufferSize = MIN(BufferSize, vramSize * 3 / 4); + HsaSVMRange SysBuffer(BufferSize, defaultGPUNode); SysBuffer.Fill(0x1);