From 53ed978c3d52b62eacbd64867cf2719b0f6651a8 Mon Sep 17 00:00:00 2001
From: Yifan Zhang <yifan1.zhang@amd.com>
Date: Thu, 23 Feb 2023 13:51:51 +0800
Subject: [PATCH] kfdtest: add non paged wptr judging API.

If MES is enabled, wptr has to be non paged memory,
Add an API to check this condition.

Change-Id: I53af1f6687d5332d102e7062c3d760e33b96e722
Signed-off-by: Yifan Zhang <yifan1.zhang@amd.com>
---
 tests/kfdtest/src/KFDBaseComponentTest.cpp | 5 +++++
 tests/kfdtest/src/KFDBaseComponentTest.hpp | 1 +
 2 files changed, 6 insertions(+)

diff --git a/tests/kfdtest/src/KFDBaseComponentTest.cpp b/tests/kfdtest/src/KFDBaseComponentTest.cpp
index bb510e040b..2c0625972e 100644
--- a/tests/kfdtest/src/KFDBaseComponentTest.cpp
+++ b/tests/kfdtest/src/KFDBaseComponentTest.cpp
@@ -164,6 +164,11 @@ unsigned int KFDBaseComponentTest::GetFamilyIdFromNodeId(unsigned int nodeId)
     return  FamilyIdFromNode(m_NodeInfo.GetNodeProperties(nodeId));
 }
 
+bool KFDBaseComponentTest::NeedNonPagedWptr(unsigned int nodeId)
+{
+    return GetFamilyIdFromNodeId(nodeId) >= FAMILY_GFX11;
+}
+
 int KFDBaseComponentTest::FindDRMRenderNode(int gpuNode) {
     HsaNodeProperties *nodeProperties;
     _HSAKMT_STATUS status;
diff --git a/tests/kfdtest/src/KFDBaseComponentTest.hpp b/tests/kfdtest/src/KFDBaseComponentTest.hpp
index cc87465ad5..8f714d54b3 100644
--- a/tests/kfdtest/src/KFDBaseComponentTest.hpp
+++ b/tests/kfdtest/src/KFDBaseComponentTest.hpp
@@ -58,6 +58,7 @@ class KFDBaseComponentTest : public testing::Test {
 // @return DRM Render Node if successful or -1 on failure
     int FindDRMRenderNode(int gpuNode);
     unsigned int GetFamilyIdFromNodeId(unsigned int nodeId);
+    bool NeedNonPagedWptr(unsigned int nodeId);
     unsigned int GetFamilyIdFromDefaultNode(){ return m_FamilyId; }
 
     // @brief Executed before the first test that uses KFDBaseComponentTest.