AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

kfdtest: Add thunk test for KFD memory availability ioctl

Browse Source 
Signed-off-by: Daniel Phillips <Daniel.Phillips@amd.com>
Change-Id: Ic4c1ffefdc3570718a1fce4e53ca5f1ebde8c479
This commit is contained in:
Daniel Phillips
2022-06-09 14:57:40 -07:00
committed by David Yat Sin
parent 14aa475905
commit 169673a435
1 changed files with 25 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/kfdtest/src/KFDMemoryTest.cpp
+25
View File
@@ -249,6 +249,31 @@ TEST_F(KFDMemoryTest, MemoryAlloc) {
TEST_END
}
// Basic test for hsaKmtAllocMemory
TEST_F(KFDMemoryTest, MemoryAllocAll) {
TEST_START(TESTPROFILE_RUNALL)
int defaultGPUNode = m_NodeInfo.HsaDefaultGPUNode();
HsaMemFlags memFlags = {0};
memFlags.ui32.NonPaged = 1; // sys mem vs vram
HSAuint64 available;
void *object = NULL;
int shrink = 21, success = HSAKMT_STATUS_NO_MEMORY;
EXPECT_SUCCESS(hsaKmtAvailableMemory(defaultGPUNode, &available));
LOG() << "Available: " << available << " bytes" << std::endl;
for (int i = 0; i < available >> shrink; i++) {
HSAuint64 size = available - ((HSAuint64)i << shrink);
if (hsaKmtAllocMemory(defaultGPUNode, size, memFlags, &object) == HSAKMT_STATUS_SUCCESS) {
LOG() << "Allocated: " << size << " bytes" << std::endl;
success = hsaKmtFreeMemory(object, available);
break;
}
}
EXPECT_SUCCESS(success);
TEST_END
}
TEST_F(KFDMemoryTest, AccessPPRMem) {
TEST_START(TESTPROFILE_RUNALL)