From 808a4428b6d9c7f32e7966517a0fd0571605ed04 Mon Sep 17 00:00:00 2001 From: Yifan Zhang Date: Thu, 28 Dec 2023 17:17:31 +0800 Subject: [PATCH] kfdtest: add APUs judgement in LargestVramBufferTest criteria This patch is to add APUs judgement in LargestVramBufferTest criteria. Change-Id: Ic69093f8ebed8be0b1c58787e2a294d86fb49bb0 --- tests/kfdtest/src/KFDMemoryTest.cpp | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) 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;