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 <Prike.Liang@amd.com>


[ROCm/ROCR-Runtime commit: c86a0b8332]
This commit is contained in:
Prike Liang
2022-03-25 15:30:34 +08:00
parent 45f9fa1a55
commit 30cb62cca4
@@ -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));