From 29820a088755a36acb50272dc75e4c913b72d904 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 [ROCm/ROCR-Runtime commit: 808a4428b6d9c7f32e7966517a0fd0571605ed04] --- projects/rocr-runtime/tests/kfdtest/src/KFDMemoryTest.cpp | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/projects/rocr-runtime/tests/kfdtest/src/KFDMemoryTest.cpp b/projects/rocr-runtime/tests/kfdtest/src/KFDMemoryTest.cpp index 6e44852cc8..2e76e546a0 100644 --- a/projects/rocr-runtime/tests/kfdtest/src/KFDMemoryTest.cpp +++ b/projects/rocr-runtime/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;