From 4ebb9857ee0fbb774d02a05baff6fd8593e70630 Mon Sep 17 00:00:00 2001
From: Yifan Zhang <yifan1.zhang@amd.com>
Date: Wed, 15 Feb 2023 18:08:11 +0800
Subject: [PATCH] Fix rocrtstFunc.Memory_Max_Mem failure for APUs w/ small VRAM

Signed-off-by: Yifan Zhang <yifan1.zhang@amd.com>
Change-Id: I1c0f481af8b1d2a0939d28fb184ff6887747ab03
---
 rocrtst/suites/functional/memory_basic.cc | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/rocrtst/suites/functional/memory_basic.cc b/rocrtst/suites/functional/memory_basic.cc
index dec5812ed5..715eb60328 100755
--- a/rocrtst/suites/functional/memory_basic.cc
+++ b/rocrtst/suites/functional/memory_basic.cc
@@ -233,7 +233,11 @@ void MemoryTest::MaxSingleAllocationTest(hsa_agent_t ag,
   }
 
   if (ag_type == HSA_DEVICE_TYPE_GPU) {
-    EXPECT_GE((float)max_alloc_size/pool_sz, (float)3/4);
+    if (pool_sz <= 536870912) {
+      EXPECT_GE((float)max_alloc_size/pool_sz, (float)6/10);
+    } else {
+      EXPECT_GE((float)max_alloc_size/pool_sz, (float)3/4);
+    }
   }
   if (verbosity() > 0) {
     std::cout << kSubTestSeparator << std::endl;