kfdtest: Add some asserts in BigBufferStressTest

It should have PASS/FAIL report for the vram allocated size. Change-Id: I546c02c2ed02f1cfb5278e0dfd7b18ade39faafb Signed-off-by: xinhui pan <xinhui.pan@amd.com> [ROCm/ROCR-Runtime commit: 1076075a1c]
2018-08-22 15:54:54 +08:00
parent b5cc397824
commit 3e0f1d695a
1 changed files with 6 additions and 0 deletions
@@ -772,6 +772,12 @@ void KFDMemoryTest::BigBufferVRAM(int defaultGPUNode, HSAuint64 granularityMB,
            << lastTestedSize << "MB" << std::endl;
    if (lastSize)
        *lastSize = lastTestedSize * 1024 * 1024;
+
+    /* Make sure 3/4 vram can be allocated.*/
+    EXPECT_GE(lastTestedSize * 4, vramSizeMB * 3);
+    if (lastTestedSize * 16 < vramSizeMB * 15)
+        WARN() << "The biggest allocated VRAM buffer size is smaller than the expected "
+            << vramSizeMB * 15 / 16 << "MB" << std::endl;
 }

 /* BigBufferStressTest allocs, maps/unmaps, and frees the biggest possible system