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

libhsakmt: Fix improper usage of hsaKmtRegisterMemoryToNodes

Browse Source 
It's unnecessary to register non-userptr.

Change-Id: Iefd329578365e036e2fe7e4d5c9c0c3d0976f67c
Signed-off-by: Lang Yu <Lang.Yu@amd.com>
Signed-off-by: Chris Freehill <cfreehil@amd.com>
This commit is contained in:
Lang Yu
2024-04-22 16:02:34 +08:00
committed by Chris Freehill
parent 5eb2a2660c
commit a7a712fb36
3 changed files with 8 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libhsakmt/tests/kfdtest/src/KFDLocalMemoryTest.cpp
-6
View File
@@ -529,32 +529,26 @@ TEST_F(KFDLocalMemoryTest, MapVramToGPUNodesTest) {
HsaMemMapFlags mapFlags = {0};
EXPECT_SUCCESS(hsaKmtAllocMemory(nodes[1], PAGE_SIZE, memFlags, &shared_addr));
EXPECT_SUCCESS(hsaKmtRegisterMemoryToNodes(shared_addr, PAGE_SIZE, 2, nodes));
EXPECT_SUCCESS(hsaKmtMapMemoryToGPUNodes(shared_addr, PAGE_SIZE, NULL, mapFlags, 2, nodes));
EXPECT_SUCCESS(hsaKmtQueryPointerInfo(shared_addr, &info));
EXPECT_EQ(info.NRegisteredNodes, 2);
EXPECT_EQ(info.NMappedNodes, 2);
EXPECT_SUCCESS(hsaKmtMapMemoryToGPUNodes(shared_addr, PAGE_SIZE, NULL, mapFlags, 1, &nodes[0]));
EXPECT_SUCCESS(hsaKmtQueryPointerInfo(shared_addr, &info));
EXPECT_EQ(info.NRegisteredNodes, 2);
EXPECT_EQ(info.NMappedNodes, 1);
EXPECT_EQ(info.MappedNodes[0], nodes[0]);
EXPECT_SUCCESS(hsaKmtMapMemoryToGPUNodes(shared_addr, PAGE_SIZE, NULL, mapFlags, 1, &nodes[1]));
EXPECT_SUCCESS(hsaKmtQueryPointerInfo(shared_addr, &info));
EXPECT_EQ(info.NRegisteredNodes, 2);
EXPECT_EQ(info.NMappedNodes, 1);
EXPECT_EQ(info.MappedNodes[0], nodes[1]);
EXPECT_SUCCESS(hsaKmtUnmapMemoryToGPU(shared_addr));
EXPECT_SUCCESS(hsaKmtQueryPointerInfo(shared_addr, &info));
EXPECT_EQ(info.NRegisteredNodes, 2);
EXPECT_EQ(info.NMappedNodes, 0);
EXPECT_SUCCESS(hsaKmtMapMemoryToGPUNodes(shared_addr, PAGE_SIZE, NULL, mapFlags, 1, &nodes[0]));
EXPECT_SUCCESS(hsaKmtQueryPointerInfo(shared_addr, &info));
EXPECT_EQ(info.NRegisteredNodes, 2);
EXPECT_EQ(info.NMappedNodes, 1);
EXPECT_EQ(info.MappedNodes[0], nodes[0]);