diff --git a/projects/rocr-runtime/tests/kfdtest/src/KFDMemoryTest.cpp b/projects/rocr-runtime/tests/kfdtest/src/KFDMemoryTest.cpp
index 31f199f79f..8cf24ffb1b 100644
--- a/projects/rocr-runtime/tests/kfdtest/src/KFDMemoryTest.cpp
+++ b/projects/rocr-runtime/tests/kfdtest/src/KFDMemoryTest.cpp
@@ -965,7 +965,7 @@ void KFDMemoryTest::BinarySearchLargestBuffer(int allocNode, const HsaMemFlags &
     int ret;
 
     HsaMemMapFlags mapFlags = {0};
-    HSAuint64 granularityMB = 128;
+    HSAuint64 granularityMB = highMB > 512 ? 128 : 16;
 
     /* Testing big buffers in VRAM */
     unsigned int * pDb = NULL;