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

rocrtst: Skip inaccessible agents when importing dmabuf

Browse Source 
If some agents cannot access the memory buffer directly, this will cause
the hsa_amd_interop_map_buffer API call to fail

Change-Id: If2f0e1735c2926440d657831de50775d7f304c8e


[ROCm/ROCR-Runtime commit: 2360253b3b]
This commit is contained in:
David Yat Sin
2024-08-27 20:42:46 +00:00
parent ed5b2e6661
commit a7b0f541b7
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rocr-runtime/rocrtst/suites/functional/memory_allocation.cc
+9 -1
View File
@@ -522,10 +522,18 @@ void MemoryAllocationTest::MemoryAllocateContiguousTest(hsa_agent_t agent,
uint64_t offset;
ASSERT_SUCCESS(hsa_amd_portable_export_dmabuf(memoryPtr, alloc_size, &dmabuf, &offset));
std::vector<hsa_agent_t> accessible_gpus;
for (auto gpuIter: gpus) {
hsa_amd_memory_pool_access_t access = HSA_AMD_MEMORY_POOL_ACCESS_NEVER_ALLOWED;
ASSERT_SUCCESS(hsa_amd_agent_memory_pool_get_info(gpuIter, pool, HSA_AMD_AGENT_MEMORY_POOL_INFO_ACCESS, &access));
if (access != HSA_AMD_MEMORY_POOL_ACCESS_NEVER_ALLOWED)
accessible_gpus.push_back(gpuIter);
}
void* importedPtr;
size_t importedSz;
ASSERT_SUCCESS(hsa_amd_interop_map_buffer(gpus.size(), gpus.data(), dmabuf, 0, &importedSz,
ASSERT_SUCCESS(hsa_amd_interop_map_buffer(accessible_gpus.size(), accessible_gpus.data(), dmabuf, 0, &importedSz,
&importedPtr, 0, NULL));
ASSERT_NE((uint64_t)importedPtr, 0);