From 30cb62cca445944ae082116d29f7ebde11630993 Mon Sep 17 00:00:00 2001 From: Prike Liang Date: Fri, 25 Mar 2022 15:30:34 +0800 Subject: [PATCH] kfdtest: limit vram allocation size for MigrateAccessInPlaceTest On the some platform there's only 256MB vram and then will fail to allocate 256MB vram. So let's limit a small vram allocation for ensuring vram allocated successfully. Change-Id: Iba4c469de56925675e5624b300a6153e24ab19b3 Signed-off-by: Prike Liang [ROCm/ROCR-Runtime commit: c86a0b8332cf9f680794ba7bf9d120c286194932] --- projects/rocr-runtime/tests/kfdtest/src/KFDSVMRangeTest.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/projects/rocr-runtime/tests/kfdtest/src/KFDSVMRangeTest.cpp b/projects/rocr-runtime/tests/kfdtest/src/KFDSVMRangeTest.cpp index 3d4e64b6dd..655ad4ed66 100644 --- a/projects/rocr-runtime/tests/kfdtest/src/KFDSVMRangeTest.cpp +++ b/projects/rocr-runtime/tests/kfdtest/src/KFDSVMRangeTest.cpp @@ -690,7 +690,7 @@ TEST_F(KFDSVMRangeTest, MigrateAccessInPlaceTest) { return; } - unsigned int BufferSize = 256 << 20; + unsigned int BufferSize = MIN(256ULL << 20, GetVramSize(defaultGPUNode) / 2); SDMAQueue sdmaQueue; ASSERT_SUCCESS(sdmaQueue.Create(defaultGPUNode));