diff --git a/tests/kfdtest/src/KFDMemoryTest.cpp b/tests/kfdtest/src/KFDMemoryTest.cpp
index 6e44852cc8..2e76e546a0 100644
--- a/tests/kfdtest/src/KFDMemoryTest.cpp
+++ b/tests/kfdtest/src/KFDMemoryTest.cpp
@@ -771,7 +771,11 @@ TEST_F(KFDMemoryTest, LargestVramBufferTest) {
             << lastTestedSizeMB << "MB" << std::endl;
 
     /* Make sure 3/5 vram can be allocated.*/
-    EXPECT_GE(lastTestedSizeMB * 5, vramSizeMB * 3);
+    if (vramSizeMB <= 512)
+        EXPECT_GE(lastTestedSizeMB * 5, vramSizeMB * 3);
+    else
+        EXPECT_GE(lastTestedSizeMB * 4, vramSizeMB * 3);
+
     if (lastTestedSizeMB * 16 < vramSizeMB * 15)
         WARN() << "The largest allocated VRAM buffer size is smaller than the expected "
             << vramSizeMB * 15 / 16 << "MB" << std::endl;