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));